如何捕获或保存所有gRPC流

时间:2019-07-13 07:44:33

标签: python grpc-python

我正在尝试编写python客户端来监听gRPC流(消防水带)。它不断保持流。没有“完成”。

原型:

rpc Start (StartParameters) returns (stream Progress) {}

在客户端中,我尝试编写以下内容,但是由于Start rpc不会返回“ on complete”,因此我无法将控件放入for循环中进行打印(事件)。

rsp = self.stub.Start(params)
for event in rsp:
    print(event)

有人可以在超时(2分钟)后用python代码帮助我处理或捕获rsp中的所有事件,然后在rsp中打印每个事件。

1 个答案:

答案 0 :(得分:0)

我已经开始工作了,如果有人正在寻找答案的话,发布这个消息

   def collect_responses(self, response_iterator, response_queue):
      for response in response_iterator:
         response_queue.put(response)


   def call_rpc(self)
      response_stream = stub.Start(params)
      response_queue = queue.Queue()
      thread = threading.Thread(target=self.collect_responses,
                                args=(response_stream, response_queue))
      thread.start()

      time.sleep(120) # or have a different trigger to say, cancel stream

      response_stream.cancel()
      thread.join()

      while not response_queue.empty():
         item = response_queue.get()
         print(item)