Support ToProto/FromProto for protobuf map

commit_hash:b1ed324e2b774c388cc76403e9970975d5561717

Author: ignat

yt/yt/core/misc/protobuf_helpers-inl.h

@@ -479,6 +479,17 @@ void ToProto(
     NYT::NDetail::ToProtoArrayImpl(serializedArray, originalArray);
 }
 
+template <class TKey, class TValue, class TSerializedKey, class TSerializedValue>
+void ToProto(
+    ::google::protobuf::Map<TSerializedKey, TSerializedValue>* serializedMap,
+    const THashMap<TKey, TValue>& originalMap)
+{
+    serializedMap->clear();
+    for (const auto& [key, value] : originalMap) {
+        serializedMap->insert(std::pair(ToProto<TSerializedKey>(key), ToProto<TSerializedValue>(value)));
+    }
+}
+
 template <class TOriginalArray, class TSerialized, class... TArgs>
 void FromProto(
     TOriginalArray* originalArray,
@@ -515,6 +526,19 @@ void CheckedHashSetFromProto(
 
 ////////////////////////////////////////////////////////////////////////////////
 
+template <class TKey, class TValue, class TSerializedKey, class TSerializedValue>
+void FromProto(
+    THashMap<TKey, TValue>* originalMap,
+    const ::google::protobuf::Map<TSerializedKey, TSerializedValue>& serializedMap)
+{
+    originalMap->clear();
+    for (const auto& [serializedKey, serializedValue] : serializedMap) {
+        EmplaceOrCrash(*originalMap, FromProto<TKey>(serializedKey), FromProto<TValue>(serializedValue));
+    }
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
 template <class TSerialized, NYT::NDetail::CToProtoOriginal TOriginal, class... TArgs>
 auto ToProto(const TOriginal& original, TArgs&&... args)
 {

yt/yt/core/misc/protobuf_helpers.h

@@ -123,6 +123,11 @@ void ToProto(
     ::google::protobuf::RepeatedField<TSerialized>* serializedArray,
     const TOriginalArray& originalArray);
 
+template <class TKey, class TValue, class TSerializedKey, class TSerializedValue>
+void ToProto(
+    ::google::protobuf::Map<TSerializedKey, TSerializedValue>* serializedMap,
+    const THashMap<TKey, TValue>& originalMap);
+
 template <class TOriginalArray, class TSerialized, class... TArgs>
 void FromProto(
     TOriginalArray* originalArray,
@@ -144,6 +149,11 @@ void CheckedHashSetFromProto(
     THashSet<TOriginal>* originalHashSet,
     const ::google::protobuf::RepeatedField<TSerialized>& serializedHashSet);
 
+template <class TKey, class TValue, class TSerializedKey, class TSerializedValue>
+void FromProto(
+    THashMap<TKey, TValue>* originalMap,
+    const ::google::protobuf::Map<TSerializedKey, TSerializedValue>& serializedMap);
+
 ////////////////////////////////////////////////////////////////////////////////
 
 template <class TSerialized, class T, class TTag>