删除无用代码

chenhj

8 天以前 c4174bbb103d95d6cd4714d3cc3070559444d70e

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package com.ruoyi.framework.config.properties; 
 
import com.ruoyi.framework.aspectj.lang.annotation.Anonymous; 
import org.apache.commons.lang3.RegExUtils; 
import org.springframework.beans.BeansException; 
import org.springframework.beans.factory.InitializingBean; 
import org.springframework.context.ApplicationContext; 
import org.springframework.context.ApplicationContextAware; 
import org.springframework.context.annotation.Configuration; 
import org.springframework.core.annotation.AnnotationUtils; 
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.*; 
import java.util.regex.Pattern; 
 
/** 
 * 设置Anonymous注解允许匿名访问的url 
 *  
 * @author ruoyi 
 */ 
@Configuration 
public class PermitAllUrlProperties implements InitializingBean, ApplicationContextAware 
{ 
    private static final Pattern PATTERN = Pattern.compile("\\{(.*?)\\}"); 
 
    private ApplicationContext applicationContext; 
 
    private List<String> urls = new ArrayList<>(); 
 
    public String ASTERISK = "*"; 
 
    @Override 
    public void afterPropertiesSet() 
    { 
        RequestMappingHandlerMapping mapping = applicationContext.getBean(RequestMappingHandlerMapping.class); 
        Map<RequestMappingInfo, HandlerMethod> map = mapping.getHandlerMethods(); 
 
//        map.keySet().forEach(info -> { 
//            HandlerMethod handlerMethod = map.get(info); 
// 
//            // 获取方法上边的注解 替代path variable 为 * 
//            Anonymous method = AnnotationUtils.findAnnotation(handlerMethod.getMethod(), Anonymous.class); 
//            Optional.ofNullable(method).ifPresent(anonymous -> Objects.requireNonNull(info.getPatternsCondition().getPatterns()) 
//                    .forEach(url -> urls.add(RegExUtils.replaceAll(url, PATTERN, ASTERISK)))); 
// 
//            // 获取类上边的注解, 替代path variable 为 * 
//            Anonymous controller = AnnotationUtils.findAnnotation(handlerMethod.getBeanType(), Anonymous.class); 
//            Optional.ofNullable(controller).ifPresent(anonymous -> Objects.requireNonNull(info.getPatternsCondition().getPatterns()) 
//                    .forEach(url -> urls.add(RegExUtils.replaceAll(url, PATTERN, ASTERISK)))); 
//        }); 
 
        map.keySet().forEach(info -> { 
            HandlerMethod handlerMethod = map.get(info); 
            // 获取方法上的匿名访问注解 
            Anonymous anonymous = AnnotationUtils.findAnnotation(handlerMethod.getMethod(), Anonymous.class); 
            if (Objects.nonNull(anonymous)) { 
                // 重点：Spring Boot 3 兼容性写法 
                Set<String> patterns = new HashSet<>(); 
                if (info.getPatternsCondition() != null) { 
                    patterns.addAll(info.getPatternsCondition().getPatterns()); 
                } 
                if (info.getPathPatternsCondition() != null) { 
                    patterns.addAll(info.getPathPatternsCondition().getPatternValues()); 
                } 
 
                patterns.forEach(url -> urls.add(RegExUtils.replaceAll(url, PATTERN, "*"))); 
            } 
        }); 
    } 
 
    @Override 
    public void setApplicationContext(ApplicationContext context) throws BeansException 
    { 
        this.applicationContext = context; 
    } 
 
    public List<String> getUrls() 
    { 
        return urls; 
    } 
 
    public void setUrls(List<String> urls) 
    { 
        this.urls = urls; 
    } 
}