cloud/filestore/apps/client/lib/command.h - main - githubmirror/nbs

cloud/filestore/apps/client/lib/command.h
263 строки6 KB
Andrei Strelkovskii
filestore: TFuture::ExtractValue() -> UnsafeExtractValue() (#6210)
14 июн 2026, 15:20
14 июн 2026, 15:20c686734
#pragma once

#include "public.h"

#include "bootstrap.h"

#include <cloud/filestore/libs/client/client.h>
#include <cloud/filestore/libs/client/config.h>
#include <cloud/filestore/libs/client/session.h>
#include <cloud/filestore/libs/service/context.h>
#include <cloud/filestore/libs/service/endpoint.h>
#include <cloud/filestore/libs/service/filestore.h>
#include <cloud/filestore/libs/service/request.h>

#include <cloud/filestore/public/api/protos/fs.pb.h>

#include <cloud/storage/core/libs/common/error.h>
#include <cloud/storage/core/libs/common/public.h>
#include <cloud/storage/core/libs/diagnostics/logging.h>
#include <cloud/storage/core/libs/diagnostics/monitoring.h>
#include <cloud/storage/core/libs/grpc/public.h>
#include <cloud/storage/core/libs/iam/iface/client.h>

#include <contrib/ydb/library/actors/util/should_continue.h>
#include <library/cpp/getopt/small/last_getopt.h>
#include <library/cpp/threading/future/future.h>

#include <util/generic/maybe.h>
#include <util/generic/string.h>
#include <util/system/condvar.h>
#include <util/system/mutex.h>

namespace NCloud::NFileStore::NClient {

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

class TCommand
{
protected:
    NLastGetopt::TOpts Opts;
    TMaybe<NLastGetopt::TOptsParseResultException> OptsParseResult;

    TString VerboseLevel;
    TLogSettings LogSettings;
    ILoggingServicePtr Logging;
    TLog Log;

    TString MonitoringAddress;
    ui32 MonitoringPort = 0;
    ui32 MonitoringThreads = 0;
    IMonitoringServicePtr Monitoring;
    ICertificateProviderPtr CertificateProvider;

    ITimerPtr Timer;
    ISchedulerPtr Scheduler;

    TString ServerAddress;
    ui32 ServerPort = 0;
    ui32 SecurePort = 0;
    TString ServerUnixSocketPath;
    bool SkipCertVerification = false;
    TString ConfigFile;

    TClientConfigPtr ClientConfig;

    TMutex WaitMutex;
    TCondVar WaitCondVar;

    bool JsonOutput = false;

    TProgramShouldContinue ProgramShouldContinue;

    std::shared_ptr<TClientFactories> ClientFactories;
    TString IamConfigFile;
    NCloud::NIamClient::IIamTokenClientPtr IamClient;

    TProtoMessagePrinter ProtoMessagePrinter;

public:
    TCommand();

    virtual ~TCommand() = default;

    int Run(int argc, char** argv);
    void Stop(int exitCode);

    void SetClientFactories(std::shared_ptr<TClientFactories> clientFactories);

    NLastGetopt::TOpts& GetOpts()
    {
        return Opts;
    }

    TLog& AccessLog()
    {
        return Log;
    }

protected:
    virtual void Init();
    void InitIamTokenClient();

    virtual void Start();
    virtual void Stop();

    virtual bool Execute() = 0;

    static TCallContextPtr PrepareCallContext()
    {
        return MakeIntrusive<TCallContext>(CreateRequestId());
    }

    template <typename T>
    T WaitFor(NThreading::TFuture<T> future, bool extract = true)
    {
        if (!future.HasValue()) {
            auto* ptr = reinterpret_cast<NThreading::TFuture<void>*>(&future);
            if (!WaitForI(*ptr)) {
                return TErrorResponse(E_REJECTED, "request cancelled");
            }
        }
        return extract ? UnsafeExtractValue(future) : future.GetValue();
    }

private:
    bool WaitForI(const NThreading::TFuture<void>& future);

    TString GetIamTokenFromClient();
};

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

class TFileStoreServiceCommand
    : public TCommand
{
protected:
    TString ClientId;
    TString FileSystemId;
    bool DisableMultiTabletForwarding = false;

    IFileStoreServicePtr Client;

public:
    TFileStoreServiceCommand() = default;

protected:
    void Init() override;

    void Start() override;
    void Stop() override;
};

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

class TFileStoreCommand
    : public TFileStoreServiceCommand
{
public:
    TFileStoreCommand(bool isClientIdRequired = false);

    void Start() override;
    void Stop() override;

protected:
    template <typename T>
    std::shared_ptr<T> CreateRequest()
    {
        auto request = std::make_shared<T>();
        request->SetFileSystemId(FileSystemId);
        request->MutableHeaders()->SetDisableMultiTabletForwarding(
            DisableMultiTabletForwarding);

        return request;
    }

    template <typename T>
    static void CheckResponse(const T& response)
    {
        if (HasError(response)) {
            STORAGE_THROW_SERVICE_ERROR(response.GetError());
        }
    }

    struct TPathEntry
    {
        NProto::TNodeAttr Node;
        TStringBuf Name;
    };

    NProto::TNodeAttr ResolveNode(
        ISession& session,
        ui64 parentNodeId,
        TString name,
        bool ignoreMissing);
    TVector<TPathEntry> ResolvePath(
        ISession& session,
        TStringBuf path,
        bool ignoreMissing);
    NProto::TListNodesResponse ListAll(
        ISession& session,
        const TString& fsId,
        ui64 parentId,
        bool disableMultiTabletForwarding,
        ui32 maxBytes);
    TString ReadLink(ISession& session, ui64 nodeId);

    class TSessionGuard final
    {
        TFileStoreCommand& FileStoreCmd;
        ISessionPtr Session;
        TLog& Log;

    public:
        TSessionGuard(TFileStoreCommand& fileStoreCmd, ISessionPtr session)
            : FileStoreCmd(fileStoreCmd)
            , Session(std::move(session))
            , Log(FileStoreCmd.AccessLog())
        {
        }

        ~TSessionGuard()
        {
            try {
                FileStoreCmd.DestroySession(*Session);
            } catch (...) {
                STORAGE_ERROR("~TSessionGuard: " << CurrentExceptionMessage());
            }
        }

        ISession& AccessSession()
        {
            return *Session;
        }
    };

    [[nodiscard]] TSessionGuard CreateSession();
    [[nodiscard]] TSessionGuard CreateCustomSession(
        TString fsId,
        TString clientId);

private:
    void DestroySession(ISession& session);
};

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

class TEndpointCommand
    : public TCommand
{
protected:
    IEndpointManagerPtr Client;

public:
    TEndpointCommand();

protected:
    void Init() override;

    void Start() override;
    void Stop() override;
};

}   // namespace NCloud::NFileStore::NClient