Fix error value handling in create list/map/struct

PiperOrigin-RevId: 736119595

Author: jcking

common/type_reflector_test.cc

@@ -15,7 +15,7 @@
 #ifndef THIRD_PARTY_CEL_CPP_COMMON_VALUE_H_
 #define THIRD_PARTY_CEL_CPP_COMMON_VALUE_H_
 
-#include <algorithm>
+#include <cstddef>
 #include <cstdint>
 #include <cstring>
 #include <memory>
@@ -2630,6 +2630,11 @@ static_assert(std::is_nothrow_move_constructible_v<Value>);
 static_assert(std::is_nothrow_move_assignable_v<Value>);
 static_assert(std::is_nothrow_swappable_v<Value>);
 
+inline absl::Status ErrorValueAssign::operator()(absl::Status status) const {
+  *value_ = ErrorValue(std::move(status));
+  return absl::OkStatus();
+}
+
 namespace common_internal {
 
 template <typename Base>
@@ -2855,9 +2860,11 @@ class ValueBuilder {
  public:
   virtual ~ValueBuilder() = default;
 
-  virtual absl::Status SetFieldByName(absl::string_view name, Value value) = 0;
+  virtual absl::StatusOr<absl::optional<ErrorValue>> SetFieldByName(
+      absl::string_view name, Value value) = 0;
 
-  virtual absl::Status SetFieldByNumber(int64_t number, Value value) = 0;
+  virtual absl::StatusOr<absl::optional<ErrorValue>> SetFieldByNumber(
+      int64_t number, Value value) = 0;
 
   virtual absl::StatusOr<Value> Build() && = 0;
 };

common/value.h

@@ -19,6 +19,7 @@
 #define THIRD_PARTY_CEL_CPP_COMMON_VALUES_ERROR_VALUE_H_
 
 #include <cstddef>
+#include <memory>
 #include <new>
 #include <ostream>
 #include <string>
@@ -217,6 +218,25 @@ class ErrorValueReturn final {
   }
 };
 
+class ErrorValueAssign final {
+ public:
+  ErrorValueAssign() = delete;
+
+  explicit ErrorValueAssign(Value& value ABSL_ATTRIBUTE_LIFETIME_BOUND)
+      : ErrorValueAssign(std::addressof(value)) {}
+
+  explicit ErrorValueAssign(
+      absl::Nonnull<Value*> value ABSL_ATTRIBUTE_LIFETIME_BOUND)
+      : value_(value) {
+    ABSL_DCHECK(value != nullptr);
+  }
+
+  absl::Status operator()(absl::Status status) const;
+
+ private:
+  absl::Nonnull<Value*> value_;
+};
+
 template <>
 struct ArenaTraits<ErrorValue> {
   static bool trivially_destructible(const ErrorValue& value) {

common/values/error_value.h

@@ -483,9 +483,11 @@ class StructValueBuilder {
  public:
   virtual ~StructValueBuilder() = default;
 
-  virtual absl::Status SetFieldByName(absl::string_view name, Value value) = 0;
+  virtual absl::StatusOr<absl::optional<ErrorValue>> SetFieldByName(
+      absl::string_view name, Value value) = 0;
 
-  virtual absl::Status SetFieldByNumber(int64_t number, Value value) = 0;
+  virtual absl::StatusOr<absl::optional<ErrorValue>> SetFieldByNumber(
+      int64_t number, Value value) = 0;
 
   virtual absl::StatusOr<StructValue> Build() && = 0;
 };