1 files changed, 103 insertions, 0 deletions
diff --git a/prometheus_pgbouncer_exporter/cli.py b/prometheus_pgbouncer_exporter/cli.py
new file mode 100644
index 0000000..7284baa
--- /dev/null
+++ b/prometheus_pgbouncer_exporter/cli.py
@@ -0,0 +1,103 @@
+#!/usr/bin/python3
+
+# Copyright (C) 2015  Christopher Baines <mail@cbaines.net>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import logging
+from http.server import HTTPServer
+
+import configargparse
+from prometheus_client.core import REGISTRY
+
+from .utils import get_connection
+from .exposition import RequestHandler
+from .collectors import StatsCollector, ListsCollector, PoolsCollector, \
+    DatabasesCollector
+
+
+def main():
+    p = configargparse.ArgParser(
+        default_config_files=[
+            '/etc/prometheus-pgbouncer-exporter/config',
+        ],
+    )
+
+    p.add(
+        '-c',
+        '--config',
+        is_config_file=True,
+        help='config file path',
+    )
+
+    p.add(
+        '--port',
+        default='6432',
+        help="Port to connect to pgbouncer",
+        env_var='PGBOUNCER_PORT',
+    )
+    p.add(
+        '--user',
+        default='pgbouncer',
+        help="User to connect to pgbouncer with",
+        env_var='PGBOUNCER_USER',
+    )
+
+    p.add(
+        '--database',
+        action='append',
+        help="Databases to report metrics for, if this is not specified, all metrics will be reported",
+        env_var='PGBOUNCER_DATABASES',
+    )
+
+    options = p.parse_args()
+
+    logging.basicConfig(level=logging.DEBUG)
+
+    logging.info(p.format_values())
+
+    connection = get_connection(options.user, options.port)
+
+    REGISTRY.register(StatsCollector(
+        connection=connection,
+        databases=options.database,
+    ))
+
+    REGISTRY.register(PoolsCollector(
+        connection=connection,
+        databases=options.database,
+    ))
+
+    REGISTRY.register(DatabasesCollector(
+        connection=connection,
+        databases=options.database,
+    ))
+
+    REGISTRY.register(ListsCollector(
+        connection=connection,
+    ))
+
+    host = '0.0.0.0'
+    port = 9127
+
+    httpd = HTTPServer((host, port), RequestHandler)
+
+    logging.info("Listing on port %s:%d" % (host, port))
+
+    try:
+        httpd.serve_forever()
+    except KeyboardInterrupt:
+        pass
+    finally:
+        connection.close()