Function LookupInMap

eval/eval/container_access_step.cc:92–159

}

void LookupInMap(const MapValue& cel_map, const Value& key,
                 ExecutionFrameBase& frame, Value& result) {
  if (frame.options().enable_heterogeneous_equality) {
    // Double isn't a supported key type but may be convertible to an integer.
    absl::optional<Number> number = CelNumberFromValue(key);
    if (number.has_value()) {
      // Consider uint as uint first then try coercion (prefer matching the
      // original type of the key value).
      if (key->Is<UintValue>()) {
        auto lookup =
            cel_map.Find(key, frame.descriptor_pool(), frame.message_factory(),
                         frame.arena(), &result);
        if (!lookup.ok()) {
          result = cel::ErrorValue(std::move(lookup).status());
          return;
        }
        if (*lookup) {
          ABSL_DCHECK(!result.IsUnknown());
          return;
        }
      }
      // double / int / uint -> int
      if (number->LosslessConvertibleToInt()) {
        auto lookup =
            cel_map.Find(IntValue(number->AsInt()), frame.descriptor_pool(),
                         frame.message_factory(), frame.arena(), &result);
        if (!lookup.ok()) {
          result = cel::ErrorValue(std::move(lookup).status());
          return;
        }
        if (*lookup) {
          ABSL_DCHECK(!result.IsUnknown());
          return;
        }
      }
      // double / int -> uint
      if (number->LosslessConvertibleToUint()) {
        auto lookup =
            cel_map.Find(UintValue(number->AsUint()), frame.descriptor_pool(),
                         frame.message_factory(), frame.arena(), &result);
        if (!lookup.ok()) {
          result = cel::ErrorValue(std::move(lookup).status());
          return;
        }
        if (*lookup) {
          ABSL_DCHECK(!result.IsUnknown());
          return;
        }
      }
      result = cel::ErrorValue(CreateNoSuchKeyError(key->DebugString()));
      return;
    }
  }

  absl::Status status = CheckMapKeyType(key);
  if (!status.ok()) {
    result = cel::ErrorValue(std::move(status));
    return;