重新注册时使用最后一次的注册实例信息 (#595)

Author: jianliu

polaris-discovery/polaris-discovery-client/src/main/java/com/tencent/polaris/discovery/client/flow/RegisterStateManager.java

@@ -55,7 +55,9 @@ public static RegisterState putRegisterState(SDKContext sdkContext,
         String registerStateKey = buildRegisterStateKey(instanceRegisterRequest);
         Map<String, RegisterState> sdkRegisterStates = REGISTER_STATES.computeIfAbsent(
                 sdkContext.getValueContext().getClientId(), clientId -> new ConcurrentHashMap<>());
-        if (sdkRegisterStates.containsKey(registerStateKey)) {
+        RegisterState existsRegisterState = sdkRegisterStates.get(registerStateKey);
+        if (existsRegisterState != null) {
+             existsRegisterState.setInstanceRegisterRequest(instanceRegisterRequest);
             return null;
         }
         return sdkRegisterStates.computeIfAbsent(registerStateKey, unused -> {
@@ -112,7 +114,7 @@ private static String buildRegisterStateKey(CommonProviderBaseEntity baseEntity)
 
     public static class RegisterState {
 
-        private InstanceRegisterRequest instanceRegisterRequest;
+        private volatile InstanceRegisterRequest instanceRegisterRequest;
         private long firstRegisterTime;
         private ScheduledFuture<?> taskFuture;
         private final AtomicInteger heartbeatFailCounter = new AtomicInteger(0);