Skip to content

Commit 6148b89

Browse files
authored
[ISSUE #9174] Add a collection of predefined Groups and common checking methods in the MixAll (#9175)
Signed-off-by: ltamber <ltamber12@gmail.com>
1 parent c24f654 commit 6148b89

File tree

2 files changed

+23
-19
lines changed

2 files changed

+23
-19
lines changed

common/src/main/java/org/apache/rocketmq/common/MixAll.java

+22
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@
4444
import java.util.concurrent.atomic.AtomicLong;
4545
import java.util.function.Predicate;
4646

47+
import com.google.common.collect.ImmutableSet;
4748
import org.apache.commons.lang3.StringUtils;
4849
import org.apache.rocketmq.common.annotation.ImportantField;
4950
import org.apache.rocketmq.common.constant.LoggerName;
@@ -120,6 +121,23 @@ public class MixAll {
120121

121122
private static final String OS = System.getProperty("os.name").toLowerCase();
122123

124+
private static final Set<String> PREDEFINE_GROUP_SET = ImmutableSet.of(
125+
DEFAULT_CONSUMER_GROUP,
126+
DEFAULT_PRODUCER_GROUP,
127+
TOOLS_CONSUMER_GROUP,
128+
SCHEDULE_CONSUMER_GROUP,
129+
FILTERSRV_CONSUMER_GROUP,
130+
MONITOR_CONSUMER_GROUP,
131+
CLIENT_INNER_PRODUCER_GROUP,
132+
SELF_TEST_PRODUCER_GROUP,
133+
SELF_TEST_CONSUMER_GROUP,
134+
ONS_HTTP_PROXY_GROUP,
135+
CID_ONSAPI_PERMISSION_GROUP,
136+
CID_ONSAPI_OWNER_GROUP,
137+
CID_ONSAPI_PULL_GROUP,
138+
CID_SYS_RMQ_TRANS
139+
);
140+
123141
public static boolean isWindows() {
124142
return OS.contains("win");
125143
}
@@ -160,6 +178,10 @@ public static boolean isSysConsumerGroup(final String consumerGroup) {
160178
return consumerGroup.startsWith(CID_RMQ_SYS_PREFIX);
161179
}
162180

181+
public static boolean isPredefinedGroup(final String consumerGroup) {
182+
return PREDEFINE_GROUP_SET.contains(consumerGroup);
183+
}
184+
163185
public static String getDLQTopic(final String consumerGroup) {
164186
return DLQ_GROUP_TOPIC_PREFIX + consumerGroup;
165187
}

tools/src/main/java/org/apache/rocketmq/tools/admin/DefaultMQAdminExtImpl.java

+1-19
Original file line numberDiff line numberDiff line change
@@ -128,24 +128,6 @@
128128
public class DefaultMQAdminExtImpl implements MQAdminExt, MQAdminExtInner {
129129

130130
private static final String SOCKS_PROXY_JSON = "socksProxyJson";
131-
private static final Set<String> SYSTEM_GROUP_SET = new HashSet<>();
132-
133-
static {
134-
SYSTEM_GROUP_SET.add(MixAll.DEFAULT_CONSUMER_GROUP);
135-
SYSTEM_GROUP_SET.add(MixAll.DEFAULT_PRODUCER_GROUP);
136-
SYSTEM_GROUP_SET.add(MixAll.TOOLS_CONSUMER_GROUP);
137-
SYSTEM_GROUP_SET.add(MixAll.SCHEDULE_CONSUMER_GROUP);
138-
SYSTEM_GROUP_SET.add(MixAll.FILTERSRV_CONSUMER_GROUP);
139-
SYSTEM_GROUP_SET.add(MixAll.MONITOR_CONSUMER_GROUP);
140-
SYSTEM_GROUP_SET.add(MixAll.CLIENT_INNER_PRODUCER_GROUP);
141-
SYSTEM_GROUP_SET.add(MixAll.SELF_TEST_PRODUCER_GROUP);
142-
SYSTEM_GROUP_SET.add(MixAll.SELF_TEST_CONSUMER_GROUP);
143-
SYSTEM_GROUP_SET.add(MixAll.ONS_HTTP_PROXY_GROUP);
144-
SYSTEM_GROUP_SET.add(MixAll.CID_ONSAPI_PERMISSION_GROUP);
145-
SYSTEM_GROUP_SET.add(MixAll.CID_ONSAPI_OWNER_GROUP);
146-
SYSTEM_GROUP_SET.add(MixAll.CID_ONSAPI_PULL_GROUP);
147-
SYSTEM_GROUP_SET.add(MixAll.CID_SYS_RMQ_TRANS);
148-
}
149131

150132
private final Logger logger = LoggerFactory.getLogger(DefaultMQAdminExtImpl.class);
151133
private final DefaultMQAdminExt defaultMQAdminExt;
@@ -1698,7 +1680,7 @@ public SubscriptionGroupWrapper getUserSubscriptionGroup(final String brokerAddr
16981680
Iterator<Entry<String, SubscriptionGroupConfig>> iterator = subscriptionGroupWrapper.getSubscriptionGroupTable().entrySet().iterator();
16991681
while (iterator.hasNext()) {
17001682
Map.Entry<String, SubscriptionGroupConfig> configEntry = iterator.next();
1701-
if (MixAll.isSysConsumerGroup(configEntry.getKey()) || SYSTEM_GROUP_SET.contains(configEntry.getKey())) {
1683+
if (MixAll.isSysConsumerGroup(configEntry.getKey()) || MixAll.isPredefinedGroup(configEntry.getKey())) {
17021684
iterator.remove();
17031685
}
17041686
}

0 commit comments

Comments
 (0)