publish 2.0.0.rc4

2024-07-11 15:21:59 +08:00
parent fb1f7c4e37
commit 5da83bb477
33 changed files with 240 additions and 250 deletions
--- a/springboot-login/src/main/java/com/tiesheng/login/config/TokenWebMvcConfigurer.java
+++ b/springboot-login/src/main/java/com/tiesheng/login/config/TokenWebMvcConfigurer.java
@@ -0,0 +1,92 @@
+package com.tiesheng.login.config;
+
+import cn.hutool.core.util.ObjUtil;
+import cn.hutool.core.util.StrUtil;
+import com.tiesheng.annotation.token.TokenIgnore;
+import com.tiesheng.util.config.TsTokenConfig;
+import com.tiesheng.util.pojos.TokenBean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.MethodParameter;
+import org.springframework.web.bind.support.WebDataBinderFactory;
+import org.springframework.web.context.request.NativeWebRequest;
+import org.springframework.web.method.HandlerMethod;
+import org.springframework.web.method.support.HandlerMethodArgumentResolver;
+import org.springframework.web.method.support.ModelAndViewContainer;
+import org.springframework.web.servlet.HandlerInterceptor;
+import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.List;
+
+
+@Configuration
+public class TokenWebMvcConfigurer implements WebMvcConfigurer {
+
+    @Resource
+    TsTokenConfig tsTokenConfig;
+
+    @Override
+    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
+        resolvers.add(new HandlerMethodArgumentResolver() {
+            @Override
+            public boolean supportsParameter(MethodParameter parameter) {
+                return parameter.getParameterType().isAssignableFrom(TokenBean.class);
+            }
+
+            @Override
+            public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
+                                          NativeWebRequest webRequest, WebDataBinderFactory binderFactory) {
+                String header = webRequest.getHeader(TsTokenConfig.TOKEN_KEY);
+                return tsTokenConfig.validToken(header, true);
+            }
+        });
+    }
+
+
+    @Override
+    public void addInterceptors(InterceptorRegistry registry) {
+        registry.addInterceptor(new HandlerInterceptor() {
+            @Override
+            public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
+
+                String requestURI = request.getRequestURI();
+                String[] ignorePaths = tsTokenConfig.getIgnorePaths();
+                if (ObjUtil.isNotEmpty(ignorePaths)) {
+                    for (String path : ignorePaths) {
+                        if (path.contains("/**") &&
+                                StrUtil.startWith(requestURI, path.replace("/**", ""))) {
+                            // 通配路径
+                            return true;
+                        } else if (requestURI.equals(path)) {
+                            // 完整路径
+                            return true;
+                        }
+                    }
+                }
+
+                // 过滤不要需要验证的接口（注解）
+                HandlerMethod handlerMethod = (HandlerMethod) handler;
+                TokenIgnore annotation = handlerMethod.getBeanType().getAnnotation(TokenIgnore.class);
+                if (annotation != null) {
+                    return true;
+                }
+                TokenIgnore apiTokenIgnore = handlerMethod.getMethodAnnotation(TokenIgnore.class);
+                if (apiTokenIgnore != null) {
+                    return true;
+                }
+
+                // token验证
+                tsTokenConfig.validToken(request, true);
+
+                return true;
+            }
+        });
+    }
+}
+
+
+
+