From ac8522fc973eaedee01d867b9886c807fdc5e945 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Fri, 30 Aug 2024 18:39:40 +0800
Subject: [PATCH] 1

---
 websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java |   18 +++++-------------
 1 files changed, 5 insertions(+), 13 deletions(-)

diff --git a/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java b/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java
index 701c459..9638232 100644
--- a/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java
+++ b/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java
@@ -2,6 +2,7 @@
 
 import cn.hutool.json.JSONUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import org.apache.commons.lang3.StringUtils;
 import org.example.pojo.User;
 import org.example.server.UserService;
 import org.example.server.impl.UserServiceImpl;
@@ -35,7 +36,7 @@
         }else {
             token = request.getHeader("token");
         }
-        if(null == token){
+        if(StringUtils.isEmpty(token)){
             // 未登录,返回未授权错误码或重定向到登录页
             response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);  // 返回 401 未授权状态码
             response.setCharacterEncoding("UTF-8");
@@ -43,24 +44,15 @@
             return false;
         }
         User user = JwtUtil.verify(token);
-        if(null == user){
+        String s = RedisUtil.get(user.getId().toString());
+        if(StringUtils.isEmpty(s) && s.equals(token)){
             // 未登录,返回未授权错误码或重定向到登录页
             response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);  // 返回 401 未授权状态码
             response.setCharacterEncoding("UTF-8");
             response.getWriter().write("请登录");
             return false;
         }
-
-        if (user.getIsRoot() == 1) {
-            // 已登录,允许请求继续
-            return true;
-        } else {
-            // 未登录,返回未授权错误码或重定向到登录页
-            response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);  // 返回 401 未授权状态码
-            response.setCharacterEncoding("UTF-8");
-            response.getWriter().write("请登录");
-            return false;
-        }
+        return true;
     }
 
     @Override

--
Gitblit v1.9.3