From 94eb0782a945f6276481dc14262cd90d27dd1ebd Mon Sep 17 00:00:00 2001
From: yariks5s <yaroslav.briukhovetskyi@clickhouse.com>
Date: Wed, 10 Apr 2024 22:22:13 +0000
Subject: [PATCH] fix for multiple threads

---
 src/Processors/Transforms/BalancingTransform.cpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/Processors/Transforms/BalancingTransform.cpp b/src/Processors/Transforms/BalancingTransform.cpp
index a6a79f65ea4..73672be5da4 100644
--- a/src/Processors/Transforms/BalancingTransform.cpp
+++ b/src/Processors/Transforms/BalancingTransform.cpp
@@ -53,7 +53,7 @@ IProcessor::Status BalancingChunksTransform::prepareConsume()
                 return Status::Finished;
             }
 
-            if (input.isFinished())
+            if (input.isFinished() && !balance.isDataLeft())
             {
                 for (auto & output : outputs)
                     output.finish();
@@ -117,7 +117,10 @@ IProcessor::Status BalancingChunksTransform::prepareSend()
         ++chunk_number;
 
         if (!chunk.hasChunkInfo())
-            throw Exception(ErrorCodes::LOGICAL_ERROR, "Chunk info must be not empty in prepareGenerate()");
+        {
+            has_data = false;
+            return Status::Ready;
+        }
 
         if (was_processed)
             continue;