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)