Fix multiput's multiple counting buffer size (#10522)

ydb-platform · Nov 1, 2024 · 93c21b6 · 93c21b6
1 parent 509064d
commit 93c21b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ydb/core/blobstorage/dsproxy/dsproxy.h b/ydb/core/blobstorage/dsproxy/dsproxy.h
@@ -341,6 +341,7 @@ class TBlobStorageGroupRequestActor : public TActor<TBlobStorageGroupRequestActo
     TActorId ProxyActorId;
     std::shared_ptr<TEvBlobStorage::TExecutionRelay> ExecutionRelay;
     bool ExecutionRelayUsed = false;
+    bool FirstResponse = true;
 };
 
 void Encrypt(char *destination, const char *source, size_t shift, size_t sizeBytes, const TLogoBlobID &id,

diff --git a/ydb/core/blobstorage/dsproxy/dsproxy_request.cpp b/ydb/core/blobstorage/dsproxy/dsproxy_request.cpp
@@ -826,7 +826,7 @@ namespace NKikimr {
 
         // ensure that we are dying for the first time
         Y_ABORT_UNLESS(!Dead);
-        if (RequestHandleClass && PoolCounters) {
+        if (RequestHandleClass && PoolCounters && FirstResponse) {
             PoolCounters->GetItem(*RequestHandleClass, RequestBytes).Register(
                 RequestBytes, GeneratedSubrequests, GeneratedSubrequestBytes, Timer.Passed());
         }
@@ -857,6 +857,7 @@ namespace NKikimr {
 
         // send the reply to original request sender
         Send(source, ev.release(), 0, cookie);
+        FirstResponse = false;
     };
 
     void TBlobStorageGroupRequestActor::SendResponse(std::unique_ptr<IEventBase>&& ev, TBlobStorageGroupProxyTimeStats *timeStats) {