From 93fe0b66e85e95e3a7e77a28504d6d289b6906b9 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Fri, 08 Oct 2021 13:11:22 +0800
Subject: [PATCH] screen-manage screen-api      更改拦截器逻辑,记录token无效的请求IP以及URL。      manage记录到数据库日志中      api记录在日志文件中

---
 screen-api/src/main/java/com/moral/api/interceptor/WebInterceptor.java |   23 +++++++++++++++++++++--
 1 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/interceptor/WebInterceptor.java b/screen-api/src/main/java/com/moral/api/interceptor/WebInterceptor.java
index b0d6ed1..afe734b 100644
--- a/screen-api/src/main/java/com/moral/api/interceptor/WebInterceptor.java
+++ b/screen-api/src/main/java/com/moral/api/interceptor/WebInterceptor.java
@@ -3,13 +3,20 @@
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
+import com.alibaba.fastjson.JSON;
+import com.moral.constant.ResponseCodeEnum;
+import com.moral.constant.ResultMessage;
+import com.moral.util.WebUtils;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Component;
 import org.springframework.web.method.HandlerMethod;
 import org.springframework.web.servlet.HandlerInterceptor;
 
 import com.moral.util.TokenUtils;
 
-@Component
+import java.io.PrintWriter;
+
+@Slf4j
 public class WebInterceptor implements HandlerInterceptor {
 
     @Override
@@ -18,7 +25,19 @@
             return true;
         }*/
         String token = request.getHeader("token");
-        TokenUtils.checkToken(token);
+        boolean result = TokenUtils.checkToken(token);
+        if (!result) {
+            ResultMessage resultMessage = ResultMessage.fail(ResponseCodeEnum.TOKEN_INVALID.getCode(),
+                    ResponseCodeEnum.TOKEN_INVALID.getMsg());
+            response.setCharacterEncoding("UTF-8");
+            response.setContentType("text/html; charset=utf-8");
+            PrintWriter writer = response.getWriter();
+            writer.print(JSON.toJSONString(resultMessage));
+            log.info("token������������");
+            log.info("ip:"+ WebUtils.getIpAddr(request));
+            log.info("������URL���"+request.getRequestURL());
+            return false;
+        }
         //TokenUtils.extendTokenTime(token);
         return true;
     }

--
Gitblit v1.8.0