update 优化 只拦截系统内存在的路径 减少不必要的拦截造成的性能消耗

This commit is contained in:
疯狂的狮子li 2023-02-28 19:13:51 +08:00
parent 4bfdf357a2
commit 7c5d645639
4 changed files with 51 additions and 2 deletions

View File

@ -127,8 +127,10 @@ security:
- /**/*.html
- /**/*.css
- /**/*.js
# swagger 文档配置
# 公共路径
- /favicon.ico
- /error
# swagger 文档配置
- /*/api-docs
- /*/api-docs/**
# actuator 监控配置

View File

@ -3,7 +3,9 @@ package com.ruoyi.common.security.config;
import cn.dev33.satoken.interceptor.SaInterceptor;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import com.ruoyi.common.core.utils.SpringUtils;
import com.ruoyi.common.security.config.properties.SecurityProperties;
import com.ruoyi.common.security.handler.AllUrlHandler;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.autoconfigure.AutoConfiguration;
@ -32,10 +34,11 @@ public class SecurityConfig implements WebMvcConfigurer {
public void addInterceptors(InterceptorRegistry registry) {
// 注册路由拦截器自定义验证规则
registry.addInterceptor(new SaInterceptor(handler -> {
AllUrlHandler allUrlHandler = SpringUtils.getBean(AllUrlHandler.class);
// 登录验证 -- 排除多个路径
SaRouter
// 获取所有的
.match("/**")
.match(allUrlHandler.getUrls())
// 对未排除的路径进行检查
.check(() -> {
// 检查是否登录 是否有token

View File

@ -0,0 +1,43 @@
package com.ruoyi.common.security.handler;
import cn.hutool.core.util.ReUtil;
import com.ruoyi.common.core.utils.SpringUtils;
import lombok.Data;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.regex.Pattern;
/**
* 获取所有Url配置
*
* @author Lion Li
*/
@Data
public class AllUrlHandler implements InitializingBean {
private static final Pattern PATTERN = Pattern.compile("\\{(.*?)\\}");
private List<String> urls = new ArrayList<>();
@Override
public void afterPropertiesSet() {
RequestMappingHandlerMapping mapping = SpringUtils.getBean("requestMappingHandlerMapping", RequestMappingHandlerMapping.class);
Map<RequestMappingInfo, HandlerMethod> map = mapping.getHandlerMethods();
map.keySet().forEach(info -> {
// 获取方法上边的注解 替代path variable *
Objects.requireNonNull(info.getPathPatternsCondition().getPatterns())
.forEach(url -> urls.add(ReUtil.replaceAll(url.getPatternString(), PATTERN, "*")));
// 获取类上边的注解, 替代path variable *
Objects.requireNonNull(info.getPathPatternsCondition().getPatterns())
.forEach(url -> urls.add(ReUtil.replaceAll(url.getPatternString(), PATTERN, "*")));
});
}
}

View File

@ -1,2 +1,3 @@
com.ruoyi.common.security.handler.GlobalExceptionHandler
com.ruoyi.common.security.handler.AllUrlHandler
com.ruoyi.common.security.config.SecurityConfig