[fix](datastream sender) fix wrong result of BUCKET_SHFFULE_HASH_PART…

…ITIONED shuffle (#22973) fix wrong result of BUCKET_SHFFULE_HASH_PARTITIONED shuffle
apache · Aug 15, 2023 · 8318dfa · 8318dfa
1 parent 911bd0e
commit 8318dfa
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/be/src/vec/sink/vdata_stream_sender.cpp b/be/src/vec/sink/vdata_stream_sender.cpp
@@ -596,7 +596,9 @@ Status VDataStreamSender::send(RuntimeState* state, Block* block, bool eos) {
 
         // vectorized calculate hash
         int rows = block->rows();
-        auto element_size = _channels.size();
+        auto element_size = _part_type == TPartitionType::HASH_PARTITIONED
+                                    ? _channels.size()
+                                    : _channel_shared_ptrs.size();
         std::vector<uint64_t> hash_vals(rows);
         auto* __restrict hashes = hash_vals.data();
 
@@ -630,7 +632,6 @@ Status VDataStreamSender::send(RuntimeState* state, Block* block, bool eos) {
                             .first->update_crcs_with_value(
                                     hash_vals, _partition_expr_ctxs[j]->root()->type().type);
                 }
-                element_size = _channel_shared_ptrs.size();
                 for (int i = 0; i < rows; i++) {
                     hashes[i] = hashes[i] % element_size;
                 }