Merge pull request #28 from mahlberg/monitoring-cascade-replication

dflemstr · web-flow · commit ce9766dc4412 · 2021-07-08T17:13:51.000+02:00
Support cascade replication in get_stats_replication_delays
diff --git a/postgresql_metrics/postgres_queries.py b/postgresql_metrics/postgres_queries.py
@@ -221,6 +221,10 @@ def get_replication_delays(conn):
     sql = ("SELECT client_addr, "
            "pg_xlog_location_diff(pg_current_xlog_location(), replay_location) AS bytes_diff "
            "FROM public.pg_stat_repl")
+    if is_in_recovery(conn):
+        # pg_current_xlog_location cannot be called in a replica
+        # use pg_last_xlog_receive_location for monitoring cascade replication
+        sql = sql.replace("pg_current_xlog_location", "pg_last_xlog_receive_location")
     if conn.server_version >= 100000: # PostgreSQL 10 and higher
         sql = sql.replace('_xlog', '_wal')
         sql = sql.replace('_location', '_lsn')
@@ -273,3 +277,7 @@ def get_wal_receiver_status(conn):
         host = CONNINFO_HOST_RE.search(conn_info).groupdict().get('host', 'UNKNOWN')
         host_replication_status.append((host, status))
     return host_replication_status
+
+
+def is_in_recovery(conn):
+    return query(conn, "SELECT pg_is_in_recovery()")[0][0]
