bb32fc1fe7
- 新增 AnonymousAccess 注解用于标记支持匿名访问的方法 - 添加 AnonymousGetMapping、AnonymousPostMapping 等注解用于具体 HTTP 方法 - 实现 AnonymousUtils 工具类以获取所有匿名访问 URL - 修改 SecurityConfig 配置类，支持细粒度的匿名访问控制- 更新 LogAspect 切面，增加对匿名访问的处理
92 lines
3.8 KiB
Java
92 lines
3.8 KiB
Java
package com.youlai.boot.common.util;
|
||
|
||
import com.youlai.boot.common.annotation.AnonymousAccess;
|
||
import com.youlai.boot.common.enums.RequestMethodEnum;
|
||
import org.springframework.context.ApplicationContext;
|
||
import org.springframework.web.bind.annotation.RequestMethod;
|
||
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.stream.Collectors;
|
||
|
||
public class AnonymousUtils {
|
||
/**
|
||
* 获取所有匿名标记URL，不区分请求方式
|
||
*/
|
||
public static Set<String> getAnonymousUrls(ApplicationContext applicationContext) {
|
||
return getAllAnonymousUrls(applicationContext).values().stream().flatMap(Collection::stream).collect(Collectors.toSet());
|
||
}
|
||
|
||
/**
|
||
* 获取所有被标记的匿名类集合
|
||
*
|
||
* @return /
|
||
*/
|
||
public static Map<String, Set<String>> getAllAnonymousUrls(ApplicationContext applicationContext) {
|
||
// 搜索匿名标记
|
||
RequestMappingHandlerMapping requestMappingHandlerMapping = (RequestMappingHandlerMapping) applicationContext.getBean("requestMappingHandlerMapping");
|
||
Map<RequestMappingInfo, HandlerMethod> handlerMethodMap = requestMappingHandlerMapping.getHandlerMethods();
|
||
|
||
// 获取所以被标记的匿名类集合
|
||
return getAnonymousUrl(handlerMethodMap);
|
||
}
|
||
|
||
/**
|
||
* 获取所有被标记的匿名类集合
|
||
*
|
||
* @param handlerMethodMap 请求映射信息集合
|
||
* @return /
|
||
*/
|
||
public static Map<String, Set<String>> getAnonymousUrl(Map<RequestMappingInfo, HandlerMethod> handlerMethodMap) {
|
||
Set<String> get = new HashSet<>();
|
||
Set<String> post = new HashSet<>();
|
||
Set<String> put = new HashSet<>();
|
||
Set<String> patch = new HashSet<>();
|
||
Set<String> delete = new HashSet<>();
|
||
Set<String> all = new HashSet<>();
|
||
|
||
handlerMethodMap.forEach((key, value) -> {
|
||
AnonymousAccess anonymousAccess = value.getMethodAnnotation(AnonymousAccess.class);
|
||
if (anonymousAccess != null) {
|
||
ArrayList<RequestMethod> requestMethods = new ArrayList<>(key.getMethodsCondition().getMethods());
|
||
RequestMethodEnum request = RequestMethodEnum.find(requestMethods.isEmpty() ? RequestMethodEnum.ALL.getType() : requestMethods.get(0).name());
|
||
switch (Objects.requireNonNull(request)) {
|
||
case GET:
|
||
get.addAll(key.getDirectPaths());
|
||
break;
|
||
case POST:
|
||
post.addAll(key.getDirectPaths());
|
||
break;
|
||
case PUT:
|
||
put.addAll(key.getDirectPaths());
|
||
break;
|
||
case PATCH:
|
||
patch.addAll(key.getDirectPaths());
|
||
break;
|
||
case DELETE:
|
||
delete.addAll(key.getDirectPaths());
|
||
break;
|
||
default:
|
||
all.addAll(key.getDirectPaths());
|
||
}
|
||
}
|
||
});
|
||
|
||
return Map.ofEntries(
|
||
entry(RequestMethodEnum.GET.getType(), get),
|
||
entry(RequestMethodEnum.POST.getType(), post),
|
||
entry(RequestMethodEnum.PUT.getType(), put),
|
||
entry(RequestMethodEnum.PATCH.getType(), patch),
|
||
entry(RequestMethodEnum.DELETE.getType(), delete),
|
||
entry(RequestMethodEnum.ALL.getType(), all)
|
||
);
|
||
}
|
||
|
||
public static Map.Entry<String, Set<String>> entry(String key, Collection<String> collection) {
|
||
return Map.entry(key, collection.stream().filter(it -> !it.isEmpty()).collect(Collectors.toUnmodifiableSet()));
|
||
}
|
||
|
||
}
|