From d3f358ed7766b6e8d641fa669fde7febdfbfa019 Mon Sep 17 00:00:00 2001
From: ivyazmitinov <ivanvyazmitinov@gmail.com>
Date: Thu, 1 Feb 2024 15:10:57 +0100
Subject: [PATCH] - Address review - run citus_indent

---
 .../distributed/connection/connection_management.c  |  6 ------
 .../connection/shared_connection_stats.c            | 13 +++++--------
 src/backend/distributed/utils/maintenanced.c        |  2 +-
 3 files changed, 6 insertions(+), 15 deletions(-)

diff --git a/src/backend/distributed/connection/connection_management.c b/src/backend/distributed/connection/connection_management.c
index fa7f9f3bf..09e50bfa7 100644
--- a/src/backend/distributed/connection/connection_management.c
+++ b/src/backend/distributed/connection/connection_management.c
@@ -42,12 +42,6 @@
 #include "distributed/time_constants.h"
 #include "distributed/version_compat.h"
 #include "distributed/worker_log_messages.h"
-#include "mb/pg_wchar.h"
-#include "pg_config.h"
-#include "portability/instr_time.h"
-#include "storage/ipc.h"
-#include "utils/hsearch.h"
-#include "utils/memutils.h"
 
 
 int NodeConnectionTimeout = 30000;
diff --git a/src/backend/distributed/connection/shared_connection_stats.c b/src/backend/distributed/connection/shared_connection_stats.c
index 69ad0166a..4ad3f287e 100644
--- a/src/backend/distributed/connection/shared_connection_stats.c
+++ b/src/backend/distributed/connection/shared_connection_stats.c
@@ -13,6 +13,7 @@
 #include "postgres.h"
 
 #include "libpq-fe.h"
+#include "math.h"
 #include "miscadmin.h"
 #include "pgstat.h"
 
@@ -32,10 +33,8 @@
 #include "distributed/placement_connection.h"
 #include "distributed/shared_connection_stats.h"
 #include "distributed/time_constants.h"
-#include "distributed/worker_manager.h"
 #include "distributed/tuplestore.h"
 #include "distributed/worker_manager.h"
-#include "math.h"
 
 #define REMOTE_CONNECTION_STATS_COLUMNS 4
 
@@ -163,7 +162,6 @@ static void DecrementSharedConnectionCounterInternal(uint32 externalFlags, const
 
 PG_FUNCTION_INFO_V1(citus_remote_connection_stats);
 
-
 /*
  * citus_remote_connection_stats returns all the avaliable information about all
  * the remote connections (a.k.a., connections to remote nodes).
@@ -964,8 +962,7 @@ SharedWorkerNodeDatabaseHashHash(const void *key, Size keysize)
 {
 	SharedWorkerNodeDatabaseConnStatsHashKey *entry =
 		(SharedWorkerNodeDatabaseConnStatsHashKey *) key;
-	uint32 hash = string_hash(entry->workerNodeKey.hostname, NAMEDATALEN);
-	hash = hash_combine(hash, hash_uint32(entry->workerNodeKey.port));
+	uint32 hash = SharedConnectionHashHash(&(entry->workerNodeKey), keysize);
 	hash = hash_combine(hash, hash_uint32(entry->database));
 
 	return hash;
@@ -991,9 +988,9 @@ SharedWorkerNodeDatabaseHashCompare(const void *a, const void *b, Size keysize)
 	SharedWorkerNodeDatabaseConnStatsHashKey *cb =
 		(SharedWorkerNodeDatabaseConnStatsHashKey *) b;
 
-	return strncmp(ca->workerNodeKey.hostname, cb->workerNodeKey.hostname,
-				   MAX_NODE_LENGTH) != 0 ||
-		   ca->workerNodeKey.port != cb->workerNodeKey.port ||
+	int sharedConnectionHashCompare =
+		SharedConnectionHashCompare(&(ca->workerNodeKey), &(cb->workerNodeKey), keysize);
+	return sharedConnectionHashCompare ||
 		   ca->database != cb->database;
 }
 
diff --git a/src/backend/distributed/utils/maintenanced.c b/src/backend/distributed/utils/maintenanced.c
index f329ae001..785608729 100644
--- a/src/backend/distributed/utils/maintenanced.c
+++ b/src/backend/distributed/utils/maintenanced.c
@@ -49,6 +49,7 @@
 
 #include "distributed/background_jobs.h"
 #include "distributed/citus_safe_lib.h"
+#include "distributed/connection_management.h"
 #include "distributed/coordinator_protocol.h"
 #include "distributed/distributed_deadlock_detection.h"
 #include "distributed/maintenanced.h"
@@ -60,7 +61,6 @@
 #include "distributed/statistics_collection.h"
 #include "distributed/transaction_recovery.h"
 #include "distributed/version_compat.h"
-#include "distributed/connection_management.h"
 
 /*
  * Shared memory data for all maintenance workers.