fix(stream-call): hungry mode to receive responses

gnes-ai · Oct 14, 2019 · 01531f7 · 01531f7
1 parent 0ac92e4
commit 01531f7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/gnes/service/frontend.py b/gnes/service/frontend.py
@@ -128,6 +128,11 @@ def get_response(num_recv, blocked=False):
                         self.pending_request -= 1
                         yield self.remove_envelope(msg)
 
+                while zmq_client.receiver.poll(5):
+                    msg = zmq_client.recv_message(**self.send_recv_kwargs)
+                    self.pending_request -= 1
+                    yield self.remove_envelope(msg)
+
             with self.zmq_context as zmq_client:
 
                 for request in request_iterator: