cloud/filestore/apps/client/lib/factory.cpp - main - githubmirror/nbs

cloud/filestore/apps/client/lib/factory.cpp
120 строк4 KB
Jan Szumski
Introducing a DiagnoseFilesystem command (#6493)
17 июл 2026, 16:44
17 июл 2026, 16:44f15c4f0
#include "factory.h"

#include <util/generic/map.h>
#include <util/string/subst.h>

#include <functional>

namespace NCloud::NFileStore::NClient {

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

TCommandPtr NewAddClusterNodeCommand();
TCommandPtr NewCreateCommand();
TCommandPtr NewDescribeCommand();
TCommandPtr NewDestroyCommand();
TCommandPtr NewKickEndpointCommand();
TCommandPtr NewListClusterNodesCommand();
TCommandPtr NewListEndpointsCommand();
TCommandPtr NewListFileStoresCommand();
TCommandPtr NewLnCommand();
TCommandPtr NewLsCommand();
TCommandPtr NewMkDirCommand();
TCommandPtr NewMountCommand();
TCommandPtr NewMvCommand();
TCommandPtr NewReadCommand();
TCommandPtr NewRemoveClusterNodeCommand();
TCommandPtr NewResizeCommand();
TCommandPtr NewRmCommand();
TCommandPtr NewStartEndpointCommand();
TCommandPtr NewStopEndpointCommand();
TCommandPtr NewTouchCommand();
TCommandPtr NewWriteCommand();
TCommandPtr NewExecuteActionCommand();
TCommandPtr NewCreateSessionCommand();
TCommandPtr NewResetSessionCommand();
TCommandPtr NewDestroySessionCommand();
TCommandPtr NewReadLinkCommand();
TCommandPtr NewStatCommand();
TCommandPtr NewSetNodeAttrCommand();
TCommandPtr NewFindGarbageCommand();
TCommandPtr NewForcedCompactionCommand();
TCommandPtr NewFindCommand();
TCommandPtr NewDiffCommand();
TCommandPtr NewPingCommand();
TCommandPtr NewDiagnoseFilesystemCommand();

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

using TFactoryFunc = std::function<TCommandPtr()>;
using TFactoryMap = TMap<TString, TFactoryFunc>;

static const TMap<TString, TFactoryFunc> Commands = {
    { "addclusternode", NewAddClusterNodeCommand },
    { "create", NewCreateCommand },
    { "createsession", NewCreateSessionCommand },
    { "describe", NewDescribeCommand },
    { "destroy", NewDestroyCommand },
    { "destroysession", NewDestroySessionCommand },
    { "diff", NewDiffCommand },
    { "executeaction", NewExecuteActionCommand },
    { "find", NewFindCommand },
    { "findgarbage", NewFindGarbageCommand },
    { "forcedcompaction", NewForcedCompactionCommand },
    { "kickendpoint", NewKickEndpointCommand },
    { "listclusternodes", NewListClusterNodesCommand },
    { "listendpoints", NewListEndpointsCommand },
    { "listfilestores", NewListFileStoresCommand },
    { "ln", NewLnCommand },
    { "ls", NewLsCommand },
    { "mkdir", NewMkDirCommand },
    { "mount", NewMountCommand },
    { "mv", NewMvCommand },
    { "read", NewReadCommand },
    { "readlink", NewReadLinkCommand },
    { "removeclusternode", NewRemoveClusterNodeCommand },
    { "resetsession", NewResetSessionCommand },
    { "resize", NewResizeCommand },
    { "rm", NewRmCommand },
    { "setnodeattr", NewSetNodeAttrCommand },
    { "startendpoint", NewStartEndpointCommand },
    { "stat", NewStatCommand },
    { "stopendpoint", NewStopEndpointCommand },
    { "touch", NewTouchCommand },
    { "write", NewWriteCommand },
    { "ping", NewPingCommand },
    { "diagnosefilesystem", NewDiagnoseFilesystemCommand },
};

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

TString NormalizeCommand(TString name)
{
    name.to_lower();
    SubstGlobal(name, "-", TStringBuf{});
    SubstGlobal(name, "_", TStringBuf{});
    return name;
}

TCommandPtr GetCommand(const TString& name)
{
    if (const auto* func = Commands.FindPtr(name)) {
        return (*func)();
    }

    return nullptr;
}

TVector<TString> GetCommandNames()
{
    TVector<TString> names;
    names.reserve(Commands.size());

    for (const auto& kv: Commands) {
        names.push_back(kv.first);
    }

    return names;
}

}   // namespace NCloud::NFileStore::NClient