diff --git a/include/Arg_helper.h b/include/Arg_helper.h index bda4fce08..bc2896232 100644 --- a/include/Arg_helper.h +++ b/include/Arg_helper.h @@ -23,7 +23,6 @@ #include "RocketMQClient.h" namespace rocketmq { -// m_args; }; -//& mqs); - virtual void persistConsumerOffset(); - virtual void persistConsumerOffsetByResetOffset(); - virtual void updateTopicSubscribeInfo(const std::string& topic, std::vector& info); - virtual ConsumeType getConsumeType(); - virtual ConsumeFromWhere getConsumeFromWhere(); - virtual void getSubscriptions(std::vector&); - virtual void updateConsumeOffset(const MQMessageQueue& mq, int64 offset); - virtual void removeConsumeOffset(const MQMessageQueue& mq); - //& info); + ConsumeFromWhere getConsumeFromWhere(); + void getSubscriptions(std::vector&); + void updateConsumeOffset(const MQMessageQueue& mq, int64 offset); + void removeConsumeOffset(const MQMessageQueue& mq); + + void registerMessageQueueListener(const std::string& topic, MQueueListener* pListener); int64 fetchConsumeOffset(const MQMessageQueue& mq, bool fromStore); void fetchMessageQueuesInBalance(const std::string& topic, std::vector mqs); - // temp persist consumer offset interface, only valid with - // RemoteBrokerOffsetStore, updateConsumeOffset should be called before. void persistConsumerOffset4PullConsumer(const MQMessageQueue& mq); private: DefaultMQPullConsumerImpl* impl; }; -// #include #include -#include "CCommon.h" #include "RocketMQClient.h" namespace rocketmq { -// m_properties; }; -//& mqs, const MQMessage& msg, void* arg) = 0; }; -//& mqAll, std::vector& mqDivided) = 0; }; -// msgFoundList; }; -//& info) { impl->updateTopicSubscribeInfo(topic, info); } -ConsumeType DefaultMQPullConsumer::getConsumeType() { - return impl->getConsumeType(); -} + ConsumeFromWhere DefaultMQPullConsumer::getConsumeFromWhere() { return impl->getConsumeFromWhere(); } diff --git a/src/consumer/DefaultMQPushConsumer.cpp b/src/consumer/DefaultMQPushConsumer.cpp index cede0c9a8..b0373c0dd 100644 --- a/src/consumer/DefaultMQPushConsumer.cpp +++ b/src/consumer/DefaultMQPushConsumer.cpp @@ -35,9 +35,10 @@ void DefaultMQPushConsumer::shutdown() { impl->shutdown(); } -ConsumeType DefaultMQPushConsumer::getConsumeType() { - return impl->getConsumeType(); -} +// ConsumeType DefaultMQPushConsumer::getConsumeType() { +// return impl->getConsumeType(); +//} + ConsumeFromWhere DefaultMQPushConsumer::getConsumeFromWhere() { return impl->getConsumeFromWhere(); } diff --git a/test/src/message/MQMessageIdTest.cpp b/test/src/message/MQMessageIdTest.cpp index d83de2f28..6e97480f8 100644 --- a/test/src/message/MQMessageIdTest.cpp +++ b/test/src/message/MQMessageIdTest.cpp @@ -45,12 +45,15 @@ TEST(messageId, id) { EXPECT_EQ(host, inet_addr("127.0.0.2")); EXPECT_EQ(port, 10092); EXPECT_EQ(id.getOffset(), 2048); + + MQMessageId id2 = id; + EXPECT_EQ(id2.getOffset(), 2048); } int main(int argc, char* argv[]) { InitGoogleMock(&argc, argv); - testing::GTEST_FLAG(filter) = "messageId.id"; + testing::GTEST_FLAG(filter) = "messageId.*"; int itestts = RUN_ALL_TESTS(); return itestts; } diff --git a/test/src/message/MQMessageTest.cpp b/test/src/message/MQMessageTest.cpp index 100e73eb2..0f55cbb29 100644 --- a/test/src/message/MQMessageTest.cpp +++ b/test/src/message/MQMessageTest.cpp @@ -70,6 +70,13 @@ TEST(message, Init) { EXPECT_EQ(messageSix.getTags(), "tagTest"); EXPECT_EQ(messageSix.getKeys(), "testKey"); EXPECT_EQ(messageSix.getFlag(), 1); + + MQMessage messageSeven = messageSix; + EXPECT_EQ(messageSeven.getTopic(), "test"); + EXPECT_EQ(messageSeven.getBody(), "testBody"); + EXPECT_EQ(messageSeven.getTags(), "tagTest"); + EXPECT_EQ(messageSeven.getKeys(), "testKey"); + EXPECT_EQ(messageSeven.getFlag(), 1); } TEST(message, info) { @@ -145,6 +152,15 @@ TEST(message, properties) { EXPECT_EQ(message.getProperty(MQMessage::PROPERTY_TRANSACTION_PREPARED), "false"); } +TEST(message, Keys) { + MQMessage message; + vector keys; + keys.push_back("abc"); + keys.push_back("efg"); + keys.push_back("hij"); + message.setKeys(keys); + EXPECT_EQ(message.getKeys(), "abc efg hij"); +} int main(int argc, char* argv[]) { InitGoogleMock(&argc, argv);