zhuyijun/nacos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

For #5831, Remove the sets in connection event lister (#5832)

Browse Source
This commit is contained in:
杨翊 SionYang 2021-05-24 20:38:34 +08:00 committed by GitHub
parent f9d696beb4
commit 3d8b2c0b17
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
client/src/main/java/com/alibaba/nacos/client/naming/remote/gprc/NamingGrpcConnectionEventListener.java
View File

@ -42,7 +42,7 @@ public class NamingGrpcConnectionEventListener implements ConnectionEventListene
private final NamingGrpcClientProxy clientProxy;
private final ConcurrentMap<String, Set<Instance>> registeredInstanceCached = new ConcurrentHashMap<String, Set<Instance>>();
private final ConcurrentMap<String, Instance> registeredInstanceCached = new ConcurrentHashMap<>();
private final Set<String> subscribes = new ConcurrentHashSet<String>();
@ -70,21 +70,19 @@ public class NamingGrpcConnectionEventListener implements ConnectionEventListene
private void redoRegisterEachService() {
LogUtils.NAMING_LOGGER.info("Grpc re-connect, redo register services");
for (Map.Entry<String, Set<Instance>> each : registeredInstanceCached.entrySet()) {
for (Map.Entry<String, Instance> each : registeredInstanceCached.entrySet()) {
String serviceName = NamingUtils.getServiceName(each.getKey());
String groupName = NamingUtils.getGroupName(each.getKey());
redoRegisterEachInstance(serviceName, groupName, each.getValue());
}
}
private void redoRegisterEachInstance(String serviceName, String groupName, Set<Instance> instances) {
for (Instance each : instances) {
try {
clientProxy.registerService(serviceName, groupName, each);
} catch (NacosException e) {
LogUtils.NAMING_LOGGER
.warn(String.format("redo register for service %s@@%s failed", groupName, serviceName), e);
}
private void redoRegisterEachInstance(String serviceName, String groupName, Instance instance) {
try {
clientProxy.registerService(serviceName, groupName, instance);
} catch (NacosException e) {
LogUtils.NAMING_LOGGER.warn(String
.format("redo register for service %s@@%s, %s failed", groupName, serviceName, instance.toString()), e);
}
}
@ -102,8 +100,7 @@ public class NamingGrpcConnectionEventListener implements ConnectionEventListene
*/
public void cacheInstanceForRedo(String serviceName, String groupName, Instance instance) {
String key = NamingUtils.getGroupedName(serviceName, groupName);
registeredInstanceCached.putIfAbsent(key, new ConcurrentHashSet<Instance>());
registeredInstanceCached.get(key).add(instance);
registeredInstanceCached.put(key, instance);
}
/**
@ -115,10 +112,7 @@ public class NamingGrpcConnectionEventListener implements ConnectionEventListene
*/
public void removeInstanceForRedo(String serviceName, String groupName, Instance instance) {
String key = NamingUtils.getGroupedName(serviceName, groupName);
Set<Instance> instances = registeredInstanceCached.get(key);
if (null != instances) {
instances.remove(instance);
}
registeredInstanceCached.remove(key);
}
public void cacheSubscriberForRedo(String fullServiceName, String cluster) {