[#20] Optimize memory usage

Provide custom buffer and use ArrayPool

Signed-off-by: Pavel Gross <p.gross@yando.com>

src/FrostFS.SDK.ClientV2/Parameters/PrmObjectPut.cs

@@ -1,6 +1,7 @@
-using System.Collections.Specialized;
+using FrostFS.SDK.ModelsV2;
+using System.Collections.Specialized;
 using System.IO;
-using FrostFS.SDK.ModelsV2;
+
 namespace FrostFS.SDK.ClientV2.Parameters;
 
 public sealed class PrmObjectPut : IContext, ISessionToken
@@ -31,6 +32,11 @@ public sealed class PrmObjectPut : IContext, ISessionToken
     /// <value>Size of the buffer</value>
     public int BufferMaxSize { get; set; }
 
+    /// <summary>
+    /// Allows to define a buffer for chunks to manage by the memory allocation and releasing.
+    /// </summary>
+    public byte[]? CustomBuffer { get; set; }
+
     /// <summary>
     /// FrostFS request X-Headers
     /// </summary>

src/FrostFS.SDK.ClientV2/Services/ObjectServiceProvider.cs

@@ -307,7 +307,21 @@ internal class ObjectServiceProvider(ObjectService.ObjectServiceClient client, C
 
         chunkSize = (int)Math.Min(restBytes, (ulong)chunkSize);
 
-        var chunkBuffer = ArrayPool<byte>.Shared.Rent(chunkSize);
+        bool isRentBuffer = false;
+        byte[]? chunkBuffer = null;
+        
+        try
+        {
+            if (args.CustomBuffer != null)
+            {
+                chunkBuffer = args.CustomBuffer;
+            }
+            else
+            {
+                chunkBuffer = env.GetArrayPool(Constants.ObjectChunkSize).Rent(chunkSize);
+                isRentBuffer = true;
+            }
+
             var sentBytes = 0;
 
             // 0 means no limit from client, so server side cut is performed
@@ -347,6 +361,14 @@ internal class ObjectServiceProvider(ObjectService.ObjectServiceClient client, C
 
             return new PutObjectResult(ObjectId.FromHash(response.Body.ObjectId.Value.ToByteArray()), sentBytes);
         }
+        finally
+        {
+            if (isRentBuffer && chunkBuffer != null)
+            {
+                ArrayPool<byte>.Shared.Return(chunkBuffer);
+            }
+        }
+    }
 
     private async Task<ObjectStreamer> GetUploadStream(PrmObjectPut args, Context ctx)
     {

src/FrostFS.SDK.ClientV2/Tools/ClientEnvironment.cs

@@ -4,11 +4,14 @@ using Grpc.Net.Client;
 using System;
 using System.Security.Cryptography;
 using FrostFS.SDK.Cryptography;
+using System.Buffers;
 
 namespace FrostFS.SDK.ClientV2;
 
 public class ClientEnvironment(Client client, ECDsa key, OwnerId owner, GrpcChannel channel, ModelsV2.Version version) : IDisposable
 {
+    private ArrayPool<byte> _arrayPool;
+
     internal OwnerId Owner { get; } = owner;
     internal GrpcChannel Channel { get; private set; } = channel;
     internal ModelsV2.Version Version { get; } = version;
@@ -18,6 +21,16 @@ public class ClientEnvironment(Client client, ECDsa key, OwnerId owner, GrpcChan
 
     internal ClientKey Key { get; } = new ClientKey(key);
 
+    /// <summary>
+    /// Custom pool is used for predefined sizes of buffers like grpc chunk 
+    /// </summary>
+    internal ArrayPool<byte> GetArrayPool(int size)
+    {
+        _arrayPool ??= ArrayPool<byte>.Create(size, 256);
+        
+        return _arrayPool;
+    }
+
     public void Dispose()
     {
         Dispose(true);