Adds support for the WSGI server to read from a different registry

Thomas Desrosiers · Thomas Desrosiers · commit 5c5d997ad548 · 2016-06-09T17:48:11.000-07:00
diff --git a/prometheus_client/exposition.py b/prometheus_client/exposition.py
@@ -28,21 +28,21 @@
 '''Content type of the latest text format'''
 
 
-def make_wsgi_app():
-    '''Create a WSGI app which serves the metrics from the registry.'''
+def make_wsgi_app(registry=core.REGISTRY):
+    '''Create a WSGI app which serves the metrics from a registry.'''
     def prometheus_app(environ, start_response):
         status = str('200 OK')
         headers = [(str('Content-type'), CONTENT_TYPE_LATEST)]
         start_response(status, headers)
-        return [generate_latest(core.REGISTRY)]
+        return [generate_latest(registry)]
     return prometheus_app
 
 
-def start_wsgi_server(port, addr=''):
+def start_wsgi_server(port, addr='', registry=core.REGISTRY):
     """Starts a WSGI server for prometheus metrics as a daemon thread."""
     class PrometheusMetricsServer(threading.Thread):
         def run(self):
-            httpd = make_server(addr, port, make_wsgi_app())
+            httpd = make_server(addr, port, make_wsgi_app(registry))
             httpd.serve_forever()
     t = PrometheusMetricsServer()
     t.daemon = True
