YT-18920: Add `TryDecodeEnumValue()`

1 files changed, 21 insertions, 4 deletions

diff --git a/library/cpp/yt/string/enum.cpp b/library/cpp/yt/string/enum.cpp
index 7cb8e5c6b6..e61b813fd2 100644
--- a/library/cpp/yt/string/enum.cpp
+++ b/library/cpp/yt/string/enum.cpp
@@ -6,18 +6,35 @@ namespace NYT {
 
 ////////////////////////////////////////////////////////////////////////////////
 
-TString DecodeEnumValue(TStringBuf value)
+template <bool failOnError>
+std::optional<TString> DecodeEnumValueImpl(TStringBuf value)
 {
     auto camelValue = UnderscoreCaseToCamelCase(value);
     auto underscoreValue = CamelCaseToUnderscoreCase(camelValue);
     if (value != underscoreValue) {
-        throw TSimpleException(Format("Enum value %Qv is not in a proper underscore case; did you mean %Qv?",
-            value,
-            underscoreValue));
+        if constexpr (failOnError) {
+            throw TSimpleException(Format("Enum value %Qv is not in a proper underscore case; did you mean %Qv?",
+                value,
+                underscoreValue));
+        } else {
+            return std::nullopt;
+        }
     }
     return camelValue;
 }
 
+std::optional<TString> TryDecodeEnumValue(TStringBuf value)
+{
+    return DecodeEnumValueImpl<false>(value);
+}
+
+TString DecodeEnumValue(TStringBuf value)
+{
+    auto decodedValue = DecodeEnumValueImpl<true>(value);
+    YT_VERIFY(decodedValue);
+    return *decodedValue;
+}
+
 TString EncodeEnumValue(TStringBuf value)
 {
     return CamelCaseToUnderscoreCase(value);