def run(self):
env = getdb()
client = KafkaClient(self.kafka_host)
consumer = self.consumer(client)
for resp in consumer:
if not resp.message.value:
continue
with env.begin(write=True) as txn:
message = IWAMessage.loads(resp.message.value)
if message:
txn.put(bytes(message.key), bytes(message.value))
def test_dumps_then_loads(self):
msg = IWAMessage('foo', 'bar')
print(msg.dumps())
assert msg == IWAMessage.loads(msg.dumps())
