mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-04 13:32:13 +00:00
53 lines
1.7 KiB
C++
53 lines
1.7 KiB
C++
#include <IO/ReadBufferFromFile.h>
|
|
#include <IO/WriteBufferFromFile.h>
|
|
#include <IO/copyData.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Common/TerminalSize.h>
|
|
#include "ICommand.h"
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class CommandRead final : public ICommand
|
|
{
|
|
public:
|
|
CommandRead()
|
|
{
|
|
command_name = "read";
|
|
description = "Read a file from `path-from` to `path-to`";
|
|
options_description.add_options()("path-from", po::value<String>(), "file from which we are reading (mandatory, positional)")(
|
|
"path-to", po::value<String>(), "file to which we are writing, defaults to `stdout`");
|
|
positional_options_description.add("path-from", 1);
|
|
}
|
|
|
|
void executeImpl(const CommandLineOptions & options, DisksClient & client) override
|
|
{
|
|
auto disk = client.getCurrentDiskWithPath();
|
|
String path_from = disk.getRelativeFromRoot(getValueFromCommandLineOptionsThrow<String>(options, "path-from"));
|
|
std::optional<String> path_to = getValueFromCommandLineOptionsWithOptional<String>(options, "path-to");
|
|
|
|
auto in = disk.getDisk()->readFile(path_from);
|
|
std::unique_ptr<WriteBufferFromFileBase> out = {};
|
|
if (path_to.has_value())
|
|
{
|
|
String relative_path_to = disk.getRelativeFromRoot(path_to.value());
|
|
out = disk.getDisk()->writeFile(relative_path_to);
|
|
copyData(*in, *out);
|
|
}
|
|
else
|
|
{
|
|
out = std::make_unique<WriteBufferFromFileDescriptor>(STDOUT_FILENO);
|
|
copyData(*in, *out);
|
|
out->write('\n');
|
|
}
|
|
out->finalize();
|
|
}
|
|
};
|
|
|
|
CommandPtr makeCommandRead()
|
|
{
|
|
return std::make_shared<DB::CommandRead>();
|
|
}
|
|
|
|
}
|