Moved commit "Add ydb cli replace and fetch storage config commands" from ydb repo

Author: mregrock

include/ydb-cpp-sdk/client/bsconfig/storage_config.h

@@ -0,0 +1,53 @@
+#pragma once
+
+#include <ydb-cpp-sdk/client/common_client/settings.h>
+#include <ydb-cpp-sdk/client/driver/driver.h>
+#include <ydb-cpp-sdk/client/types/status/status.h>
+#include <ydb-cpp-sdk/client/types/request_settings.h>
+#include <ydb-cpp-sdk/client/types/ydb.h>
+
+#include <memory>
+#include <string>
+
+namespace NYdb::inline V3::NStorageConfig {
+
+struct TFetchStorageConfigResult : public TStatus {
+    TFetchStorageConfigResult(
+            TStatus&& status,
+            std::string&& config)
+        : TStatus(std::move(status))
+        , Config_(std::move(config))
+    {}
+
+    const std::string& GetConfig() const {
+        return Config_;
+    }
+
+private:
+    std::string Config_;
+};
+
+using TAsyncFetchStorageConfigResult = NThreading::TFuture<TFetchStorageConfigResult>;
+
+struct TStorageConfigSettings : public NYdb::TOperationRequestSettings<TStorageConfigSettings> {};
+
+class TStorageConfigClient {
+public:
+
+    explicit TStorageConfigClient(const TDriver& driver, const TCommonClientSettings& settings = {});
+
+    ~TStorageConfigClient();
+
+    // Replace config
+    TAsyncStatus ReplaceStorageConfig(const std::string& config);
+
+    // Fetch current cluster storage config
+    TAsyncFetchStorageConfigResult FetchStorageConfig(const TStorageConfigSettings& settings = {});
+
+private:
+    class TImpl;
+
+    std::unique_ptr<TImpl> Impl_;
+};
+
+}

src/client/CMakeLists.txt

@@ -1,3 +1,4 @@
+add_subdirectory(bsconfig)
 add_subdirectory(draft)
 add_subdirectory(helpers)
 add_subdirectory(iam)

src/client/bsconfig/CMakeLists.txt

@@ -0,0 +1,20 @@
+_ydb_sdk_add_library(client-bsconfig)
+
+target_link_libraries(client-bsconfig
+  PUBLIC
+    client-ydb_driver
+    client-ydb_common_client
+    client-ydb_types-status
+    client-ydb_types
+  PRIVATE
+    api-grpc
+    impl-ydb_internal-make_request
+    client-ydb_common_client-impl
+)
+
+target_sources(client-bsconfig
+  PRIVATE
+    storage_config.cpp
+)
+
+_ydb_sdk_make_client_component(BSConfig client-bsconfig)

src/client/bsconfig/storage_config.cpp

@@ -0,0 +1,67 @@
+#include <ydb-cpp-sdk/client/bsconfig/storage_config.h>
+
+#include <src/client/common_client/impl/client.h>
+#include <src/client/impl/ydb_internal/make_request/make.h>
+
+#include <src/api/grpc/ydb_bsconfig_v1.grpc.pb.h>
+
+namespace NYdb::inline V3::NStorageConfig {
+
+class TStorageConfigClient::TImpl : public TClientImplCommon<TStorageConfigClient::TImpl> {
+public:
+    TImpl(std::shared_ptr<TGRpcConnectionsImpl> connections, const TCommonClientSettings& settings)
+        : TClientImplCommon(std::move(connections), settings)
+    {
+    }
+
+    TAsyncStatus ReplaceStorageConfig(const std::string& config) {
+        auto request = MakeRequest<Ydb::BSConfig::ReplaceStorageConfigRequest>();
+        request.set_yaml_config(config);
+
+        return RunSimple<Ydb::BSConfig::V1::BSConfigService, Ydb::BSConfig::ReplaceStorageConfigRequest, Ydb::BSConfig::ReplaceStorageConfigResponse>(
+            std::move(request),
+            &Ydb::BSConfig::V1::BSConfigService::Stub::AsyncReplaceStorageConfig);
+    }
+
+    TAsyncFetchStorageConfigResult FetchStorageConfig(const TStorageConfigSettings& settings = {}) {
+        auto request = MakeOperationRequest<Ydb::BSConfig::FetchStorageConfigRequest>(settings);
+        auto promise = NThreading::NewPromise<TFetchStorageConfigResult>();
+
+        auto extractor = [promise] (google::protobuf::Any* any, TPlainStatus status) mutable {
+                NYdb::TStringType config;
+                if (Ydb::BSConfig::FetchStorageConfigResult result; any && any->UnpackTo(&result)) {
+                    config = result.yaml_config();
+                }
+
+                TFetchStorageConfigResult val(TStatus(std::move(status)), std::string{std::move(config)});
+                promise.SetValue(std::move(val));
+            };
+
+        Connections_->RunDeferred<Ydb::BSConfig::V1::BSConfigService, Ydb::BSConfig::FetchStorageConfigRequest, Ydb::BSConfig::FetchStorageConfigResponse>(
+            std::move(request),
+            extractor,
+            &Ydb::BSConfig::V1::BSConfigService::Stub::AsyncFetchStorageConfig,
+            DbDriverState_,
+            INITIAL_DEFERRED_CALL_DELAY,
+            TRpcRequestSettings::Make(settings));
+        return promise.GetFuture();
+    }
+
+};
+
+TStorageConfigClient::TStorageConfigClient(const TDriver& driver, const TCommonClientSettings& settings)
+    : Impl_(new TStorageConfigClient::TImpl(CreateInternalInterface(driver), settings))
+{}
+
+TStorageConfigClient::~TStorageConfigClient() = default;
+
+TAsyncStatus TStorageConfigClient::ReplaceStorageConfig(const std::string& config) {
+    return Impl_->ReplaceStorageConfig(config);
+}
+
+TAsyncFetchStorageConfigResult TStorageConfigClient::FetchStorageConfig(const TStorageConfigSettings& settings) {
+    return Impl_->FetchStorageConfig(settings);
+}
+
+
+}