Remove extra flush when closing piped output stream

ClickHouse · May 10, 2022 · c3dbfe3 · c3dbfe3
1 parent 8eb0fa7
commit c3dbfe3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/clickhouse-client/src/main/java/com/clickhouse/client/stream/BlockingPipedOutputStream.java b/clickhouse-client/src/main/java/com/clickhouse/client/stream/BlockingPipedOutputStream.java
@@ -102,7 +102,10 @@ public void close() throws IOException {
             Thread.currentThread().interrupt();
             throw new IOException("Thread was interrupted when putting EMPTY buffer into queue", e);
         } finally {
-            super.close();
+            closed = true;
+            if (postCloseAction != null) {
+                postCloseAction.run();
+            }
         }
     }
 

diff --git a/...house-client/src/main/java/com/clickhouse/client/stream/NonBlockingPipedOutputStream.java b/...house-client/src/main/java/com/clickhouse/client/stream/NonBlockingPipedOutputStream.java
@@ -116,7 +116,11 @@ public void close() throws IOException {
             for (int i = 0, len = buckets.length; i < len; i++) {
                 buckets[i] = null;
             }
-            super.close();
+
+            closed = true;
+            if (postCloseAction != null) {
+                postCloseAction.run();
+            }
         }
     }