| 150 | } |
| 151 | |
| 152 | Status IcebergDeleteNode::NextProbeRowBatchFromChild( |
| 153 | RuntimeState* state, RowBatch* out_batch, bool* eos) { |
| 154 | DCHECK_ENUM_EQ(probe_state_, ProbeState::PROBING_END_BATCH); |
| 155 | DCHECK(probe_batch_pos_ == probe_batch_->num_rows() || probe_batch_pos_ == -1); |
| 156 | *eos = false; |
| 157 | do { |
| 158 | // Loop until we find a non-empty row batch. |
| 159 | probe_batch_->TransferResourceOwnership(out_batch); |
| 160 | if (out_batch->AtCapacity()) { |
| 161 | // This out batch is full. Need to return it before getting the next batch. |
| 162 | probe_batch_pos_ = -1; |
| 163 | return Status::OK(); |
| 164 | } |
| 165 | if (probe_side_eos_) { |
| 166 | probe_batch_pos_ = -1; |
| 167 | *eos = true; |
| 168 | return Status::OK(); |
| 169 | } |
| 170 | RETURN_IF_ERROR(child(0)->GetNext(state, probe_batch_.get(), &probe_side_eos_)); |
| 171 | COUNTER_ADD(probe_row_counter_, probe_batch_->num_rows()); |
| 172 | } while (probe_batch_->num_rows() == 0); |
| 173 | |
| 174 | ResetForProbe(); |
| 175 | return Status::OK(); |
| 176 | } |
| 177 | |
| 178 | Status IcebergDeleteNode::ProcessProbeBatch(RowBatch* out_batch) { |
| 179 | DCHECK_ENUM_EQ(probe_state_, ProbeState::PROBING_IN_BATCH); |
nothing calls this directly
no test coverage detected