Implemented resending of data when performing a flash in buffered output

ClickHouse · Dec 17, 2024 · 0f1d5b9 · 0f1d5b9
1 parent 2a49a25
commit 0f1d5b9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/clickhouse/base/output.cpp b/clickhouse/base/output.cpp
@@ -81,7 +81,14 @@ void BufferedOutput::Reset() {
 
 void BufferedOutput::DoFlush() {
     if (array_output_.Data() != buffer_.data()) {
-        destination_->Write(buffer_.data(), array_output_.Data() - buffer_.data());
+        size_t len = array_output_.Data() - buffer_.data();
+        const uint8_t* buf = buffer_.data();
+        while (len > 0) {
+            const size_t written = destination_->Write(buf, len);
+            buf += written;
+            len -= written;
+        }
+
         destination_->Flush();
 
         array_output_.Reset(buffer_.data(), buffer_.size());