mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
69 lines
1.6 KiB
C++
69 lines
1.6 KiB
C++
#include "ICommand.h"
|
|
#include <Interpreters/Context.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int BAD_ARGUMENTS;
|
|
}
|
|
|
|
class CommandListDisks final : public ICommand
|
|
{
|
|
public:
|
|
CommandListDisks()
|
|
{
|
|
command_name = "list-disks";
|
|
description = "List disks names";
|
|
usage = "list-disks [OPTION]";
|
|
}
|
|
|
|
void processOptions(
|
|
Poco::Util::LayeredConfiguration &,
|
|
po::variables_map &) const override
|
|
{}
|
|
|
|
void execute(
|
|
const std::vector<String> & command_arguments,
|
|
std::shared_ptr<DiskSelector> &,
|
|
Poco::Util::LayeredConfiguration & config) override
|
|
{
|
|
if (!command_arguments.empty())
|
|
{
|
|
printHelpMessage();
|
|
throw DB::Exception(DB::ErrorCodes::BAD_ARGUMENTS, "Bad Arguments");
|
|
}
|
|
|
|
constexpr auto config_prefix = "storage_configuration.disks";
|
|
constexpr auto default_disk_name = "default";
|
|
|
|
Poco::Util::AbstractConfiguration::Keys keys;
|
|
config.keys(config_prefix, keys);
|
|
|
|
bool has_default_disk = false;
|
|
|
|
/// For the output to be ordered
|
|
std::set<String> disks;
|
|
|
|
for (const auto & disk_name : keys)
|
|
{
|
|
if (disk_name == default_disk_name)
|
|
has_default_disk = true;
|
|
disks.insert(disk_name);
|
|
}
|
|
|
|
if (!has_default_disk)
|
|
disks.insert(default_disk_name);
|
|
|
|
for (const auto & disk : disks)
|
|
std::cout << disk << '\n';
|
|
}
|
|
};
|
|
}
|
|
|
|
std::unique_ptr <DB::ICommand> makeCommandListDisks()
|
|
{
|
|
return std::make_unique<DB::CommandListDisks>();
|
|
}
|