From 28e375a90931917fd2d410478336ba359c014e5a Mon Sep 17 00:00:00 2001
From: Thomas Kluyver <thomas@kluyver.me.uk>
Date: Wed, 24 Jan 2024 11:51:40 +0000
Subject: [PATCH] Log success setting up Kakfa producer as well as failure
---
webservice/job_monitor.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/webservice/job_monitor.py b/webservice/job_monitor.py
index 7947f0046..f54de2dcb 100644
--- a/webservice/job_monitor.py
+++ b/webservice/job_monitor.py
@@ -43,7 +43,7 @@ class NoOpProducer:
def init_kafka_producer(config):
try:
- return KafkaProducer(
+ kp = KafkaProducer(
value_serializer=lambda d: json.dumps(d).encode('utf-8'),
max_block_ms=2000, # Don't get stuck trying to send Kafka messages
**config['kafka']['producer-config'].to_dict()
@@ -52,6 +52,10 @@ def init_kafka_producer(config):
log.warning("Problem initialising Kafka producer; "
"Kafka notifications will not be sent.", exc_info=True)
return NoOpProducer()
+ else:
+ log.info("Connected to Kafka broker (%s) to send notifications",
+ kp.config['bootstrap_servers'])
+ return kp
def slurm_status(filter_user=True):
--
GitLab