improve log sensor service daemon

edenia · Nov 3, 2023 · 7025c76 · 7025c76
1 parent 6dea026
commit 7025c76
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/tikuna/sensor/log_sensor.py b/tikuna/sensor/log_sensor.py
@@ -5,6 +5,7 @@
 import os
 import threading
 import logging
+import sys
 
 from os.path import join, dirname
 from dotenv import load_dotenv

diff --git a/tikuna/sensor/sensor_daemon.py b/tikuna/sensor/sensor_daemon.py
@@ -17,19 +17,29 @@
 
 class SensorDaemon(Daemon):
 
+    def __init__(self, pidfile, stdin='/dev/null',
+                 stdout='/dev/null', stderr='/dev/null'):
+        super(Daemon, self).__init__()
+        self.pidfile = pidfile
+        self.stdin = stdin
+        self.stdout = stdout
+        self.stderr = stderr
+        self.log_sensor_service = None
+
     def run(self):
         logging.basicConfig(filename=log_file,
                             level=logging.INFO)
         logging.info('Creating tikuna client services...')
-        log_sensor_service = LogSensor("prysm-beacon")
+        self.log_sensor_service = LogSensor("prysm-beacon")
         # Start the services.
         logging.info('Starting the tikuna client services...')
-        log_sensor_service.start()
+        self.log_sensor_service.start()
         logging.info("Sensors started.")
         logging.info('tikuna service client started...')
 
     def stop(self):
-        # TODO: do something here?
+        if self.log_sensor_service is not None:
+            self.log_sensor_service.stop()
         logging.info('Service tikuna stopped ...')
 
 if __name__ == "__main__":