Skip to content
Snippets Groups Projects
Commit e33ba8b8 authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Telemetry backend component:

- Added call to close() method
- Fixed exception log messages in method close()
parent bd059d3d
No related branches found
No related tags found
2 merge requests!359Release TeraFlowSDN 5.0,!318Resolve "(CTTC) Analytics Module Enhancements"
...@@ -151,7 +151,7 @@ class AnalyticsBackendService(GenericGrpcService): ...@@ -151,7 +151,7 @@ class AnalyticsBackendService(GenericGrpcService):
LOGGER.error("Failed to stop Dask Streamer. ERROR: {:}".format(e)) LOGGER.error("Failed to stop Dask Streamer. ERROR: {:}".format(e))
return False return False
def close(self): # TODO: Is this function needed? def close(self):
""" """
Close the producer and cluster cleanly. Close the producer and cluster cleanly.
""" """
...@@ -159,11 +159,15 @@ class AnalyticsBackendService(GenericGrpcService): ...@@ -159,11 +159,15 @@ class AnalyticsBackendService(GenericGrpcService):
try: try:
self.central_producer.flush() self.central_producer.flush()
LOGGER.info("Kafka producer flushed and closed.") LOGGER.info("Kafka producer flushed and closed.")
except Exception as e: except:
LOGGER.error(f"Error closing Kafka producer: {e}") LOGGER.exception("Error closing Kafka producer")
if self.cluster: if self.cluster:
try: try:
self.cluster.close() self.cluster.close()
LOGGER.info("Dask cluster closed.") LOGGER.info("Dask cluster closed.")
except Exception as e: except:
LOGGER.error(f"Error closing Dask cluster: {e}") LOGGER.exception("Error closing Dask cluster")
def stop(self):
self.close()
return super().stop()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment