Method maybeMakeTraceConfig

bolt/exec/Task.cpp:3046–3109

}

std::optional<trace::TraceConfig> Task::maybeMakeTraceConfig() const {
  const auto& queryConfig = queryCtx_->queryConfig();
  if (!queryConfig.queryTraceEnabled()) {
    return std::nullopt;
  }

  BOLT_USER_CHECK(
      !queryConfig.queryTraceDir().empty(),
      "Query trace enabled but the trace dir is not set");

  BOLT_USER_CHECK(
      !queryConfig.queryTraceTaskRegExp().empty(),
      "Query trace enabled but the trace task regexp is not set");

  if (!RE2::FullMatch(taskId_, queryConfig.queryTraceTaskRegExp())) {
    return std::nullopt;
  }

  const auto traceNodes = queryConfig.queryTraceNodeIds();
  BOLT_USER_CHECK(!traceNodes.empty(), "Query trace nodes are not set");

  const auto traceDir = trace::getTaskTraceDirectory(
      queryConfig.queryTraceDir(), queryCtx_->queryId(), taskId_);

  std::vector<std::string> traceNodeIds;
  folly::split(',', traceNodes, traceNodeIds);
  std::unordered_set<std::string> traceNodeIdSet(
      traceNodeIds.begin(), traceNodeIds.end());
  BOLT_USER_CHECK_EQ(
      traceNodeIdSet.size(),
      traceNodeIds.size(),
      "Duplicate trace nodes found: {}",
      folly::join(", ", traceNodeIds));

  bool foundTraceNode{false};
  for (const auto& traceNodeId : traceNodeIds) {
    if (core::PlanNode::findFirstNode(
            planFragment_.planNode.get(),
            [traceNodeId](const core::PlanNode* node) -> bool {
              return node->id() == traceNodeId;
            })) {
      foundTraceNode = true;
      break;
    }
  }
  BOLT_USER_CHECK(
      foundTraceNode,
      "Trace plan nodes not found from task {}: {}",
      taskId_,
      folly::join(",", traceNodeIdSet));

  LOG(INFO) << "Trace input for plan nodes " << traceNodes << " from task "
            << taskId_;

  trace::UpdateAndCheckTraceLimitCB updateAndCheckTraceLimitCB =
      [this](uint64_t bytes) {
        queryCtx_->updateTracedBytesAndCheckLimit(bytes);
      };