From 1040b69682b06d78284b39a277be63cd67edf961 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 14 Aug 2024 07:59:32 +0800
Subject: [PATCH] 1
---
websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java | 34 +++++++++++++++++++++++++++++-----
1 files changed, 29 insertions(+), 5 deletions(-)
diff --git a/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java b/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java
index e69c348..9638232 100644
--- a/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java
+++ b/websocketSerivce/src/main/java/org/example/util/LoginInterceptor.java
@@ -1,5 +1,13 @@
package org.example.util;
+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;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;
@@ -12,23 +20,39 @@
* @description:
* @create: 2024-07-29 11:15
**/
+@Component
public class LoginInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
throws Exception {
- HttpSession session = request.getSession();
-
// 假设登录状态信息保存在 session 中，这里简单示例，实际项目中可能会有自己的登录状态判断逻辑
- if (session.getAttribute("loggedInUser") != null) {
- // 已登录，允许请求继续
+ String token = null;
+ // 如果是 OPTIONS 请求，我们就让他通过，不管他
+ if (request.getMethod().equals("OPTIONS")) {
+ response.setStatus(HttpServletResponse.SC_OK);
return true;
- } else {
+ }else {
+ token = request.getHeader("token");
+ }
+ if(StringUtils.isEmpty(token)){
// 未登录，返回未授权错误码或重定向到登录页
response.setStatus(HttpServletResponse.SC_UNAUTHORIZED); // 返回 401 未授权状态码
+ response.setCharacterEncoding("UTF-8");
+ response.getWriter().write("请登录");
return false;
}
+ User user = JwtUtil.verify(token);
+ 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;
+ }
+ return true;
}
@Override
--
Gitblit v1.9.3