Method PostVisitMap

eval/compiler/flat_expr_builder.cc:1581–1618

  }

  void PostVisitMap(const cel::Expr& expr,
                    const cel::MapExpr& map_expr) override {
    for (const auto& entry : map_expr.entries()) {
      ValidateOrError(entry.has_key(), "Map entry missing key");
      ValidateOrError(entry.has_value(), "Map entry missing value");
    }

    if (!comprehension_stack_.empty()) {
      const ComprehensionStackRecord& comprehension =
          comprehension_stack_.back();
      if (comprehension.is_optimizable_map_insert) {
        if (&(comprehension.comprehension->accu_init()) == &expr) {
          if (PlanRecursiveProgram()) {
            SetRecursiveStep(CreateDirectMutableMapStep(expr.id()), 1);
            return;
          }
          AddStep(CreateMutableMapStep(expr.id()));
          return;
        }
      }
    }

    if (auto depth = RecursionEligible(); depth.has_value()) {
      auto deps = ExtractRecursiveDependencies();
      if (deps.size() != 2 * map_expr.entries().size()) {
        SetProgressStatusError(absl::InternalError(
            "Unexpected number of plan elements for CreateStruct expr"));
        return;
      }
      auto step = CreateDirectCreateMapStep(
          std::move(deps), MakeOptionalIndicesSet(map_expr), expr.id());
      SetRecursiveStep(std::move(step), *depth + 1);
      return;
    }
    AddStep(CreateCreateStructStepForMap(map_expr.entries().size(),
                                         MakeOptionalIndicesSet(map_expr),
                                         expr.id()));
  }

  absl::Status progress_status() const { return progress_status_; }