#include <condition_variable>
#include "gflags/gflags.h"
#include "rocketmq/Message.h"
#include "rocketmq/Producer.h"
using namespace ROCKETMQ_NAMESPACE;
Semaphore(std::size_t permits) : permits_(permits) {
std::unique_lock<std::mutex> lk(mtx_);
cv_.wait(lk, [this]() { return permits_ > 0; });
std::unique_lock<std::mutex> lk(mtx_);
std::size_t permits_{0};
std::condition_variable cv_;
const std::string& alphaNumeric() {
static std::string alpha_numeric("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ");
std::string randomString(std::string::size_type len) {
std::mt19937 generator(rd());
std::string source(alphaNumeric());
std::string::size_type generated = 0;
while (generated < len) {
std::shuffle(source.begin(), source.end(), generator);
std::string::size_type delta = std::min({len - generated, source.length()});
result.append(source.substr(0, delta));
DEFINE_string(topic, "xxxx", "Topic to which messages are published");
DEFINE_string(access_point, "rocketmq-xxxx:8080", "Service access URL, provided by your service provider");
DEFINE_int32(message_body_size, 4096, "Message body size");
DEFINE_uint32(total, 256, "Number of sample messages to publish");
DEFINE_string(access_key, "xxxxx", "Your access key ID");
DEFINE_string(access_secret, "xxxx", "Your access secret");
DEFINE_bool(tls, false, "Use HTTP2 with TLS/SSL");
int main(int argc, char* argv[]) {
gflags::ParseCommandLineFlags(&argc, &argv, true);
auto& logger = getLogger();
logger.setConsoleLevel(Level::Info);
logger.setLevel(Level::Info);
CredentialsProviderPtr credentials_provider;
if (!FLAGS_access_key.empty() && !FLAGS_access_secret.empty()) {
credentials_provider = std::make_shared<StaticCredentialsProvider>(FLAGS_access_key, FLAGS_access_secret);
auto producer = Producer::newBuilder()
.withConfiguration(Configuration::newBuilder()
.withEndpoints(FLAGS_access_point)
.withCredentialsProvider(credentials_provider)
.withTopics({FLAGS_topic})
std::atomic_bool stopped;
std::atomic_long count(0);
auto stats_lambda = [&] {
while (!stopped.load(std::memory_order_relaxed)) {
long cnt = count.load(std::memory_order_relaxed);
while (!count.compare_exchange_weak(cnt, 0)) {
cnt = count.load(std::memory_order_relaxed);
std::this_thread::sleep_for(std::chrono::seconds(1));
std::cout << "QPS: " << cnt << std::endl;
std::thread stats_thread(stats_lambda);
std::string body = randomString(FLAGS_message_body_size);
std::size_t completed = 0;
std::condition_variable cv;
std::unique_ptr<Semaphore> semaphore(new Semaphore(FLAGS_concurrency));
auto send_callback = [&](const std::error_code& ec, const SendReceipt& receipt) {
std::unique_lock<std::mutex> lk(mtx);
if (completed >= FLAGS_total) {
for (std::size_t i = 0; i < FLAGS_total; ++i) {
auto message = Message::newBuilder()
.withKeys({"Key-" + std::to_string(i)})
producer.send(std::move(message), send_callback);
std::unique_lock<std::mutex> lk(mtx);
cv.wait(lk, [&]() { return completed >= FLAGS_total; });
std::cout << "Completed: " << completed << ", total: " << FLAGS_total << std::endl;
stopped.store(true, std::memory_order_relaxed);
if (stats_thread.joinable()) {