-
Notifications
You must be signed in to change notification settings - Fork 165
Expand file tree
/
Copy pathRequestReply.cpp
More file actions
115 lines (96 loc) · 3.92 KB
/
RequestReply.cpp
File metadata and controls
115 lines (96 loc) · 3.92 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <iostream>
#include "../src/common/UtilAll.h"
#include "common.h"
#include "MessageUtil.h"
#include "DefaultMQProducer.h"
#include "DefaultMQPushConsumer.h"
using namespace rocketmq;
class MyResponseMessageListener : public MessageListenerConcurrently {
public:
MyResponseMessageListener(DefaultMQProducer* replyProducer) : m_replyProducer(replyProducer) {}
virtual ~MyResponseMessageListener() = default;
ConsumeStatus consumeMessage(std::vector<MQMessageExt>& msgs) override {
for (const auto& msg : msgs) {
try {
std::cout << "handle message: " << msg.toString() << std::endl;
const auto& replyTo = MessageUtil::getReplyToClient(msg);
// create reply message with given util, do not create reply message by yourself
MQMessage replyMessage = MessageUtil::createReplyMessage(msg, "reply message contents.");
// send reply message with producer
SendResult replyResult = m_replyProducer->send(replyMessage, 10000);
std::cout << "reply to " << replyTo << ", " << replyResult.toString() << std::endl;
} catch (const std::exception& e) {
std::cout << e.what() << std::endl;
}
}
return CONSUME_SUCCESS;
}
private:
DefaultMQProducer* m_replyProducer;
};
int main(int argc, char* argv[]) {
RocketmqSendAndConsumerArgs info;
if (!ParseArgs(argc, argv, &info)) {
exit(-1);
}
PrintRocketmqSendAndConsumerArgs(info);
DefaultMQProducer producer("");
producer.set_namesrv_addr(info.namesrv);
producer.set_send_msg_timeout(3000);
producer.set_retry_times(info.retrytimes);
producer.set_retry_times_for_async(info.retrytimes);
producer.set_send_latency_fault_enable(!info.selectUnactiveBroker);
producer.set_tcp_transport_try_lock_timeout(1000);
producer.set_tcp_transport_connect_timeout(400);
producer.start();
DefaultMQPushConsumer consumer(info.groupname);
consumer.set_namesrv_addr(info.namesrv);
consumer.set_tcp_transport_try_lock_timeout(1000);
consumer.set_tcp_transport_connect_timeout(400);
consumer.set_consume_thread_nums(info.thread_count);
consumer.set_consume_from_where(CONSUME_FROM_LAST_OFFSET);
// recommend client configs
consumer.set_pull_time_delay_millis_when_exception(0L);
consumer.subscribe(info.topic, "*");
MyResponseMessageListener msglistener(&producer);
consumer.registerMessageListener(&msglistener);
try {
consumer.start();
} catch (MQClientException& e) {
std::cout << e << std::endl;
return -1;
}
// std::this_thread::sleep_for(std::chrono::seconds(10));
int msg_count = g_msg_count.load();
for (int count = 0; count < msg_count; count++) {
try {
MQMessage msg(info.topic, "Hello world");
auto begin = UtilAll::currentTimeMillis();
MQMessage retMsg = producer.request(msg, 10000);
auto cost = UtilAll::currentTimeMillis() - begin;
std::cout << count << " >>> request to <" << info.topic << "> cost: " << cost
<< " replyMessage: " << retMsg.toString() << std::endl;
} catch (const std::exception& e) {
std::cout << count << " >>> " << e.what() << std::endl;
}
}
consumer.shutdown();
producer.shutdown();
return 0;
}