|
17 | 17 | package org.apache.rocketmq.broker.slave; |
18 | 18 |
|
19 | 19 | import java.io.IOException; |
| 20 | +import java.util.Iterator; |
| 21 | +import java.util.Map; |
20 | 22 | import java.util.concurrent.ConcurrentHashMap; |
21 | 23 | import java.util.concurrent.ConcurrentMap; |
22 | 24 |
|
23 | 25 | import org.apache.commons.lang3.StringUtils; |
24 | 26 | import org.apache.rocketmq.broker.BrokerController; |
25 | 27 | import org.apache.rocketmq.broker.loadbalance.MessageRequestModeManager; |
26 | 28 | import org.apache.rocketmq.broker.subscription.SubscriptionGroupManager; |
| 29 | +import org.apache.rocketmq.broker.topic.TopicConfigManager; |
27 | 30 | import org.apache.rocketmq.common.MixAll; |
28 | 31 | import org.apache.rocketmq.common.TopicConfig; |
29 | 32 | import org.apache.rocketmq.common.constant.LoggerName; |
@@ -77,20 +80,28 @@ private void syncTopicConfig() { |
77 | 80 | try { |
78 | 81 | TopicConfigAndMappingSerializeWrapper topicWrapper = |
79 | 82 | this.brokerController.getBrokerOuterAPI().getAllTopicConfig(masterAddrBak); |
80 | | - if (!this.brokerController.getTopicConfigManager().getDataVersion() |
81 | | - .equals(topicWrapper.getDataVersion())) { |
| 83 | + TopicConfigManager topicConfigManager = this.brokerController.getTopicConfigManager(); |
| 84 | + if (!topicConfigManager.getDataVersion().equals(topicWrapper.getDataVersion())) { |
82 | 85 |
|
83 | | - this.brokerController.getTopicConfigManager().getDataVersion() |
84 | | - .assignNewOne(topicWrapper.getDataVersion()); |
| 86 | + topicConfigManager.getDataVersion().assignNewOne(topicWrapper.getDataVersion()); |
85 | 87 |
|
86 | 88 | ConcurrentMap<String, TopicConfig> newTopicConfigTable = topicWrapper.getTopicConfigTable(); |
| 89 | + ConcurrentMap<String, TopicConfig> topicConfigTable = topicConfigManager.getTopicConfigTable(); |
| 90 | + |
87 | 91 | //delete |
88 | | - ConcurrentMap<String, TopicConfig> topicConfigTable = this.brokerController.getTopicConfigManager().getTopicConfigTable(); |
89 | | - topicConfigTable.entrySet().removeIf(item -> !newTopicConfigTable.containsKey(item.getKey())); |
| 92 | + Iterator<Map.Entry<String, TopicConfig>> iterator = topicConfigTable.entrySet().iterator(); |
| 93 | + while (iterator.hasNext()) { |
| 94 | + Map.Entry<String, TopicConfig> entry = iterator.next(); |
| 95 | + if (!newTopicConfigTable.containsKey(entry.getKey())) { |
| 96 | + iterator.remove(); |
| 97 | + } |
| 98 | + topicConfigManager.deleteTopicConfig(entry.getKey()); |
| 99 | + } |
| 100 | + |
90 | 101 | //update |
91 | | - topicConfigTable.putAll(newTopicConfigTable); |
| 102 | + newTopicConfigTable.values().forEach(topicConfigManager::updateSingleTopicConfigWithoutPersist); |
92 | 103 |
|
93 | | - this.brokerController.getTopicConfigManager().persist(); |
| 104 | + topicConfigManager.persist(); |
94 | 105 | } |
95 | 106 | if (topicWrapper.getTopicQueueMappingDetailMap() != null |
96 | 107 | && !topicWrapper.getMappingDataVersion().equals(this.brokerController.getTopicQueueMappingManager().getDataVersion())) { |
@@ -165,19 +176,24 @@ private void syncSubscriptionGroupConfig() { |
165 | 176 |
|
166 | 177 | if (!this.brokerController.getSubscriptionGroupManager().getDataVersion() |
167 | 178 | .equals(subscriptionWrapper.getDataVersion())) { |
168 | | - SubscriptionGroupManager subscriptionGroupManager = |
169 | | - this.brokerController.getSubscriptionGroupManager(); |
170 | | - subscriptionGroupManager.getDataVersion().assignNewOne( |
171 | | - subscriptionWrapper.getDataVersion()); |
| 179 | + SubscriptionGroupManager subscriptionGroupManager = this.brokerController.getSubscriptionGroupManager(); |
| 180 | + subscriptionGroupManager.getDataVersion().assignNewOne(subscriptionWrapper.getDataVersion()); |
172 | 181 |
|
173 | 182 | ConcurrentMap<String, SubscriptionGroupConfig> curSubscriptionGroupTable = |
174 | 183 | subscriptionGroupManager.getSubscriptionGroupTable(); |
175 | 184 | ConcurrentMap<String, SubscriptionGroupConfig> newSubscriptionGroupTable = |
176 | 185 | subscriptionWrapper.getSubscriptionGroupTable(); |
177 | 186 | // delete |
178 | | - curSubscriptionGroupTable.entrySet().removeIf(e -> !newSubscriptionGroupTable.containsKey(e.getKey())); |
| 187 | + Iterator<Map.Entry<String, SubscriptionGroupConfig>> iterator = curSubscriptionGroupTable.entrySet().iterator(); |
| 188 | + while (iterator.hasNext()) { |
| 189 | + Map.Entry<String, SubscriptionGroupConfig> configEntry = iterator.next(); |
| 190 | + if (!newSubscriptionGroupTable.containsKey(configEntry.getKey())) { |
| 191 | + iterator.remove(); |
| 192 | + } |
| 193 | + subscriptionGroupManager.deleteSubscriptionGroupConfig(configEntry.getKey()); |
| 194 | + } |
179 | 195 | // update |
180 | | - curSubscriptionGroupTable.putAll(newSubscriptionGroupTable); |
| 196 | + newSubscriptionGroupTable.values().forEach(subscriptionGroupManager::updateSubscriptionGroupConfigWithoutPersist); |
181 | 197 | // persist |
182 | 198 | subscriptionGroupManager.persist(); |
183 | 199 | LOGGER.info("Update slave Subscription Group from master, {}", masterAddrBak); |
|
0 commit comments