Move to use prometheus_client

.gitignore

@@ -0,0 +1 @@
+.pyenv/

Dockerfile

@@ -12,7 +12,7 @@ ADD requirements.txt .
 RUN python -m pip install -r requirements.txt
 
 WORKDIR /app
-ADD . /app
+ADD dockstat.py /app
 
 # Switching to a non-root user, please refer to https://aka.ms/vscode-docker-python-user-rights
 # RUN useradd appuser && chown -R appuser /app

dockstat.py

@@ -5,23 +5,34 @@ Module to act as a Prometheus Exporter for Docker containers with a
 """
 
 import argparse
-import http.server
 import os.path
-import socketserver
 import sys
-import time
+from http.server import HTTPServer
 
 import docker
+from prometheus_client import (
+    CollectorRegistry,
+    Gauge,
+    generate_latest,
+    MetricsHandler,
+)
 
 LISTEN_PORT = 8080
 HEALTHY_STR = 'healthy'
 
 
-class SimpleHTTPRequestHandler(http.server.BaseHTTPRequestHandler):
+class HTTPHandler(MetricsHandler):
     """
     Class to encompass the requirements of a Prometheus Exporter
         for Docker containers with a healthcheck configured
     """
+
+    def __init__(self, *args, **kwargs):
+        self.docker_api = docker.APIClient()
+        self.docker_client = docker.from_env()
+        super().__init__(*args, **kwargs)
+
+    # Override built-in method
     # pylint: disable=invalid-name
     def do_GET(self):
         """
@@ -33,7 +44,6 @@ class SimpleHTTPRequestHandler(http.server.BaseHTTPRequestHandler):
         if self.path == '/healthcheck':
             self._healthcheck()
 
-
     def _healthcheck(self, message=True):
         """
         Method to return 200 or 500 response and an optional message
@@ -50,35 +60,37 @@ class SimpleHTTPRequestHandler(http.server.BaseHTTPRequestHandler):
         if message:
             self.wfile.write(b'OK')
 
-
     def _metrics(self):
         """
         Method to handle the request for metrics
         """
         self._healthcheck(message=False)
 
-        api = docker.APIClient()
+        registry = CollectorRegistry()
 
-        client = docker.from_env()
+        gauge = Gauge(
-        for container in client.containers.list():
+            'container_inspect_state_health_status',
-            now = int(round(time.time() * 1000))
+            "Container's healthcheck value (binary)",
-            data = api.inspect_container(container.id)
+            labelnames=['id', 'name', 'value'],
+            registry=registry
+        )
+
+        for container in self.docker_client.containers.list():
+            data = self.docker_api.inspect_container(container.id)
 
             try:
                 health_str = data["State"]["Health"]["Status"]
+                label_values = [
+                    container.id,
+                    container.name,
+                    health_str,
+                ]
             except KeyError:
                 pass
             else:
-                self.wfile.write(
+                gauge.labels(*label_values).set(int(health_str == HEALTHY_STR))
-                    bytes(
+
-                        f'container_inspect_state_health_status{{'
+        self.wfile.write(generate_latest(registry))
-                        f'id="{container.id}",'
-                        f'name="{container.name}",'
-                        f'value="{health_str}"'
-                        f'}} '
-                        f'{int(health_str == HEALTHY_STR)} {now}\n'.encode()
-                    )
-                )
 
 
 def healthy():
@@ -105,7 +117,6 @@ if __name__ == '__main__':
 
         return parser.parse_args()
 
-
     def main():
         """
         main()
@@ -116,11 +127,8 @@ if __name__ == '__main__':
             # Invert the sense of 'healthy' for Unix CLI usage
             return not healthy()
 
-        Handler = SimpleHTTPRequestHandler
+        HTTPServer(('', LISTEN_PORT), HTTPHandler).serve_forever()
 
-        with socketserver.TCPServer(('', LISTEN_PORT), Handler) as httpd:
+        return 0
-            httpd.serve_forever()
-
-        return True
 
     sys.exit(main())

requirements.txt

@@ -1,2 +1,3 @@
 # To ensure app dependencies are ported from your virtual environment/host machine into your container, run 'pip freeze > requirements.txt' in the terminal to overwrite this file
 docker
+prometheus_client