mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 03:22:14 +00:00
73 lines
1.8 KiB
C++
73 lines
1.8 KiB
C++
#pragma once
|
|
|
|
#include "Parser.h"
|
|
#include "Commands.h"
|
|
#include <Common/ZooKeeper/ZooKeeper.h>
|
|
#include <Client/LineReader.h>
|
|
#include <IO/ReadBufferFromPocoSocket.h>
|
|
#include <IO/WriteBufferFromPocoSocket.h>
|
|
#include <Parsers/ASTLiteral.h>
|
|
#include <Poco/Net/StreamSocket.h>
|
|
#include <Poco/Util/Application.h>
|
|
#include <filesystem>
|
|
|
|
|
|
namespace fs = std::filesystem;
|
|
|
|
namespace DB
|
|
{
|
|
|
|
static const NameSet four_letter_word_commands
|
|
{
|
|
"ruok", "mntr", "srvr", "stat", "srst", "conf",
|
|
"cons", "crst", "envi", "dirs", "isro", "wchs",
|
|
"wchc", "wchp", "dump", "csnp", "lgif", "rqld",
|
|
};
|
|
|
|
class KeeperClient: public Poco::Util::Application
|
|
{
|
|
public:
|
|
KeeperClient() = default;
|
|
|
|
void initialize(Poco::Util::Application & self) override;
|
|
|
|
int main(const std::vector<String> & args) override;
|
|
|
|
void defineOptions(Poco::Util::OptionSet & options) override;
|
|
|
|
fs::path getAbsolutePath(const String & relative) const;
|
|
|
|
void askConfirmation(const String & prompt, std::function<void()> && callback);
|
|
|
|
String executeFourLetterCommand(const String & command);
|
|
|
|
zkutil::ZooKeeperPtr zookeeper;
|
|
std::filesystem::path cwd = "/";
|
|
std::function<void()> confirmation_callback;
|
|
|
|
inline static std::map<String, Command> commands;
|
|
|
|
protected:
|
|
void runInteractive();
|
|
void runInteractiveReplxx();
|
|
void runInteractiveInputStream();
|
|
|
|
bool processQueryText(const String & text);
|
|
|
|
void loadCommands(std::vector<Command> && new_commands);
|
|
|
|
std::vector<String> getCompletions(const String & prefix) const;
|
|
|
|
String history_file;
|
|
LineReader::Suggest suggest;
|
|
|
|
zkutil::ZooKeeperArgs zk_args;
|
|
|
|
bool ask_confirmation = true;
|
|
bool waiting_confirmation = false;
|
|
|
|
std::vector<String> registered_commands_and_four_letter_words;
|
|
};
|
|
|
|
}
|