浏览代码

统一rediskey

1018653686@qq.com 1 年之前
父节点
当前提交
aded1d58c7

+ 2 - 1
hx-xmhjc/src/main/java/com/fjhx/xmhjc/aspect/LoginValidAspect.java

@@ -7,6 +7,7 @@ import com.fjhx.xmhjc.anno.LoginValid;
 import com.fjhx.xmhjc.constants.LoginConstant;
 import com.fjhx.xmhjc.entity.website.po.WebsiteUsers;
 import com.fjhx.xmhjc.exception.LoginException;
+import com.fjhx.xmhjc.utils.WebsiteUserUtil;
 import com.ruoyi.common.core.redis.RedisCache;
 import com.ruoyi.common.utils.ServletUtils;
 import com.ruoyi.common.utils.spring.SpringUtils;
@@ -45,7 +46,7 @@ public class LoginValidAspect {
         if (StrUtil.isBlank(token)) {
             throw new LoginException("请先登录");
         }
-        WebsiteUsers websiteUsers = redisCache.getCacheObject(LoginConstant.TOKEN_PREFIX+token);
+        WebsiteUsers websiteUsers = redisCache.getCacheObject(WebsiteUserUtil.buildRealKey(token));
         if (ObjectUtil.isNull(websiteUsers)) {
             throw new LoginException("请先登录");
         }

+ 3 - 4
hx-xmhjc/src/main/java/com/fjhx/xmhjc/controller/open/OpenWesiteUserController.java

@@ -69,7 +69,7 @@ public class OpenWesiteUserController {
             throw new ServiceException("密码错误");
         }
         String token = IdUtil.fastSimpleUUID();
-        redisCache.setCacheObject(LoginConstant.TOKEN_PREFIX+token, websiteUsers, LoginConstant.LOGIN_EXPIRED, TimeUnit.HOURS);
+        redisCache.setCacheObject(WebsiteUserUtil.buildRealKey(token), websiteUsers, LoginConstant.LOGIN_EXPIRED, TimeUnit.HOURS);
         return token;
     }
 
@@ -83,7 +83,7 @@ public class OpenWesiteUserController {
     @PostMapping("/logout")
     public void logout(){
         String websiteUserToken = WebsiteUserUtil.getWebsiteUserToken();
-        redisCache.deleteObject(LoginConstant.TOKEN_PREFIX+websiteUserToken);
+        redisCache.deleteObject(WebsiteUserUtil.buildRealKey(websiteUserToken));
     }
 
     /**
@@ -94,8 +94,7 @@ public class OpenWesiteUserController {
     @LoginValid
     @PostMapping("/info")
     public UserInfoVo info(){
-        String websiteUserToken = WebsiteUserUtil.getWebsiteUserToken();
-        WebsiteUsers websiteUsers = redisCache.getCacheObject(LoginConstant.TOKEN_PREFIX+websiteUserToken);
+        WebsiteUsers websiteUsers = WebsiteUserUtil.getLoginWebsiteUser();
         return BeanUtil.copyProperties(websiteUsers, UserInfoVo.class);
     }
 

+ 5 - 1
hx-xmhjc/src/main/java/com/fjhx/xmhjc/utils/WebsiteUserUtil.java

@@ -20,7 +20,11 @@ public class WebsiteUserUtil {
 
 
     public static WebsiteUsers getLoginWebsiteUser() {
-        WebsiteUsers websiteUsers = redisCache.getCacheObject(LoginConstant.TOKEN_PREFIX+getWebsiteUserToken());
+        WebsiteUsers websiteUsers = redisCache.getCacheObject(buildRealKey(getWebsiteUserToken()));
         return websiteUsers;
     }
+
+    public static String buildRealKey(String token){
+        return LoginConstant.TOKEN_PREFIX+token;
+    }
 }