Merge branch 'main' into one-sync-always-on

PeerDB-io · May 27, 2024 · 921aa29 · 921aa29
2 parents 8e2a2d1 + 3485158
commit 921aa29
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/flow/activities/flowable.go b/flow/activities/flowable.go
@@ -512,13 +512,7 @@ func (a *FlowableActivity) SendWALHeartbeat(ctx context.Context) error {
 		return err
 	}
 
-	command := `
-		BEGIN;
-		DROP AGGREGATE IF EXISTS PEERDB_EPHEMERAL_HEARTBEAT(float4);
-		CREATE AGGREGATE PEERDB_EPHEMERAL_HEARTBEAT(float4) (SFUNC = float4pl, STYPE = float4);
-		DROP AGGREGATE PEERDB_EPHEMERAL_HEARTBEAT(float4);
-		END;
-		`
+	command := peerdbenv.PeerDBWALHeartbeatQuery()
 	// run above command for each Postgres peer
 	for _, pgPeer := range pgPeers {
 		activity.RecordHeartbeat(ctx, pgPeer.Name)

diff --git a/flow/peerdbenv/config.go b/flow/peerdbenv/config.go
@@ -101,6 +101,15 @@ func PeerDBEnableWALHeartbeat() bool {
 	return getEnvBool("PEERDB_ENABLE_WAL_HEARTBEAT", false)
 }
 
+// PEERDB_WAL_HEARTBEAT_QUERY
+func PeerDBWALHeartbeatQuery() string {
+	return GetEnvString("PEERDB_WAL_HEARTBEAT_QUERY", `BEGIN;
+DROP AGGREGATE IF EXISTS PEERDB_EPHEMERAL_HEARTBEAT(float4);
+CREATE AGGREGATE PEERDB_EPHEMERAL_HEARTBEAT(float4) (SFUNC = float4pl, STYPE = float4);
+DROP AGGREGATE PEERDB_EPHEMERAL_HEARTBEAT(float4);
+END;`)
+}
+
 // PEERDB_ENABLE_PARALLEL_SYNC_NORMALIZE
 func PeerDBEnableParallelSyncNormalize() bool {
 	return getEnvBool("PEERDB_ENABLE_PARALLEL_SYNC_NORMALIZE", false)