From efe6f63ebaa468e5405bb4b8b0cb95e1dc6b51d9 Mon Sep 17 00:00:00 2001
From: Konstantin Knizhnik <knizhnik@neon.tech>
Date: Tue, 21 Jan 2025 12:19:23 +0200
Subject: [PATCH] Use keepalives for compute to PS connection

---
 pgxn/neon/libpagestore.c | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/pgxn/neon/libpagestore.c b/pgxn/neon/libpagestore.c
index 4460e3b40c..7348bc58ce 100644
--- a/pgxn/neon/libpagestore.c
+++ b/pgxn/neon/libpagestore.c
@@ -373,8 +373,8 @@ pageserver_connect(shardno_t shard_no, int elevel)
 	{
 	case PS_Disconnected:
 	{
-		const char *keywords[3];
-		const char *values[3];
+		const char *keywords[4];
+		const char *values[4];
 		int			n_pgsql_params;
 		TimestampTz	now;
 		int64		us_since_last_attempt;
@@ -421,12 +421,14 @@ pageserver_connect(shardno_t shard_no, int elevel)
 		 */
 		keywords[0] = "dbname";
 		values[0] = connstr;
-		n_pgsql_params = 1;
+		keywords[1] = "keepalives";
+		values[1] = "1";
+		n_pgsql_params = 2;
 
 		if (neon_auth_token)
 		{
-			keywords[1] = "password";
-			values[1] = neon_auth_token;
+			keywords[2] = "password";
+			values[2] = neon_auth_token;
 			n_pgsql_params++;
 		}