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

60 lines
1.6 KiB
C++

#pragma once
#include "ICommand.h"
namespace DB
{
namespace ErrorCodes
{
extern const int BAD_ARGUMENTS;
}
class CommandMove : public ICommand
{
public:
CommandMove()
{
command_name = "move";
command_option_description.emplace(createOptionsDescription("Help Message for move", getTerminalWidth()));
description = "Move file or directory from `from_path` to `to_path`\nPath should be in format './' or './path' or 'path'";
usage = "Usage: move [OPTION]... <FROM_PATH> <TO_PATH>";
}
void processOptions(
Poco::Util::LayeredConfiguration &,
po::variables_map &) const override{}
void executeImpl(
const DB::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 = config.getString("disk", "default");
String path_from = pos_arguments[0];
String path_to = pos_arguments[1];
DiskPtr disk = global_context->getDisk(disk_name);
String full_path_from = fullPathWithValidate(disk, path_from);
String full_path_to = fullPathWithValidate(disk, path_to);
if (disk->isFile(full_path_from))
disk->moveFile(full_path_from, full_path_to);
else
disk->moveDirectory(full_path_from, full_path_to);
}
};
}
std::unique_ptr <DB::ICommand> makeCommandMove()
{
return std::make_unique<DB::CommandMove>();
}