ClickHouse/programs/disks/CommandCopy.cpp
2022-06-06 16:52:58 +03:00

71 lines
2.2 KiB
C++

#pragma once
#include "ICommand.h"
namespace DB
{
namespace ErrorCodes
{
extern const int BAD_ARGUMENTS;
}
class CommandCopy : public ICommand
{
public:
CommandCopy()
{
command_name = "copy";
command_option_description.emplace(createOptionsDescription("Help Message for copy", getTerminalWidth()));
description = "Recursively copy data containing at `from_path` to `to_path`\nPath should be in format './' or './path' or 'path'";
usage = "Usage: copy [OPTION]... <FROM_PATH> <TO_PATH>";
command_option_description->add_options()
("diskFrom", po::value<String>(), "set name for disk from which we do operations")
("diskTo", po::value<String>(), "set name for disk to which we do operations")
;
}
void processOptions(
Poco::Util::LayeredConfiguration & config,
po::variables_map & options) const override
{
if (options.count("diskFrom"))
config.setString("diskFrom", options["diskFrom"].as<String>());
if (options.count("diskTo"))
config.setString("diskTo", options["diskTo"].as<String>());
}
void executeImpl(
const ContextMutablePtr & global_context,
const Poco::Util::LayeredConfiguration & config) const override
{
if (pos_arguments.size() != 2)
{
printHelpMessage();
throw DB::Exception("Bad Arguments", DB::ErrorCodes::BAD_ARGUMENTS);
}
String disk_name_from = config.getString("diskFrom", config.getString("disk", "default"));
String disk_name_to = config.getString("diskTo", config.getString("disk", "default"));
String path_from = pos_arguments[0];
String path_to = pos_arguments[1];
DiskPtr disk_from = global_context->getDisk(disk_name_from);
DiskPtr disk_to = global_context->getDisk(disk_name_to);
String full_path_from = fullPathWithValidate(disk_from, path_from);
String full_path_to = fullPathWithValidate(disk_to, path_to);
disk_from->copy(full_path_from, disk_to, full_path_to);
}
};
}
std::unique_ptr <DB::ICommand> makeCommandCopy()
{
return std::make_unique<DB::CommandCopy>();
}