Updated app.py

This commit is contained in:
Brage 2023-07-18 02:48:09 +02:00
parent bf5fcd9093
commit d2263aaad4

View File

@ -56,6 +56,12 @@ class ProducerDataValueSchema:
return cls(referenceId, statusType, errorMessage, data) return cls(referenceId, statusType, errorMessage, data)
def decode_key(key_bytes):
return key_bytes.decode('utf-8') if key_bytes else None
def decode_value(value_bytes):
return json.loads(value_bytes.decode('utf-8')) if value_bytes else None
# Kafka consumer-klasse # Kafka consumer-klasse
class KafkaConsumerThread(threading.Thread): class KafkaConsumerThread(threading.Thread):
@ -67,7 +73,13 @@ class KafkaConsumerThread(threading.Thread):
self.shutdown = threading.Event() self.shutdown = threading.Event()
def run(self): def run(self):
consumer = KafkaConsumer(self.topic, bootstrap_servers=self.bootstrap_servers, group_id=self.consumer_group) consumer = KafkaConsumer(
self.topic,
bootstrap_servers=self.bootstrap_servers,
group_id=self.consumer_group,
key_deserializer=lambda x: decode_key(x),
value_deserializer=lambda x: decode_value(x)
)
logger.info("Kafka Consumer started") logger.info("Kafka Consumer started")