Method InputReceived

cpp/src/arrow/acero/hash_join_node.cc:902–926

  }

  Status InputReceived(ExecNode* input, ExecBatch batch) override {
    auto scope = TraceInputReceived(batch);
    ARROW_DCHECK(std::find(inputs_.begin(), inputs_.end(), input) != inputs_.end());
    if (complete_.load()) {
      return Status::OK();
    }

    size_t thread_index = plan_->query_context()->GetThreadIndex();
    int side = (input == inputs_[0]) ? 0 : 1;

    if (side == 0) {
      ARROW_RETURN_NOT_OK(OnProbeSideBatch(thread_index, std::move(batch)));
    } else {
      ARROW_RETURN_NOT_OK(OnBuildSideBatch(thread_index, std::move(batch)));
    }

    if (batch_count_[side].Increment()) {
      if (side == 0) {
        return OnProbeSideFinished(thread_index);
      } else {
        return OnBuildSideFinished(thread_index);
      }
    }
    return Status::OK();
  }

  Status InputFinished(ExecNode* input, int total_batches) override {
    ARROW_DCHECK(std::find(inputs_.begin(), inputs_.end(), input) != inputs_.end());