replace ping with SELECT 1 (#1375)

Sometimes with replication connections `-- ping` isn't playing well, replace with `SELECT 1` --------- Co-authored-by: Philip Dubé <[email protected]>
PeerDB-io · Feb 25, 2024 · 91bc2e9 · 91bc2e9
1 parent a989cab
commit 91bc2e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flow/connectors/postgres/postgres.go b/flow/connectors/postgres/postgres.go
@@ -240,7 +240,7 @@ func (c *PostgresConnector) ConnectionActive(ctx context.Context) error {
 	if c.conn == nil {
 		return errors.New("connection is nil")
 	}
-	pingErr := c.conn.Ping(ctx)
+	_, pingErr := c.conn.Exec(ctx, "SELECT 1")
 	return pingErr
 }
 

diff --git a/flow/connectors/postgres/ssh_wrapped_pool.go b/flow/connectors/postgres/ssh_wrapped_pool.go
@@ -112,7 +112,7 @@ func (tunnel *SSHTunnel) NewPostgresConnFromConfig(
 
 	host := connConfig.Host
 	err = retryWithBackoff(logger, func() error {
-		err = conn.Ping(ctx)
+		_, err := conn.Exec(ctx, "SELECT 1")
 		if err != nil {
 			logger.Error("Failed to ping pool", slog.Any("error", err), slog.String("host", host))
 			return err