#include "gflags/gflags.h"
#include "rocketmq/Message.h"
#include "rocketmq/Producer.h"
using namespace ROCKETMQ_NAMESPACE;
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);
auto checker = [](const Message& message) -> TransactionState {
std::cout << "Recovery orphan transactional message[topic=" << message.topic() << ", MsgId=" << message.id()
<< ", txn-id=" << message.extension().transaction_id << std::endl;
return TransactionState::COMMIT;
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})
.withTransactionChecker(checker)
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);
auto message = Message::newBuilder().withTopic(FLAGS_topic).withTag("TagA").withBody(body).build();
auto transaction = producer.beginTransaction();
producer.send(std::move(message), ec, *transaction);
if (!transaction->commit()) {
std::cerr << "Failed to commit message" << std::endl;
std::cerr << "Failed to send transactional message to topic: " << FLAGS_topic << std::endl;
std::cerr << "Ah...No!!!" << std::endl;
stopped.store(true, std::memory_order_relaxed);
if (stats_thread.joinable()) {
std::cout << "Wait 5 minutes for potential unresolved transactional message callback" << std::endl;
std::this_thread::sleep_for(std::chrono::minutes(5));