mirror of
https://github.com/hanzoai/datastore.git
synced 2026-09-26 13:55:07 +00:00
Azat KhuzhinandClaude
8ea6465c1e
shell-like completion support in clickhouse keeper-client (handle completions of arguments in quotes, i.e. 'foo ba', handle escaped arguments, i.e. foo\ ba, make ls print quoted nodes, if they have whitespaces) Also remove on_complete_modify_callback hack, replace it with pure getCompletions implementation Co-Authored-By: Claude <[email protected]>
45 lines
1.0 KiB
C++
45 lines
1.0 KiB
C++
#pragma once
|
|
|
|
#include <Client/LineReader.h>
|
|
#include <Poco/Util/Application.h>
|
|
#include <Common/ZooKeeper/KeeperClientCLI/KeeperClient.h>
|
|
|
|
#include <iostream>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class KeeperClient: public Poco::Util::Application, public KeeperClientBase
|
|
{
|
|
public:
|
|
KeeperClient() : KeeperClientBase(std::cout, std::cerr) {}
|
|
|
|
void initialize(Poco::Util::Application & self) override;
|
|
|
|
int main(const std::vector<String> & args) override;
|
|
|
|
void defineOptions(Poco::Util::OptionSet & options) override;
|
|
|
|
String executeFourLetterCommand(const String & command) final;
|
|
|
|
protected:
|
|
void runInteractive();
|
|
void runInteractiveReplxx();
|
|
void runInteractiveInputStream();
|
|
|
|
void connectToKeeper();
|
|
|
|
bool processQueryText(const String & text, bool is_interactive);
|
|
|
|
std::vector<String> getCompletions(const String & prefix) const;
|
|
|
|
zkutil::ZooKeeperArgs zk_args;
|
|
|
|
String history_file;
|
|
UInt32 history_max_entries = 0; /// Maximum number of entries in the history file.
|
|
|
|
LineReader::Suggest suggest;
|
|
};
|
|
|
|
}
|