-
Notifications
You must be signed in to change notification settings - Fork 165
Expand file tree
/
Copy pathPushConsumer.cpp
More file actions
91 lines (76 loc) · 2.56 KB
/
PushConsumer.cpp
File metadata and controls
91 lines (76 loc) · 2.56 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
/*
* 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 "common.h"
#include "concurrent/latch.hpp"
#include "DefaultMQPushConsumer.h"
#include "UtilAll.h"
using namespace rocketmq;
TpsReportService g_tps;
latch g_finished(1);
class MyMsgListener : public MessageListenerConcurrently {
public:
virtual ~MyMsgListener() = default;
ConsumeStatus consumeMessage(std::vector<MQMessageExt>& msgs) override {
auto old = g_msg_count.fetch_sub(msgs.size());
if (old > 0) {
for (size_t i = 0; i < msgs.size(); ++i) {
g_tps.Increment();
if (msgs[i] == nullptr) {
std::cout << "error!!!" << std::endl;
} else {
std::cout << msgs[i].msg_id() << ", body: " << msgs[i].body() << std::endl;
}
}
if (old <= msgs.size()) {
g_finished.count_down();
}
return CONSUME_SUCCESS;
} else {
return RECONSUME_LATER;
}
}
};
int main(int argc, char* argv[]) {
RocketmqSendAndConsumerArgs info;
if (!ParseArgs(argc, argv, &info)) {
exit(-1);
}
PrintRocketmqSendAndConsumerArgs(info);
auto* consumer = new DefaultMQPushConsumer(info.groupname);
consumer->set_namesrv_addr(info.namesrv);
consumer->set_group_name(info.groupname);
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);
if (info.broadcasting) {
consumer->set_message_model(BROADCASTING);
}
consumer->subscribe(info.topic, "*");
MyMsgListener msglistener;
consumer->registerMessageListener(&msglistener);
g_tps.start();
try {
consumer->start();
} catch (MQClientException& e) {
std::cout << e << std::endl;
}
g_finished.wait();
consumer->shutdown();
delete consumer;
return 0;
}