ClickHouse/programs/disks/CommandWrite.cpp

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

81 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>
#include <IO/ReadBufferFromFile.h>
#include <IO/WriteBufferFromFile.h>
#include <IO/copyData.h>
namespace DB
{
namespace ErrorCodes
{
extern const int BAD_ARGUMENTS;
}
class CommandWrite final : public ICommand
{
public:
CommandWrite()
{
command_name = "write";
2022-06-16 17:38:33 +00:00
command_option_description.emplace(createOptionsDescription("Allowed options", getTerminalWidth()));
description = "Write File `from_path` or stdin to `to_path`";
2022-06-16 17:38:33 +00:00
usage = "write [OPTION]... <FROM_PATH> <TO_PATH>\nor\nstdin | write [OPTION]... <TO_PATH>\nPath should be in format './' or './path' or 'path'";
command_option_description->add_options()
("input", po::value<String>(), "set path to file to which we are write")
;
}
void processOptions(
Poco::Util::LayeredConfiguration & config,
po::variables_map & options) const override
{
if (options.count("input"))
config.setString("input", options["input"].as<String>());
}
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("Bad Arguments", DB::ErrorCodes::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);
2022-06-02 18:24:13 +00:00
String path_input = config.getString("input", "");
std::unique_ptr<ReadBufferFromFileBase> in;
if (path_input.empty())
{
in = std::make_unique<ReadBufferFromFileDescriptor>(STDIN_FILENO);
}
else
{
2022-12-08 17:43:54 +00:00
String relative_path_input = validatePathAndGetAsRelative(path_input);
in = disk->readFile(relative_path_input);
}
2022-12-08 17:43:54 +00:00
auto out = disk->writeFile(relative_path);
copyData(*in, *out);
out->finalize();
}
};
}
std::unique_ptr <DB::ICommand> makeCommandWrite()
{
return std::make_unique<DB::CommandWrite>();
}