def receive_msg(self, in_key): encoded_msg = self.msg_handler.receive(in_key) if not encoded_msg: return None msg = message_decode(encoded_msg) msg.record_entry(self.component_name, self.logger) return msg
def test_message_encode(): msg = create_msg() encoded_msg = message_encode(msg) decoded_msg = message_decode(encoded_msg) assert msg.id == decoded_msg.id assert msg.source_address == decoded_msg.source_address assert msg.history == decoded_msg.history
def test_message_encode_shared_memory(): generator = DummyGenerator() msg = create_msg() encoded_msg = message_encode(msg, generator) decoded_msg = message_decode(encoded_msg) assert msg.id == decoded_msg.id assert msg.source_address == decoded_msg.source_address assert msg.history == decoded_msg.history generator.cleanup()
def test_prediction_payload_decode_encode(): preds = {"test": 1} msg = Message(preds.copy(), "localhost") assert isinstance(msg.payload, PredictionPayload) encoded_message = message_encode(msg) decoded_message = message_decode(encoded_message) decoded_message_data = decoded_message.get_payload() assert preds == decoded_message_data
def receive_msg(self, in_key, most_recent=True): if most_recent: encoded_msg = self.msg_handler.read_most_recent_msg(in_key) else: encoded_msg = self.msg_handler.receive(in_key) if not encoded_msg: return None msg = message_decode(encoded_msg) msg.record_entry(self.component_name, self.logger) return msg
def test_frame_metadata_payload_decode_encode(): metadata_dictionary = {"id": 2} img = np.random.rand(576, 720, 3) msg = Message((img, metadata_dictionary.copy()), "localhost") assert isinstance(msg.payload, FrameMetadataPayload) encoded_message = message_encode(msg) decoded_message = message_decode(encoded_message) decoded_message_data = decoded_message.get_payload() assert decoded_message_data[1] == metadata_dictionary
def main_logic(self, *args, **kwargs): encoded_msg = self.read_method(self.in_key) if encoded_msg: msg = message_decode(encoded_msg) msg.record_entry(self.component_name, self.logger) success = self.q_handler.deque_non_blocking_put(msg) return success else: time.sleep(0) return None
def main_logic(self, *args, **kwargs): encoded_msg = self.msg_handler.read_most_recent_msg(self.redis_read_key) if encoded_msg: msg = message_decode(encoded_msg) msg.record_entry(self.component_name, self.logger) try: self.message_queue.put(msg, block=False) return True except Full: try: self.message_queue.get(block=False) except Empty: pass finally: self.message_queue.put(msg, block=False) return True else: time.sleep(0) return False