cloud/blockstore/libs/disk_agent/bootstrap.h - main - githubmirror/nbs

cloud/blockstore/libs/disk_agent/bootstrap.h
114 строк3 KB
neihar
[Storage] rename RDMA namespace to storage from blockstore (#5797)
22 апр 2026, 17:21
22 апр 2026, 17:211c11ead
#pragma once

#include "private.h"

#include <cloud/blockstore/libs/common/public.h>
#include <cloud/blockstore/libs/diagnostics/public.h>
#include <cloud/blockstore/libs/kikimr/public.h>
#include <cloud/blockstore/libs/local_nvme/public.h>
#include <cloud/blockstore/libs/nvme/public.h>
#include <cloud/blockstore/libs/rdma/config.h>
#include <cloud/blockstore/libs/service/public.h>
#include <cloud/blockstore/libs/service_local/public.h>
#include <cloud/blockstore/libs/spdk/iface/public.h>
#include <cloud/blockstore/libs/storage/disk_agent/public.h>

#include <cloud/storage/core/libs/diagnostics/stats_fetcher.h>
#include <cloud/storage/core/libs/http/simple_http_server.h>
#include <cloud/storage/core/libs/rdma/iface/public.h>

#include <contrib/ydb/core/driver_lib/run/factories.h>
#include <contrib/ydb/library/actors/util/should_continue.h>

#include <library/cpp/logger/log.h>

namespace NCloud::NBlockStore::NServer {

////////////////////////////////////////////////////////////////////////////////

struct TConfigInitializer;

////////////////////////////////////////////////////////////////////////////////

struct TSpdkParts
{
    NSpdk::ISpdkEnvPtr Env;
    std::function<void(TLog& log)> LogInitializer;
};

struct TServerModuleFactories
{
    std::function<TSpdkParts(NSpdk::TSpdkEnvConfigPtr config)> SpdkFactory;
    std::function<NCloud::NStorage::NRdma::IServerPtr(
        ILoggingServicePtr logging,
        IMonitoringServicePtr monitoring,
        NCloud::NStorage::NRdma::TServerConfigPtr config)> RdmaServerFactory;
};

////////////////////////////////////////////////////////////////////////////////

class TBootstrap
{
private:
    std::shared_ptr<NKikimr::TModuleFactories> ModuleFactories;
    std::shared_ptr<TServerModuleFactories> ServerModuleFactories;

    std::unique_ptr<TConfigInitializer> Configs;

    ILoggingServicePtr BootstrapLogging;
    TLog Log;

    ITimerPtr Timer;
    ISchedulerPtr Scheduler;
    ITaskQueuePtr BackgroundThreadPool;
    IActorSystemPtr ActorSystem;
    IAsyncLoggerPtr AsyncLogger;
    ILoggingServicePtr Logging;
    NCloud::NStorage::IStatsFetcherPtr StatsFetcher;
    IMonitoringServicePtr Monitoring;
    TVector<ITraceReaderPtr> TraceReaders;
    ITraceProcessorPtr TraceProcessor;
    IProfileLogPtr ProfileLog;
    IBlockDigestGeneratorPtr BlockDigestGenerator;
    IFileIOServiceProviderPtr FileIOServiceProvider;
    NSpdk::ISpdkEnvPtr Spdk;
    std::function<void(TLog& log)> SpdkLogInitializer;
    ICachingAllocatorPtr Allocator;
    IStorageProviderPtr LocalStorageProvider;
    NNvme::INvmeManagerPtr NvmeManager;
    NCloud::NStorage::NRdma::IServerPtr RdmaServer;
    ILocalNVMeServicePtr LocalNVMeService;

    TProgramShouldContinue ShouldContinue;
    TVector<TString> PostponedCriticalEvents;

    std::unique_ptr<NCloud::NStorage::TSimpleHttpServer> StubMonPageServer;
    bool Initialized = false;

public:
    TBootstrap(
        std::shared_ptr<NKikimr::TModuleFactories> moduleFactories,
        std::shared_ptr<TServerModuleFactories> serverModuleFactories);
    ~TBootstrap();

    void ParseOptions(int argc, char** argv);
    void Init();

    void Start();
    void Stop();

    TProgramShouldContinue& GetShouldContinue();

private:
    void InitLWTrace();
    void InitProfileLog();
    bool InitKikimrService();

    void InitHTTPServer();

    void InitRdmaServer(NRdma::TRdmaConfig& config);

    bool InitBackend();
};

}   // namespace NCloud::NBlockStore::NServer