[enum_parser] generate ToStringBuf specialization

IGNIETFERRO-1350

ref:bb9bb1755b6abfc5a27639daf049b8a7cd748fe4

2 files changed, 40 insertions, 10 deletions

diff --git a/tools/enum_parser/enum_serialization_runtime/dispatch_methods.h b/tools/enum_parser/enum_serialization_runtime/dispatch_methods.h
index db752ae7108..fa2c8aa833b 100644
--- a/tools/enum_parser/enum_serialization_runtime/dispatch_methods.h
+++ b/tools/enum_parser/enum_serialization_runtime/dispatch_methods.h
@@ -27,6 +27,21 @@ namespace NEnumSerializationRuntime {
     }
 
     template <class TNameBufs, typename EEnum>
+    inline TStringBuf DispatchToStringBufFn(EEnum n) {
+        constexpr auto order = TNameBufs::NamesOrder;
+        if constexpr (order >= ::NEnumSerializationRuntime::ESortOrder::DirectMapping) {
+            return TNameBufs::ToStringBufDirect(n, TNameBufs::EnumInitializationData);
+        } else if constexpr (std::size(TNameBufs::EnumInitializationData.NamesInitializer) <= LINEAR_SEARCH_KEYS_SIZE_THRESHOLD) {
+            return TNameBufs::ToStringBufFullScan(n, TNameBufs::EnumInitializationData);
+        } else if constexpr (order >= ::NEnumSerializationRuntime::ESortOrder::Ascending) {
+            return TNameBufs::ToStringBufSorted(n, TNameBufs::EnumInitializationData);
+        } else {
+            const TNameBufs& names = TNameBufs::Instance();
+            return names.ToStringBuf(n);
+        }
+    }
+
+    template <class TNameBufs, typename EEnum>
     inline EEnum DispatchFromStringImplFn(const char* data, size_t len) {
         const TStringBuf name{data, len};
         constexpr auto order = TNameBufs::ValuesOrder;
diff --git a/tools/enum_parser/enum_serialization_runtime/enum_runtime.h b/tools/enum_parser/enum_serialization_runtime/enum_runtime.h
index cac142608da..7c85ed3934a 100644
--- a/tools/enum_parser/enum_serialization_runtime/enum_runtime.h
+++ b/tools/enum_parser/enum_serialization_runtime/enum_runtime.h
@@ -102,34 +102,51 @@ namespace NEnumSerializationRuntime {
     public:
         using TBase::TBase;
 
+        // ToString
+        // Return reference to singleton preallocated string
         const TString& ToString(const EEnum key) const {
             return TBase::ToString(TCast::CastToRepresentationType(key));
         }
 
-        bool FromString(const TStringBuf name, EEnum& ret) const {
-            return MapFindResult(TBase::TryFromString(name), ret);
+        // ToStringBuf
+        TStringBuf ToStringBuf(EEnum key) const {
+            return TBase::ToStringBuf(TCast::CastToRepresentationType(key));
         }
-
-        EEnum FromString(const TStringBuf name) const {
-            return TCast::CastFromRepresentationType(TBase::FromString(name));
+        static TStringBuf ToStringBufFullScan(const EEnum key, const TInitializationData& enumInitData) {
+            return TBase::ToStringBufFullScan(TCast::CastToRepresentationType(key), enumInitData);
+        }
+        static TStringBuf ToStringBufSorted(const EEnum key, const TInitializationData& enumInitData) {
+            return TBase::ToStringBufSorted(TCast::CastToRepresentationType(key), enumInitData);
+        }
+        static TStringBuf ToStringBufDirect(const EEnum key, const TInitializationData& enumInitData) {
+            return TBase::ToStringBufDirect(TCast::CastToRepresentationType(key), enumInitData);
         }
 
+        // TryFromString-like functons
+        // Return false for unknown enumeration names
+        bool FromString(const TStringBuf name, EEnum& ret) const {
+            return MapFindResult(TBase::TryFromString(name), ret);
+        }
         static bool TryFromStringFullScan(const TStringBuf name, EEnum& ret, const TInitializationData& enumInitData) {
             return MapFindResult(TBase::TryFromStringFullScan(name, enumInitData), ret);
         }
-
         static bool TryFromStringSorted(const TStringBuf name, EEnum& ret, const TInitializationData& enumInitData) {
             return MapFindResult(TBase::TryFromStringSorted(name, enumInitData), ret);
         }
 
+        // FromString
+        // Throw exception for unknown enumeration names
+        EEnum FromString(const TStringBuf name) const {
+            return TCast::CastFromRepresentationType(TBase::FromString(name));
+        }
         static EEnum FromStringFullScan(const TStringBuf name, const TInitializationData& enumInitData) {
             return TCast::CastFromRepresentationType(TBase::FromStringFullScan(name, enumInitData));
         }
-
         static EEnum FromStringSorted(const TStringBuf name, const TInitializationData& enumInitData) {
             return TCast::CastFromRepresentationType(TBase::FromStringSorted(name, enumInitData));
         }
 
+        // Inspection
         TMappedDictView<EEnum, TString> EnumNames() const noexcept {
             return {TBase::TypelessEnumNames()};
         }
@@ -138,18 +155,16 @@ namespace NEnumSerializationRuntime {
             return {TBase::TypelessEnumValues()};
         }
 
+        // Out
         void Out(IOutputStream* os, const EEnum key) const {
             TBase::Out(os, TCast::CastToRepresentationType(key));
         }
-
         static void OutFullScan(IOutputStream* os, const EEnum key, const TInitializationData& enumInitData) {
             TBase::OutFullScan(os, TCast::CastToRepresentationType(key), enumInitData);
         }
-
         static void OutSorted(IOutputStream* os, const EEnum key, const TInitializationData& enumInitData) {
             TBase::OutSorted(os, TCast::CastToRepresentationType(key), enumInitData);
         }
-
         static void OutDirect(IOutputStream* os, const EEnum key, const TInitializationData& enumInitData) {
             TBase::OutDirect(os, TCast::CastToRepresentationType(key), enumInitData);
         }