Backport #65771 to 24.3: PostgreSQL source support cancel query

2024-11-22 23:52:03 +00:00 · 2024-07-01 14:07:42 +00:00 · 2024-07-01 14:07:42 +00:00 · 45f1361433
commit 45f1361433
parent 297ceee4fc
1 changed files with 6 additions and 0 deletions
--- 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();
        }