Copy cached responses in TObjectServiceCache

2 files changed, 19 insertions, 0 deletions

diff --git a/library/cpp/yt/memory/ref.cpp b/library/cpp/yt/memory/ref.cpp
index d48277f84c..31680b6c61 100644
--- a/library/cpp/yt/memory/ref.cpp
+++ b/library/cpp/yt/memory/ref.cpp
@@ -386,6 +386,21 @@ TString TSharedRefArray::ToString() const
     return result;
 }
 
+TSharedRefArray TSharedRefArray::MakeCopy(
+    const TSharedRefArray& array,
+    TRefCountedTypeCookie tagCookie)
+{
+    TSharedRefArrayBuilder builder(
+        array.Size(),
+        array.ByteSize(),
+        tagCookie);
+    for (const auto& part : array) {
+        auto partCopy = builder.AllocateAndAdd(part.Size());
+        ::memcpy(partCopy.Begin(), part.Begin(), part.Size());
+    }
+    return builder.Finish();
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 TSharedRefArrayBuilder::TSharedRefArrayBuilder(
diff --git a/library/cpp/yt/memory/ref.h b/library/cpp/yt/memory/ref.h
index 78b5cf58cb..a47188180c 100644
--- a/library/cpp/yt/memory/ref.h
+++ b/library/cpp/yt/memory/ref.h
@@ -302,6 +302,10 @@ public:
     std::vector<TSharedRef> ToVector() const;
     TString ToString() const;
 
+    //! Creates a copy of a given TSharedRefArray.
+    //! The memory is marked with a given tag.
+    static TSharedRefArray MakeCopy(const TSharedRefArray& array, TRefCountedTypeCookie tagCookie);
+
 private:
     friend class TSharedRefArrayBuilder;