From f4585e65716404c8ee5ad22758b8967dad1966e7 Mon Sep 17 00:00:00 2001 From: Christopher Baines Date: Sat, 27 Feb 2016 17:53:34 +0000 Subject: Support connecting to pgbouncer on a different host --- prometheus_pgbouncer_exporter/cli.py | 12 +++++++++++- prometheus_pgbouncer_exporter/utils.py | 3 ++- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/prometheus_pgbouncer_exporter/cli.py b/prometheus_pgbouncer_exporter/cli.py index 6e4b5a0..a0b023f 100644 --- a/prometheus_pgbouncer_exporter/cli.py +++ b/prometheus_pgbouncer_exporter/cli.py @@ -73,6 +73,12 @@ def main(): help="User to connect to pgbouncer with", env_var='PGBOUNCER_USER', ) + p.add( + '--pgbouncer-host', + default=None, + help="Host on which to connect to pgbouncer", + env_var='PGBOUNCER_HOST', + ) p.add( '--database', @@ -98,7 +104,11 @@ def main(): logging.info(p.format_values()) - connection = get_connection(options.pgbouncer_user, options.pgbouncer_port) + connection = get_connection( + options.pgbouncer_user, + options.pgbouncer_port, + options.pgbouncer_host, + ) REGISTRY.register(StatsCollector( connection=connection, diff --git a/prometheus_pgbouncer_exporter/utils.py b/prometheus_pgbouncer_exporter/utils.py index 12e9ddd..7f9aae0 100644 --- a/prometheus_pgbouncer_exporter/utils.py +++ b/prometheus_pgbouncer_exporter/utils.py @@ -14,11 +14,12 @@ import psycopg2 -def get_connection(user, port): +def get_connection(user, port, host): connection = psycopg2.connect( database='pgbouncer', user=user, port=port, + host=host, ) # pgbouncer does not support transactions (as it does not make sense to), -- cgit v1.2.3