diff --git a/webservice/job_monitor.py b/webservice/job_monitor.py index 7947f004635fa723d614e506da01d0181e073098..f54de2dcb8d1a7abf32b23cc2436fbef2d508aa7 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):