#include <algorithm>
#include <atomic>
#include <iostream>
#include <memory>
#include <random>
#include <string>
#include <system_error>
#include "rocketmq/CredentialsProvider.h"
#include "rocketmq/Logger.h"
#include "rocketmq/Message.h"
#include "rocketmq/Producer.h"
using namespace ROCKETMQ_NAMESPACE;
const std::string &alphaNumeric() {
static std::string alpha_numeric("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ");
return alpha_numeric;
}
std::string randomString(std::string::size_type len) {
std::string result;
result.reserve(len);
std::random_device rd;
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));
generated += delta;
}
return result;
}
static const std::string topic = "xxx";
static const std::string access_point = "rmq-xxx.rocketmq.xxxtencenttdmq.com:8081";
static const std::string access_key = "xxx";
static const std::string access_secret = "xxx";
static const uint32_t total = 32;
static const int32_t message_body_size = 128;
int main(int argc, char *argv[]) {
CredentialsProviderPtr credentials_provider;
if (!access_key.empty() && !access_secret.empty()) {
credentials_provider = std::make_shared<StaticCredentialsProvider>(access_key, access_secret);
}
auto producer = Producer::newBuilder()
.withConfiguration(Configuration::newBuilder()
.withEndpoints(access_point)
.withCredentialsProvider(credentials_provider)
.withSsl(false)
.build())
.withTopics({topic})
.build();
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)) {
break;
}
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(message_body_size);
try {
for (std::size_t i = 0; i < total; ++i) {
auto message = Message::newBuilder()
.withTopic(topic)
.withTag("TagA")
.withKeys({"Key-" + std::to_string(i)})
.withBody(body)
.build();
std::error_code ec;
SendReceipt send_receipt = producer.send(std::move(message), ec);
if (ec) {
std::cerr << "Failed to publish message to " << topic << ". Cause: " << ec.message() << std::endl;
} else {
std::cout << "Publish message to " << topic << " OK. Message-ID: " << send_receipt.message_id
<< std::endl;
count++;
}
}
} catch (...) {
std::cerr << "Ah...No!!!" << std::endl;
}
stopped.store(true, std::memory_order_relaxed);
if (stats_thread.joinable()) {
stats_thread.join();
}
return EXIT_SUCCESS;
}
Was this page helpful?