opt

apache · Feb 4, 2024 · 2e0b742 · 2e0b742
1 parent 744c395
commit 2e0b742
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/core/src/main/java/org/apache/seata/core/rpc/processor/server/RegRmProcessor.java b/core/src/main/java/org/apache/seata/core/rpc/processor/server/RegRmProcessor.java
@@ -89,9 +89,9 @@ private void onRegRmMessage(ChannelHandlerContext ctx, RpcMessage rpcMessage) {
             }
         }
         RegisterRMResponse response = new RegisterRMResponse(isSuccess);
-        response.setResultCode(resultCode);
-        if (checkAuthHandler.refreshAuthToken(message) != null) {
-            response.setExtraData(checkAuthHandler.refreshAuthToken(message));
+        String newToken = checkAuthHandler.refreshAuthToken(message);
+        if (StringUtils.isNotBlank(newToken)) {
+            response.setExtraData(newToken);
         }
         if (StringUtils.isNotEmpty(errorInfo)) {
             response.setMsg(errorInfo);

diff --git a/core/src/main/java/org/apache/seata/core/rpc/processor/server/RegTmProcessor.java b/core/src/main/java/org/apache/seata/core/rpc/processor/server/RegTmProcessor.java
@@ -92,8 +92,9 @@ private void onRegTmMessage(ChannelHandlerContext ctx, RpcMessage rpcMessage) {
             }
         }
         RegisterTMResponse response = new RegisterTMResponse(isSuccess);
-        if (checkAuthHandler.refreshAuthToken(message) != null) {
-            response.setExtraData(checkAuthHandler.refreshAuthToken(message));
+        String newToken = checkAuthHandler.refreshAuthToken(message);
+        if (StringUtils.isNotBlank(newToken)) {
+            response.setExtraData(newToken);
         }
         if (StringUtils.isNotEmpty(errorInfo)) {
             response.setMsg(errorInfo);