Method OperatorReplayerBase

bolt/tool/trace/OperatorReplayerBase.cpp:47–90

using namespace bytedance::bolt;
namespace bytedance::bolt::tool::trace {
OperatorReplayerBase::OperatorReplayerBase(
    std::string traceDir,
    std::string queryId,
    std::string taskId,
    std::string nodeId,
    std::string operatorType,
    const std::string& driverIds,
    uint64_t queryCapacity,
    folly::Executor* executor)
    : queryId_(std::string(std::move(queryId))),
      taskId_(std::move(taskId)),
      nodeId_(std::move(nodeId)),
      operatorType_(std::move(operatorType)),
      taskTraceDir_(
          exec::trace::getTaskTraceDirectory(traceDir, queryId_, taskId_)),
      nodeTraceDir_(exec::trace::getNodeTraceDirectory(taskTraceDir_, nodeId_)),
      fs_(filesystems::getFileSystem(taskTraceDir_, nullptr)),
      pipelineIds_(exec::trace::listPipelineIds(nodeTraceDir_, fs_)),
      driverIds_(
          driverIds.empty() ? exec::trace::listDriverIds(
                                  nodeTraceDir_,
                                  pipelineIds_.front(),
                                  fs_)
                            : exec::trace::extractDriverIds(driverIds)),
      queryCapacity_(queryCapacity == 0 ? memory::kMaxMemory : queryCapacity),
      executor_(executor) {
  BOLT_USER_CHECK(!taskTraceDir_.empty());
  BOLT_USER_CHECK(!taskId_.empty());
  BOLT_USER_CHECK(!nodeId_.empty());
  BOLT_USER_CHECK(!operatorType_.empty());
  if (operatorType_ == "HashJoin") {
    BOLT_USER_CHECK_EQ(pipelineIds_.size(), 2);
  } else {
    BOLT_USER_CHECK_EQ(pipelineIds_.size(), 1);
  }
  BOLT_CHECK_NOT_NULL(executor_);

  const auto taskMetaReader = exec::trace::TaskTraceMetadataReader(
      taskTraceDir_, memory::MemoryManager::getInstance()->tracePool());
  queryConfigs_ = taskMetaReader.queryConfigs();
  connectorConfigs_ = taskMetaReader.connectorProperties();
  planFragment_ = taskMetaReader.queryPlan();
  queryConfigs_[core::QueryConfig::kQueryTraceEnabled] = "false";
}

RowVectorPtr OperatorReplayerBase::run(bool copyResults) {
  auto queryCtx = createQueryCtx();