[Fix](regressiontest) fix the p2 pipeline hang in multi stream sender

HappenLee · Jul 17, 2023 · 3e53d2e · 3e53d2e
1 parent 83e5a29
commit 3e53d2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/be/src/pipeline/exec/operator.h b/be/src/pipeline/exec/operator.h
@@ -273,7 +273,7 @@ class DataSinkOperator : public OperatorBase {
 
     Status sink(RuntimeState* state, vectorized::Block* in_block,
                 SourceState source_state) override {
-        if (in_block->rows() > 0) {
+        if (in_block->rows() > 0 || source_state == SourceState::FINISHED) {
             auto st = _sink->send(state, in_block, source_state == SourceState::FINISHED);
             // TODO: improvement: if sink returned END_OF_FILE, pipeline task can be finished
             if (st.template is<ErrorCode::END_OF_FILE>()) {