From eadb18e7583fa4ae7b18dc25934ba4f3c99c7637 Mon Sep 17 00:00:00 2001
From: Onder Kalaci <onderkalaci@gmail.com>
Date: Wed, 8 Apr 2020 16:26:27 +0200
Subject: [PATCH] Receive cancellations

---
 src/backend/distributed/connection/shared_connection_stats.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/backend/distributed/connection/shared_connection_stats.c b/src/backend/distributed/connection/shared_connection_stats.c
index af4bac71f..341d53020 100644
--- a/src/backend/distributed/connection/shared_connection_stats.c
+++ b/src/backend/distributed/connection/shared_connection_stats.c
@@ -274,6 +274,8 @@ WaitLoopForSharedConnection(const char *hostname, int port)
 {
 	while (!TryToIncrementSharedConnectionCounter(hostname, port))
 	{
+		CHECK_FOR_INTERRUPTS();
+
 		WaitForSharedConnection();
 	}