optimization, scheduling pull

Author: bigmontz

packages/bolt-connection/src/bolt/stream-observers.js

@@ -95,6 +95,7 @@ class ResultStreamObserver extends StreamObserver {
     this._setState(reactive ? _states.READY : _states.READY_STREAMING)
     this._setupAuoPull(fetchSize)
     this._pullMode = false;
+    this._pullScheduled = false;
   }
 
   setPullMode(pullMode) {
@@ -360,10 +361,11 @@ class ResultStreamObserver extends StreamObserver {
 
   _handleStreaming () {
     if (this._head && this._observers.some(o => o.onNext || o.onCompleted)) {
-      if (!this._pullMode && (this._discard || this._autoPull)) {
+      if (!this._pullMode && (this._discard || this._autoPull) || this._pullScheduled) {
         this._more()
       }
     }
+    this._pullScheduled = false
   }
 
   _more () {
@@ -373,6 +375,11 @@ class ResultStreamObserver extends StreamObserver {
       this._moreFunction(this._queryId, this._fetchSize, this)
     }
     this._setState(_states.STREAMING)
+    this._pullScheduled = false
+  }
+
+  _schedulePull () {
+    this._pullScheduled = true
   }
 
   _storeMetadataForCompletion (meta) {
@@ -631,7 +638,7 @@ const _states = {
     name: () => {
       return 'STREAMING'
     },
-    pull: () => {}
+    pull: streamObserver => streamObserver._schedulePull()
   },
   FAILED: {
     onError: error => {