ClickHouse/programs/disks/CommandList.cpp

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

99 lines
2.6 KiB
C++
Raw Normal View History

#include "ICommand.h"
2022-07-20 20:30:16 +00:00
#include <Interpreters/Context.h>
2022-12-08 17:53:05 +00:00
#include <Common/TerminalSize.h>
namespace DB
{
namespace ErrorCodes
{
extern const int BAD_ARGUMENTS;
}
class CommandList final : public ICommand
{
public:
CommandList()
{
command_name = "list";
2022-06-23 21:55:12 +00:00
command_option_description.emplace(createOptionsDescription("Allowed options", getTerminalWidth()));
2023-08-29 12:25:04 +00:00
description = "List files at path[s]";
2022-06-16 17:38:33 +00:00
usage = "list [OPTION]... <PATH>...";
command_option_description->add_options()
2023-08-29 12:25:04 +00:00
("recursive", "recursively list all directories");
}
void processOptions(
Poco::Util::LayeredConfiguration & config,
po::variables_map & options) const override
{
if (options.count("recursive"))
config.setBool("recursive", true);
}
2022-06-16 17:38:33 +00:00
void execute(
const std::vector<String> & command_arguments,
DB::ContextMutablePtr & global_context,
Poco::Util::LayeredConfiguration & config) override
{
2022-06-16 17:38:33 +00:00
if (command_arguments.size() != 1)
{
printHelpMessage();
throw DB::Exception(DB::ErrorCodes::BAD_ARGUMENTS, "Bad Arguments");
}
String disk_name = config.getString("disk", "default");
const String & path = command_arguments[0];
DiskPtr disk = global_context->getDisk(disk_name);
2022-12-08 17:43:54 +00:00
String relative_path = validatePathAndGetAsRelative(path);
bool recursive = config.getBool("recursive", false);
if (recursive)
2022-12-08 17:43:54 +00:00
listRecursive(disk, relative_path);
else
2022-12-08 17:43:54 +00:00
list(disk, relative_path);
}
private:
2022-12-08 17:43:54 +00:00
static void list(const DiskPtr & disk, const std::string & relative_path)
{
std::vector<String> file_names;
2022-12-08 17:43:54 +00:00
disk->listFiles(relative_path, file_names);
for (const auto & file_name : file_names)
std::cout << file_name << '\n';
}
2022-12-08 17:43:54 +00:00
static void listRecursive(const DiskPtr & disk, const std::string & relative_path)
{
std::vector<String> file_names;
2022-12-08 17:43:54 +00:00
disk->listFiles(relative_path, file_names);
2022-12-08 17:43:54 +00:00
std::cout << relative_path << ":\n";
if (!file_names.empty())
{
for (const auto & file_name : file_names)
std::cout << file_name << '\n';
std::cout << "\n";
}
for (const auto & file_name : file_names)
{
2022-12-08 17:43:54 +00:00
auto path = relative_path + "/" + file_name;
if (disk->isDirectory(path))
listRecursive(disk, path);
}
}
};
}
std::unique_ptr <DB::ICommand> makeCommandList()
{
return std::make_unique<DB::CommandList>();
}