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
