[Filestore] add ListTabletBootInfoBackups request/response to TabletB…

…ootInfoBackup - later it will be used for booting tablets in emergency mode (#2913)
ydb-platform · Jan 23, 2025 · 4483d4d · 4483d4d
1 parent 80bb53f
commit 4483d4d
Show file tree

Hide file tree

Showing 4 changed files with 74 additions and 17 deletions.
diff --git a/cloud/storage/core/libs/hive_proxy/hive_proxy_events_private.h b/cloud/storage/core/libs/hive_proxy/hive_proxy_events_private.h
@@ -2,6 +2,8 @@
 
 #include "public.h"
 
+#include "tablet_boot_info.h"
+
 #include <cloud/storage/core/libs/kikimr/components.h>
 #include <cloud/storage/core/libs/kikimr/events.h>
 
@@ -95,6 +97,24 @@ struct TEvHiveProxyPrivate
         {}
     };
 
+    //
+    // ListTabletBootInfoBackups
+    //
+
+    struct TListTabletBootInfoBackupsRequest
+    {
+    };
+
+    struct TListTabletBootInfoBackupsResponse
+    {
+        TVector<TTabletBootInfo> TabletBootInfos;
+
+        explicit TListTabletBootInfoBackupsResponse(
+                TVector<TTabletBootInfo> tabletBootInfos)
+            : TabletBootInfos(std::move(tabletBootInfos))
+        {}
+    };
+
     //
     // Events declaration
     //
@@ -110,6 +130,8 @@ struct TEvHiveProxyPrivate
         EvReadTabletBootInfoBackupRequest,
         EvReadTabletBootInfoBackupResponse,
         EvUpdateTabletBootInfoBackupRequest,
+        EvListTabletBootInfoBackupsRequest,
+        EvListTabletBootInfoBackupsResponse,
 
         EvEnd
     };
@@ -128,6 +150,10 @@ struct TEvHiveProxyPrivate
         TReadTabletBootInfoBackupResponse, EvReadTabletBootInfoBackupResponse>;
     using TEvUpdateTabletBootInfoBackupRequest = TRequestEvent<
         TUpdateTabletBootInfoBackupRequest, EvUpdateTabletBootInfoBackupRequest>;
+    using TEvListTabletBootInfoBackupsRequest = TRequestEvent<
+        TListTabletBootInfoBackupsRequest, EvListTabletBootInfoBackupsRequest>;
+    using TEvListTabletBootInfoBackupsResponse = TRequestEvent<
+        TListTabletBootInfoBackupsResponse, EvListTabletBootInfoBackupsResponse>;
 };
 
 }   // namespace NCloud::NStorage
diff --git a/cloud/storage/core/libs/hive_proxy/tablet_boot_info.h b/cloud/storage/core/libs/hive_proxy/tablet_boot_info.h
@@ -0,0 +1,26 @@
+#pragma once
+
+#include "public.h"
+
+#include <cloud/storage/core/libs/kikimr/public.h>
+
+namespace NCloud::NStorage {
+
+////////////////////////////////////////////////////////////////////////////////
+
+struct TTabletBootInfo
+{
+    TTabletBootInfo() = default;
+
+    TTabletBootInfo(
+            NKikimr::TTabletStorageInfoPtr storageInfo,
+            ui64 suggestedGeneration)
+        : StorageInfo(std::move(storageInfo))
+        , SuggestedGeneration(suggestedGeneration)
+    {}
+
+    NKikimr::TTabletStorageInfoPtr StorageInfo;
+    ui64 SuggestedGeneration = 0;
+};
+
+}   // namespace NCloud::NStorage
diff --git a/cloud/storage/core/libs/hive_proxy/tablet_boot_info_backup.cpp b/cloud/storage/core/libs/hive_proxy/tablet_boot_info_backup.cpp
@@ -187,6 +187,23 @@ void TTabletBootInfoBackup::HandleBackupTabletBootInfos(
     NCloud::Reply(ctx, *ev, std::move(response));
 }
 
+void TTabletBootInfoBackup::HandleListTabletBootInfoBackups(
+    const TEvHiveProxyPrivate::TEvListTabletBootInfoBackupsRequest::TPtr& ev,
+    const TActorContext& ctx)
+{
+    TVector<TTabletBootInfo> infos;
+    for (const auto& [_, info]: BackupProto.GetData()) {
+        infos.emplace_back(
+            NKikimr::TabletStorageInfoFromProto(info.GetStorageInfo()),
+            info.GetSuggestedGeneration());
+    }
+
+    auto response =
+        std::make_unique<TEvHiveProxyPrivate::TEvListTabletBootInfoBackupsResponse>(
+            std::move(infos));
+    NCloud::Reply(ctx, *ev, std::move(response));
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 STFUNC(TTabletBootInfoBackup::StateWork)
@@ -196,6 +213,7 @@ STFUNC(TTabletBootInfoBackup::StateWork)
         HFunc(TEvHiveProxyPrivate::TEvReadTabletBootInfoBackupRequest, HandleReadTabletBootInfoBackup);
         HFunc(TEvHiveProxyPrivate::TEvUpdateTabletBootInfoBackupRequest, HandleUpdateTabletBootInfoBackup);
         HFunc(TEvHiveProxy::TEvBackupTabletBootInfosRequest, HandleBackupTabletBootInfos);
+        HFunc(TEvHiveProxyPrivate::TEvListTabletBootInfoBackupsRequest, HandleListTabletBootInfoBackups);
         default:
             HandleUnexpectedEvent(ev, LogComponent);
             break;

diff --git a/cloud/storage/core/libs/hive_proxy/tablet_boot_info_backup.h b/cloud/storage/core/libs/hive_proxy/tablet_boot_info_backup.h
@@ -21,23 +21,6 @@ namespace NCloud::NStorage {
 
 ////////////////////////////////////////////////////////////////////////////////
 
-struct TTabletBootInfo
-{
-    TTabletBootInfo() = default;
-
-    TTabletBootInfo(
-            NKikimr::TTabletStorageInfoPtr storageInfo,
-            ui64 suggestedGeneration)
-        : StorageInfo(std::move(storageInfo))
-        , SuggestedGeneration(suggestedGeneration)
-    {}
-
-    NKikimr::TTabletStorageInfoPtr StorageInfo;
-    ui64 SuggestedGeneration = 0;
-};
-
-////////////////////////////////////////////////////////////////////////////////
-
 class TTabletBootInfoBackup final
     : public NActors::TActorBootstrapped<TTabletBootInfoBackup>
 {
@@ -78,6 +61,10 @@ class TTabletBootInfoBackup final
     void HandleBackupTabletBootInfos(
         const TEvHiveProxy::TEvBackupTabletBootInfosRequest::TPtr& ev,
         const NActors::TActorContext& ctx);
+
+    void HandleListTabletBootInfoBackups(
+        const TEvHiveProxyPrivate::TEvListTabletBootInfoBackupsRequest::TPtr& ev,
+        const NActors::TActorContext& ctx);
 };
 
 }   // namespace NCloud::NStorage