[Bug][dev]Fix the token cannot be refreshed properly. (#4409)

Author: gaoyan1998

dinky-admin/src/main/java/org/dinky/controller/TokenController.java

@@ -62,6 +62,7 @@
 public class TokenController {
 
     private final TokenService tokenService;
+    private final org.dinky.service.impl.TokenService tokenServiceStp;
     private final StpLogic stpLogic;
 
     /**
@@ -92,9 +93,13 @@ public ProTableResult<SysToken> listToken(@RequestBody JsonNode params) {
             mode = SaMode.OR)
     public Result<Void> saveOrUpdateToken(@RequestBody SysToken sysToken) {
         sysToken.setSource(SysToken.Source.CUSTOM);
-        return tokenService.saveOrUpdate(sysToken)
+        Result<Void> result = tokenService.saveOrUpdate(sysToken)
                 ? Result.succeed(Status.SAVE_SUCCESS)
                 : Result.failed(Status.SAVE_FAILED);
+        if (result.isSuccess()) {
+            tokenServiceStp.injectToken(sysToken);
+        }
+        return result;
     }
 
     /**

dinky-admin/src/main/java/org/dinky/service/impl/TokenService.java

@@ -300,29 +300,33 @@ public void initRefreshThread() {
      */
     @Override
     public void init() {
-        DateTime now = DateUtil.date();
         List<SysToken> sysTokens = tokenMapper.selectList(new LambdaQueryWrapper<>());
         for (SysToken sysToken : sysTokens) {
-            Integer userId = sysToken.getUserId();
-            dataMap.put(stpLogic.splicingKeyTokenValue(sysToken.getTokenValue()), userId.toString());
-            dataMap.put(stpLogic.splicingKeyLastActiveTime(sysToken.getTokenValue()), StrUtil.toString(now.getTime()));
-            UserDTO userInfo = new UserDTO();
-            User user = new User();
-            user.setId(userId);
-            userInfo.setUser(user);
-            userInfo.setTenantList(Collections.singletonList(tenantMapper.selectById(sysToken.getTenantId())));
-            UserInfoContextHolder.set(userId, userInfo);
-            if (sysToken.getExpireType() == 1) {
-                expireMap.put(stpLogic.splicingKeyTokenValue(sysToken.getTokenValue()), NEVER_EXPIRE);
-            } else {
-                expireMap.put(
-                        stpLogic.splicingKeyTokenValue(sysToken.getTokenValue()),
-                        sysToken.getExpireEndTime().getTime());
-            }
+            injectToken(sysToken);
         }
         initRefreshThread();
     }
 
+    public void injectToken(SysToken sysToken) {
+        DateTime now = DateUtil.date();
+        Integer userId = sysToken.getUserId();
+        dataMap.put(stpLogic.splicingKeyTokenValue(sysToken.getTokenValue()), userId.toString());
+        dataMap.put(stpLogic.splicingKeyLastActiveTime(sysToken.getTokenValue()), StrUtil.toString(now.getTime()));
+        UserDTO userInfo = new UserDTO();
+        User user = new User();
+        user.setId(userId);
+        userInfo.setUser(user);
+        userInfo.setTenantList(Collections.singletonList(tenantMapper.selectById(sysToken.getTenantId())));
+        UserInfoContextHolder.set(userId, userInfo);
+        if (sysToken.getExpireType() == 1) {
+            expireMap.put(stpLogic.splicingKeyTokenValue(sysToken.getTokenValue()), NEVER_EXPIRE);
+        } else {
+            expireMap.put(
+                    stpLogic.splicingKeyTokenValue(sysToken.getTokenValue()),
+                    sysToken.getExpireEndTime().getTime());
+        }
+    }
+
     /**
      * 组件被卸载时，结束定时任务，不再定时清理过期数据
      */