def send_buffered_messages(self):
"""Send messages in out_stream to the Stream Manager"""
while not self.out_stream.is_empty():
tuple_set = self.out_stream.poll()
msg = stmgr_pb2.TupleMessage()
msg.set.CopyFrom(tuple_set)
self._stmgr_client.send_message(msg)
def _register_msg_to_handle(self): # pylint: disable=unnecessary-lambda new_instance_builder = lambda: stmgr_pb2.NewInstanceAssignmentMessage() tuple_msg_builder = lambda: stmgr_pb2.TupleMessage() hts2_msg_builder = lambda: tuple_pb2.HeronTupleSet2() self.register_on_message(new_instance_builder) self.register_on_message(tuple_msg_builder) self.register_on_message(hts2_msg_builder)