From e33ba8b88dc992169546bae89b57089215d32d6b Mon Sep 17 00:00:00 2001 From: gifrerenom <lluis.gifre@cttc.es> Date: Fri, 17 Jan 2025 15:51:33 +0000 Subject: [PATCH] Telemetry backend component: - Added call to close() method - Fixed exception log messages in method close() --- .../backend/service/AnalyticsBackendService.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/src/analytics/backend/service/AnalyticsBackendService.py b/src/analytics/backend/service/AnalyticsBackendService.py index 92332df6f..f676b2c50 100755 --- a/src/analytics/backend/service/AnalyticsBackendService.py +++ b/src/analytics/backend/service/AnalyticsBackendService.py @@ -151,7 +151,7 @@ class AnalyticsBackendService(GenericGrpcService): LOGGER.error("Failed to stop Dask Streamer. ERROR: {:}".format(e)) return False - def close(self): # TODO: Is this function needed? + def close(self): """ Close the producer and cluster cleanly. """ @@ -159,11 +159,15 @@ class AnalyticsBackendService(GenericGrpcService): try: self.central_producer.flush() LOGGER.info("Kafka producer flushed and closed.") - except Exception as e: - LOGGER.error(f"Error closing Kafka producer: {e}") + except: + LOGGER.exception("Error closing Kafka producer") if self.cluster: try: self.cluster.close() LOGGER.info("Dask cluster closed.") - except Exception as e: - LOGGER.error(f"Error closing Dask cluster: {e}") + except: + LOGGER.exception("Error closing Dask cluster") + + def stop(self): + self.close() + return super().stop() -- GitLab