Add prometheus endpoint to celery worker. (google#836)

jonathan-greig · Jul 8, 2021 · 75a60ee · 75a60ee
1 parent 963a97d
commit 75a60ee
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/turbinia/client.py b/turbinia/client.py
@@ -1244,6 +1244,13 @@ def __init__(self, jobs_denylist=None, jobs_allowlist=None):
 
   def start(self):
     """Start Turbinia Celery Worker."""
+    if config.PROMETHEUS_ENABLED:
+      if config.PROMETHEUS_PORT and config.PROMETHEUS_ADDR:
+        log.info('Starting Prometheus endpoint.')
+        start_http_server(
+            port=config.PROMETHEUS_PORT, addr=config.PROMETHEUS_ADDR)
+      else:
+        log.info('Prometheus enabled but port or address not set!')
     log.info('Running Turbinia Celery Worker.')
     self.worker.task(task_manager.task_runner, name='task_runner')
     argv = ['celery', 'worker', '--loglevel=info', '--pool=solo']