diff options
| -rw-r--r-- | prometheus_pgbouncer_exporter/cli.py | 12 | ||||
| -rw-r--r-- | 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), |