Method SendCurrentBatch

be/src/runtime/krpc-data-stream-sender.cc:637–648

}

Status KrpcDataStreamSender::PartitionRowCollector::SendCurrentBatch() {
  if (collector_batch_->IsEmpty()) {
    DCHECK_EQ(num_rows_, 0);
    return Status::OK();
  }
  num_rows_ = 0;
  RETURN_IF_ERROR(channel_->GetParent()->PrepareBatchForSend(
      collector_batch_.get(), !channel_->IsLocal()));
  RETURN_IF_ERROR(channel_->TransmitData(&collector_batch_, true /*swap_batch*/));
  collector_batch_->Reset();
  return Status::OK();
}

void KrpcDataStreamSender::Channel::EndDataStreamCompleteCb() {
  std::unique_lock<SpinLock> l(lock_);