From 65779b2b1051a587ac12b24d7f22555702795bea Mon Sep 17 00:00:00 2001
From: yx9o <yangx_soft@163.com>
Date: Mon, 13 Jan 2025 16:21:58 +0800
Subject: [PATCH] Update

---
 .../broker/client/DefaultConsumerIdsChangeListener.java         | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/broker/src/main/java/org/apache/rocketmq/broker/client/DefaultConsumerIdsChangeListener.java b/broker/src/main/java/org/apache/rocketmq/broker/client/DefaultConsumerIdsChangeListener.java
index 12af3ee9a1e..e0461769568 100644
--- a/broker/src/main/java/org/apache/rocketmq/broker/client/DefaultConsumerIdsChangeListener.java
+++ b/broker/src/main/java/org/apache/rocketmq/broker/client/DefaultConsumerIdsChangeListener.java
@@ -91,7 +91,7 @@ public void handle(ConsumerGroupEvent event, String group, Object... args) {
                             this.brokerController.getBroker2Client().notifyConsumerIdsChanged(chl, group);
                         }
                         if (isNormalCompletion) {
-                            activeGroupNotifyMap.remove(group);
+                            activeGroupNotifyMap.computeIfPresent(group, (k, val) -> val == currentNotifyTaskControl ? null : val);
                         }
                     } else {
                         consumerChannelMap.put(group, channels);