From 73bdf2bae092bd731a51238b7313e7c9ee0ec729 Mon Sep 17 00:00:00 2001
From: robot-clickhouse <robot-clickhouse@users.noreply.github.com>
Date: Mon, 1 Jul 2024 14:09:00 +0000
Subject: [PATCH] Backport #65771 to 24.6: PostgreSQL source support cancel
 query

---
 src/Processors/Sources/PostgreSQLSource.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/Processors/Sources/PostgreSQLSource.cpp b/src/Processors/Sources/PostgreSQLSource.cpp
index 4b828d6699c..66b6398f873 100644
--- a/src/Processors/Sources/PostgreSQLSource.cpp
+++ b/src/Processors/Sources/PostgreSQLSource.cpp
@@ -191,6 +191,12 @@ PostgreSQLSource<T>::~PostgreSQLSource()
     {
         try
         {
+            if (stream)
+            {
+                tx->conn().cancel_query();
+                stream->close();
+            }
+
             stream.reset();
             tx.reset();
         }