mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 08:32:02 +00:00
Add set generator
This commit is contained in:
parent
35b6b11a5d
commit
eaa0e4e542
@ -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);
|
||||
}
|
||||
|
@ -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);
|
||||
|
@ -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")
|
||||
|
Loading…
Reference in New Issue
Block a user