thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-19 16:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add set generator

Browse Source
This commit is contained in:
alesapin 2022-01-20 20:06:34 +03:00
parent 35b6b11a5d
commit eaa0e4e542
3 changed files with 53 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
utils/keeper-bench/Generator.cpp
View File

@ -136,6 +136,33 @@ ZooKeeperRequestPtr CreateRequestGenerator::generate()
}
void SetRequestGenerator::startup(Coordination::ZooKeeper & zookeeper)
{
removeRecursive(zookeeper, path_prefix);
auto promise = std::make_shared<std::promise<void>>();
auto future = promise->get_future();
auto create_callback = [promise] (const CreateResponse & response)
{
if (response.error != Coordination::Error::ZOK)
promise->set_exception(std::make_exception_ptr(zkutil::KeeperException(response.error)));
else
promise->set_value();
};
zookeeper.create(path_prefix, "", false, false, default_acls, create_callback);
future.get();
}
ZooKeeperRequestPtr SetRequestGenerator::generate()
{
auto request = std::make_shared<ZooKeeperSetRequest>();
request->path = path_prefix;
request->data = generateRandomData(data_size);
return request;
}
void GetRequestGenerator::startup(Coordination::ZooKeeper & zookeeper)
{
auto promise = std::make_shared<std::promise<void>>();
@ -242,7 +269,7 @@ std::unique_ptr<IGenerator> getGenerator(const std::string & name)
}
else if (name == "create_small_data")
{
return std::make_unique<CreateRequestGenerator>("/create_generator", 50, 32);
return std::make_unique<CreateRequestGenerator>("/create_generator", 5, 32);
}
else if (name == "create_medium_data")
{
@ -284,6 +311,11 @@ std::unique_ptr<IGenerator> getGenerator(const std::string & name)
{
return std::make_unique<ListRequestGenerator>("/list_generator", 100000, 5);
}
else if (name == "set_small_data")
{
return std::make_unique<SetRequestGenerator>("/set_generator", 5);
}
throw DB::Exception(DB::ErrorCodes::LOGICAL_ERROR, "Unknown generator {}", name);
}

19
utils/keeper-bench/Generator.h
View File

@ -104,4 +104,23 @@ private:
std::optional<uint64_t> paths_length;
};
class SetRequestGenerator final : public IGenerator
{
public:
explicit SetRequestGenerator(
std::string path_prefix_ = "/set_generator",
uint64_t data_size_ = 5)
: path_prefix(path_prefix_)
, data_size(data_size_)
{}
void startup(Coordination::ZooKeeper & zookeeper) override;
Coordination::ZooKeeperRequestPtr generate() override;
private:
std::string path_prefix;
uint64_t data_size;
};
std::unique_ptr<IGenerator> getGenerator(const std::string & name);

2
utils/keeper-bench/main.cpp
View File

@ -20,7 +20,7 @@ int main(int argc, char *argv[])
boost::program_options::options_description desc = createOptionsDescription("Allowed options", getTerminalWidth());
desc.add_options()
("help", "produce help message")
("generator", value<std::string>()->default_value("create_small_data"), "query to execute")
("generator", value<std::string>()->default_value("set_small_data"), "query to execute")
("concurrency,c", value<unsigned>()->default_value(1), "number of parallel queries")
("delay,d", value<double>()->default_value(1), "delay between intermediate reports in seconds (set 0 to disable reports)")
("iterations,i", value<size_t>()->default_value(0), "amount of queries to be executed")