adjust pg_replication_slots metrics

radondb · May 26, 2023 · 9422d8b · 9422d8b
1 parent fb316ad
commit 9422d8b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/cmd/postgres_exporter/queries.go b/cmd/postgres_exporter/queries.go
@@ -112,8 +112,9 @@ var queryOverrides = map[string][]OverrideQuery{
 		{
 			semver.MustParseRange(">=10.0.0"),
 			`
-			SELECT slot_name, database, active, pg_wal_lsn_diff(pg_current_wal_lsn(), restart_lsn)
-			FROM pg_replication_slots
+			SELECT slot_name, database, active, 
+			       (case pg_is_in_recovery() when 't' then pg_last_wal_receive_lsn() - restart_lsn else pg_current_wal_lsn() - restart_lsn end) as pg_wal_lsn_diff
+			FROM pg_replication_slots;
 			`,
 		},
 	},

diff --git a/collector/replication_slots.go b/collector/replication_slots.go
@@ -53,9 +53,9 @@ var pgReplicationSlot = map[string]*prometheus.Desc{
 
 func (PGReplicationSlotCollector) Update(ctx context.Context, db *sql.DB, ch chan<- prometheus.Metric) error {
 	rows, err := db.QueryContext(ctx,
-		`SELECT
+		`SELECT 
 			slot_name,
-			pg_current_wal_lsn() - '0/0' AS current_wal_lsn,
+			(case pg_is_in_recovery() when 't' then pg_last_wal_receive_lsn() - '0/0' else pg_current_wal_lsn() - '0/0' end) as current_wal_lsn,
 			coalesce(confirmed_flush_lsn, '0/0') - '0/0',
 			active
 		FROM