From 010bdce3aa67c5ed058a9660bb4c6cf8acb0f01d Mon Sep 17 00:00:00 2001
From: Stepanyuk Vladislav <stepanyukvladyslav@yandex.ru>
Date: Thu, 16 Jan 2025 17:41:12 +0700
Subject: [PATCH] issue-2725: refactor file and fix responce

---
 .../disk_registry_actor_acquire_release.cpp   | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/cloud/blockstore/libs/storage/disk_registry/disk_registry_actor_acquire_release.cpp b/cloud/blockstore/libs/storage/disk_registry/disk_registry_actor_acquire_release.cpp
index aae4536b8bb..6a7cb7467d3 100644
--- a/cloud/blockstore/libs/storage/disk_registry/disk_registry_actor_acquire_release.cpp
+++ b/cloud/blockstore/libs/storage/disk_registry/disk_registry_actor_acquire_release.cpp
@@ -15,16 +15,12 @@ using namespace NKikimr::NTabletFlatExecutor;
 
 namespace {
 
-
-TVector<NProto::TDeviceConfig> ExtractDevicesFromDiskInfo(TDiskInfo &diskInfo) {
+TVector<NProto::TDeviceConfig> ExtractDevicesFromDiskInfo(TDiskInfo& diskInfo)
+{
     TVector devices = std::move(diskInfo.Devices);
-
-    auto devicesToAdd = diskInfo.Migrations.size();
-    for (const auto& replica: diskInfo.Replicas) {
-        devicesToAdd += replica.size();
-    }
-
-    devices.reserve(devices.size() + devicesToAdd);
+    devices.reserve(
+        devices.size() * (diskInfo.Replicas.size() + 1) +
+        diskInfo.Migrations.size());
 
     for (auto& migration: diskInfo.Migrations) {
         devices.emplace_back(std::move(*migration.MutableTargetDevice()));
@@ -39,8 +35,7 @@ TVector<NProto::TDeviceConfig> ExtractDevicesFromDiskInfo(TDiskInfo &diskInfo) {
     return devices;
 }
 
-}  // namespace
-
+}   // namespace
 
 ////////////////////////////////////////////////////////////////////////////////
 
@@ -53,7 +48,7 @@ void TDiskRegistryActor::HandleAcquireDisk(
     auto replyWithError = [&](auto error)
     {
         auto response =
-            std::make_unique<TEvDiskRegistry::TEvReleaseDiskResponse>(
+            std::make_unique<TEvDiskRegistry::TEvAcquireDiskResponse>(
                 std::move(error));
         NCloud::Reply(ctx, *ev, std::move(response));
     };