Merge pull request ClickHouse#65929 from ClickHouse/backport/24.3/65771

Backport ClickHouse#65771 to 24.3: PostgreSQL source support cancel query
Altinity · Jul 1, 2024 · fe54cea · fe54cea
2 parents 297ceee + 45f1361
commit fe54cea
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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();
         }