From de76e06d6da8e8646d2ac59ea061e8dc2653aae8 Mon Sep 17 00:00:00 2001
From: cserwen <cserwen@163.com>
Date: Mon, 14 Feb 2022 15:11:06 +0800
Subject: [PATCH] add hook to execute after consume for pop (#3780)

---
 .../impl/consumer/ConsumeMessagePopConcurrentlyService.java    | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/client/src/main/java/org/apache/rocketmq/client/impl/consumer/ConsumeMessagePopConcurrentlyService.java b/client/src/main/java/org/apache/rocketmq/client/impl/consumer/ConsumeMessagePopConcurrentlyService.java
index 910f5926ff..f3e661bf50 100644
--- a/client/src/main/java/org/apache/rocketmq/client/impl/consumer/ConsumeMessagePopConcurrentlyService.java
+++ b/client/src/main/java/org/apache/rocketmq/client/impl/consumer/ConsumeMessagePopConcurrentlyService.java
@@ -455,6 +455,9 @@ public class ConsumeMessagePopConcurrentlyService implements ConsumeMessageServi
 
             if (ConsumeMessagePopConcurrentlyService.this.defaultMQPushConsumerImpl.hasHook()) {
                 consumeMessageContext.getProps().put(MixAll.CONSUME_CONTEXT_TYPE, returnType.name());
+                consumeMessageContext.setStatus(status.toString());
+                consumeMessageContext.setSuccess(ConsumeConcurrentlyStatus.CONSUME_SUCCESS == status);
+                ConsumeMessagePopConcurrentlyService.this.defaultMQPushConsumerImpl.executeHookAfter(consumeMessageContext);
             }
 
             ConsumeMessagePopConcurrentlyService.this.getConsumerStatsManager()