ClickHouse/programs/disks/CommandCopy.cpp

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

69 lines
2.3 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 CommandCopy final : public ICommand
{
public:
CommandCopy()
{
command_name = "copy";
2022-06-16 17:38:33 +00:00
command_option_description.emplace(createOptionsDescription("Allowed options", getTerminalWidth()));
2023-08-29 12:25:04 +00:00
description = "Recursively copy data from `FROM_PATH` to `TO_PATH`";
2022-06-16 17:38:33 +00:00
usage = "copy [OPTION]... <FROM_PATH> <TO_PATH>";
command_option_description->add_options()
("disk-from", po::value<String>(), "disk from which we copy")
("disk-to", po::value<String>(), "disk to which we copy");
}
void processOptions(
Poco::Util::LayeredConfiguration & config,
po::variables_map & options) const override
{
2023-09-04 15:01:43 +00:00
if (options.count("disk-from"))
config.setString("disk-from", options["disk-from"].as<String>());
if (options.count("disk-to"))
config.setString("disk-to", options["disk-to"].as<String>());
}
2022-06-16 17:38:33 +00:00
void execute(
const std::vector<String> & command_arguments,
2023-11-30 03:09:55 +00:00
std::shared_ptr<DiskSelector> & disk_selector,
2022-06-16 17:38:33 +00:00
Poco::Util::LayeredConfiguration & config) override
{
2022-06-16 17:38:33 +00:00
if (command_arguments.size() != 2)
{
printHelpMessage();
throw DB::Exception(DB::ErrorCodes::BAD_ARGUMENTS, "Bad Arguments");
}
2023-09-04 15:01:43 +00:00
String disk_name_from = config.getString("disk-from", config.getString("disk", "default"));
String disk_name_to = config.getString("disk-to", config.getString("disk", "default"));
const String & path_from = command_arguments[0];
const String & path_to = command_arguments[1];
2023-11-30 03:09:55 +00:00
DiskPtr disk_from = disk_selector->get(disk_name_from);
DiskPtr disk_to = disk_selector->get(disk_name_to);
2022-12-08 17:43:54 +00:00
String relative_path_from = validatePathAndGetAsRelative(path_from);
String relative_path_to = validatePathAndGetAsRelative(path_to);
disk_from->copyDirectoryContent(relative_path_from, disk_to, relative_path_to, /* read_settings= */ {}, /* write_settings= */ {}, /* cancellation_hook= */ {});
}
};
}
std::unique_ptr <DB::ICommand> makeCommandCopy()
{
return std::make_unique<DB::CommandCopy>();
}