mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 02:12:21 +00:00
224 lines
7.4 KiB
C++
224 lines
7.4 KiB
C++
#include <IO/ReadBufferFromString.h>
|
|
#include <IO/ReadHelpers.h>
|
|
#include <Poco/ConsoleChannel.h>
|
|
#include <Common/ZooKeeper/KeeperException.h>
|
|
#include <Common/ZooKeeper/ZooKeeper.h>
|
|
#include <base/LineReader.h>
|
|
#include <Common/logger_useful.h>
|
|
|
|
#include <iostream>
|
|
#include <sstream>
|
|
|
|
|
|
void printStat(const Coordination::Stat & s)
|
|
{
|
|
std::cout << "Stat:\n";
|
|
std::cout << " czxid: " << s.czxid << '\n';
|
|
std::cout << " mzxid: " << s.mzxid << '\n';
|
|
std::cout << " ctime: " << s.ctime << '\n';
|
|
std::cout << " mtime: " << s.mtime << '\n';
|
|
std::cout << " version: " << s.version << '\n';
|
|
std::cout << " cversion: " << s.cversion << '\n';
|
|
std::cout << " aversion: " << s.aversion << '\n';
|
|
std::cout << " ephemeralOwner: " << s.ephemeralOwner << '\n';
|
|
std::cout << " dataLength: " << s.dataLength << '\n';
|
|
std::cout << " numChildren: " << s.numChildren << '\n';
|
|
std::cout << " pzxid: " << s.pzxid << std::endl;
|
|
}
|
|
|
|
void waitForWatch(const zkutil::EventPtr & event)
|
|
{
|
|
std::cout << "waiting for watch" << std::endl;
|
|
event->wait();
|
|
std::cout << "watch event was signalled" << std::endl;
|
|
}
|
|
|
|
|
|
void readUntilSpace(std::string & s, DB::ReadBuffer & buf)
|
|
{
|
|
s = "";
|
|
while (!buf.eof())
|
|
{
|
|
if (isspace(*buf.position()))
|
|
return;
|
|
s.push_back(*buf.position());
|
|
++buf.position();
|
|
}
|
|
}
|
|
|
|
void readMaybeQuoted(std::string & s, DB::ReadBuffer & buf)
|
|
{
|
|
if (!buf.eof() && *buf.position() == '\'')
|
|
DB::readQuotedString(s, buf);
|
|
else
|
|
readUntilSpace(s, buf);
|
|
}
|
|
|
|
|
|
int main(int argc, char ** argv)
|
|
{
|
|
try
|
|
{
|
|
if (argc != 2)
|
|
{
|
|
std::cerr << "usage: " << argv[0] << " hosts" << std::endl;
|
|
return 2;
|
|
}
|
|
|
|
Poco::AutoPtr<Poco::ConsoleChannel> channel = new Poco::ConsoleChannel(std::cerr);
|
|
Poco::Logger::root().setChannel(channel);
|
|
Poco::Logger::root().setLevel("trace");
|
|
|
|
zkutil::ZooKeeper zk{zkutil::ZooKeeperArgs(argv[1])};
|
|
LineReader lr({}, false, {"\\"}, {});
|
|
|
|
do
|
|
{
|
|
const auto & line = lr.readLine(":3 ", ":3 ");
|
|
if (line.empty())
|
|
break;
|
|
|
|
try
|
|
{
|
|
std::stringstream ss(line); // STYLE_CHECK_ALLOW_STD_STRING_STREAM
|
|
|
|
std::string cmd;
|
|
ss >> cmd;
|
|
|
|
if (cmd == "q" || cmd == "quit" || cmd == "exit" || cmd == ":q")
|
|
break;
|
|
|
|
std::string path;
|
|
ss >> path;
|
|
if (cmd == "ls")
|
|
{
|
|
std::string w;
|
|
ss >> w;
|
|
bool watch = w == "w";
|
|
zkutil::EventPtr event = watch ? std::make_shared<Poco::Event>() : nullptr;
|
|
std::vector<std::string> v = zk.getChildren(path, nullptr, event);
|
|
for (const auto & child : v)
|
|
std::cout << child << std::endl;
|
|
if (watch)
|
|
waitForWatch(event);
|
|
}
|
|
else if (cmd == "create")
|
|
{
|
|
DB::ReadBufferFromString in(line);
|
|
|
|
std::string path_ignored;
|
|
std::string data;
|
|
std::string mode;
|
|
|
|
DB::assertString("create", in);
|
|
DB::skipWhitespaceIfAny(in);
|
|
readMaybeQuoted(path_ignored, in);
|
|
DB::skipWhitespaceIfAny(in);
|
|
readMaybeQuoted(data, in);
|
|
DB::skipWhitespaceIfAny(in);
|
|
readUntilSpace(mode, in);
|
|
|
|
int32_t m;
|
|
if (mode == "p")
|
|
m = zkutil::CreateMode::Persistent;
|
|
else if (mode == "ps")
|
|
m = zkutil::CreateMode::PersistentSequential;
|
|
else if (mode == "e")
|
|
m = zkutil::CreateMode::Ephemeral;
|
|
else if (mode == "es")
|
|
m = zkutil::CreateMode::EphemeralSequential;
|
|
else
|
|
{
|
|
std::cout << "Bad create mode" << std::endl;
|
|
continue;
|
|
}
|
|
std::cout << zk.create(path, data, m) << std::endl;
|
|
}
|
|
else if (cmd == "remove")
|
|
{
|
|
zk.remove(path);
|
|
}
|
|
else if (cmd == "rmr")
|
|
{
|
|
zk.removeRecursive(path);
|
|
}
|
|
else if (cmd == "exists")
|
|
{
|
|
std::string w;
|
|
ss >> w;
|
|
bool watch = w == "w";
|
|
zkutil::EventPtr event = watch ? std::make_shared<Poco::Event>() : nullptr;
|
|
Coordination::Stat stat;
|
|
bool e = zk.exists(path, &stat, event);
|
|
if (e)
|
|
printStat(stat);
|
|
else
|
|
std::cout << path << " does not exist" << std::endl;
|
|
if (watch)
|
|
waitForWatch(event);
|
|
}
|
|
else if (cmd == "get")
|
|
{
|
|
std::string w;
|
|
ss >> w;
|
|
bool watch = w == "w";
|
|
zkutil::EventPtr event = watch ? std::make_shared<Poco::Event>() : nullptr;
|
|
Coordination::Stat stat;
|
|
std::string data = zk.get(path, &stat, event);
|
|
std::cout << "Data: " << data << std::endl;
|
|
printStat(stat);
|
|
if (watch)
|
|
waitForWatch(event);
|
|
}
|
|
else if (cmd == "set")
|
|
{
|
|
DB::ReadBufferFromString in(line);
|
|
|
|
std::string data;
|
|
int version = -1;
|
|
|
|
DB::assertString("set", in);
|
|
DB::skipWhitespaceIfAny(in);
|
|
DB::assertString(path, in);
|
|
DB::skipWhitespaceIfAny(in);
|
|
readMaybeQuoted(data, in);
|
|
DB::skipWhitespaceIfAny(in);
|
|
|
|
if (!in.eof())
|
|
DB::readText(version, in);
|
|
|
|
Coordination::Stat stat;
|
|
zk.set(path, data, version, &stat);
|
|
printStat(stat);
|
|
}
|
|
else if (!cmd.empty())
|
|
{
|
|
std::cout << "commands:\n";
|
|
std::cout << " q\n";
|
|
std::cout << " ls path [w]\n";
|
|
std::cout << " create path data (p|ps|e|es)\n";
|
|
std::cout << " remove path\n";
|
|
std::cout << " rmr path\n";
|
|
std::cout << " exists path [w]\n";
|
|
std::cout << " get path [w]\n";
|
|
std::cout << " set path data [version]" << std::endl;
|
|
continue;
|
|
}
|
|
|
|
}
|
|
catch (const Coordination::Exception & e)
|
|
{
|
|
std::cerr << "KeeperException: " << e.displayText() << std::endl;
|
|
}
|
|
}
|
|
while (true);
|
|
}
|
|
catch (const Coordination::Exception & e)
|
|
{
|
|
std::cerr << "KeeperException: " << e.displayText() << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|