Examples with ConsumeMessageHook org.apache.rocketmq.broker.mqtrace.ConsumeMessageHook used on opensource projects

Example 1 with ConsumeMessageHook

use of org.apache.rocketmq.broker.mqtrace.ConsumeMessageHook in project rocketmq-rocketmq-all-4.1.0-incubating by lirenzuo.

the class PullMessageProcessorTest method testProcessRequest_FoundWithHook.

@Test
public void testProcessRequest_FoundWithHook() throws RemotingCommandException {
    GetMessageResult getMessageResult = createGetMessageResult();
    when(messageStore.getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), any(ExpressionMessageFilter.class))).thenReturn(getMessageResult);
    List<ConsumeMessageHook> consumeMessageHookList = new ArrayList<>();
    final ConsumeMessageContext[] messageContext = new ConsumeMessageContext[1];
    ConsumeMessageHook consumeMessageHook = new ConsumeMessageHook() {

        @Override
        public String hookName() {
            return "TestHook";
        }

        @Override
        public void consumeMessageBefore(ConsumeMessageContext context) {
            messageContext[0] = context;
        }

        @Override
        public void consumeMessageAfter(ConsumeMessageContext context) {
        }
    };
    consumeMessageHookList.add(consumeMessageHook);
    pullMessageProcessor.registerConsumeMessageHook(consumeMessageHookList);
    final RemotingCommand request = createPullMsgCommand(RequestCode.PULL_MESSAGE);
    RemotingCommand response = pullMessageProcessor.processRequest(handlerContext, request);
    assertThat(response).isNotNull();
    assertThat(response.getCode()).isEqualTo(ResponseCode.SUCCESS);
    assertThat(messageContext[0]).isNotNull();
    assertThat(messageContext[0].getConsumerGroup()).isEqualTo(group);
    assertThat(messageContext[0].getTopic()).isEqualTo(topic);
    assertThat(messageContext[0].getQueueId()).isEqualTo(1);
}

Example 2 with ConsumeMessageHook

use of org.apache.rocketmq.broker.mqtrace.ConsumeMessageHook in project rocketmq by apache.

the class PullMessageProcessorTest method testProcessRequest_FoundWithHook.

@Test
public void testProcessRequest_FoundWithHook() throws RemotingCommandException {
    GetMessageResult getMessageResult = createGetMessageResult();
    when(messageStore.getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), any(ExpressionMessageFilter.class))).thenReturn(getMessageResult);
    List<ConsumeMessageHook> consumeMessageHookList = new ArrayList<>();
    final ConsumeMessageContext[] messageContext = new ConsumeMessageContext[1];
    ConsumeMessageHook consumeMessageHook = new ConsumeMessageHook() {

        @Override
        public String hookName() {
            return "TestHook";
        }

        @Override
        public void consumeMessageBefore(ConsumeMessageContext context) {
            messageContext[0] = context;
        }

        @Override
        public void consumeMessageAfter(ConsumeMessageContext context) {
        }
    };
    consumeMessageHookList.add(consumeMessageHook);
    pullMessageProcessor.registerConsumeMessageHook(consumeMessageHookList);
    final RemotingCommand request = createPullMsgCommand(RequestCode.PULL_MESSAGE);
    RemotingCommand response = pullMessageProcessor.processRequest(handlerContext, request);
    assertThat(response).isNotNull();
    assertThat(response.getCode()).isEqualTo(ResponseCode.SUCCESS);
    assertThat(messageContext[0]).isNotNull();
    assertThat(messageContext[0].getConsumerGroup()).isEqualTo(group);
    assertThat(messageContext[0].getTopic()).isEqualTo(topic);
    assertThat(messageContext[0].getQueueId()).isEqualTo(1);
}