mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-13 09:52:38 +00:00
55 lines
1.6 KiB
C++
55 lines
1.6 KiB
C++
#include <Interpreters/Context.h>
|
|
#include "Common/Exception.h"
|
|
#include "ICommand.h"
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int BAD_ARGUMENTS;
|
|
}
|
|
|
|
|
|
class CommandRemove final : public ICommand
|
|
{
|
|
public:
|
|
CommandRemove()
|
|
{
|
|
command_name = "remove";
|
|
description = "Remove file or directory. Throws exception if file doesn't exists";
|
|
options_description.add_options()("path", po::value<String>(), "path that is going to be deleted (mandatory, positional)")(
|
|
"recursive,r", "recursively removes the directory (required to remove a directory)");
|
|
positional_options_description.add("path", 1);
|
|
}
|
|
|
|
void executeImpl(const CommandLineOptions & options, DisksClient & client) override
|
|
{
|
|
auto disk = client.getCurrentDiskWithPath();
|
|
const String & path = disk.getRelativeFromRoot(getValueFromCommandLineOptionsThrow<String>(options, "path"));
|
|
bool recursive = options.count("recursive");
|
|
if (disk.getDisk()->existsDirectory(path))
|
|
{
|
|
if (!recursive)
|
|
{
|
|
throw Exception(ErrorCodes::BAD_ARGUMENTS, "cannot remove '{}': Is a directory", path);
|
|
}
|
|
|
|
disk.getDisk()->removeRecursive(path);
|
|
}
|
|
else if (disk.getDisk()->existsFile(path))
|
|
{
|
|
disk.getDisk()->removeFileIfExists(path);
|
|
}
|
|
else
|
|
throw Exception(ErrorCodes::BAD_ARGUMENTS, "Path {} on disk {} doesn't exist", path, disk.getDisk()->getName());
|
|
}
|
|
};
|
|
|
|
CommandPtr makeCommandRemove()
|
|
{
|
|
return std::make_shared<DB::CommandRemove>();
|
|
}
|
|
|
|
}
|