项目升级到 Spring Boot 4
RuoYi
3 months ago
parent
3fa5746ae7
commit
c4c1ae4e55
@ -1,151 +0,0 @@
|
|||||||
package com.ruoyi.framework.config.properties;
|
|
||||||
|
|
||||||
import java.lang.reflect.Method;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Objects;
|
|
||||||
import org.springframework.aop.framework.Advised;
|
|
||||||
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.stereotype.Controller;
|
|
||||||
import org.springframework.web.bind.annotation.DeleteMapping;
|
|
||||||
import org.springframework.web.bind.annotation.GetMapping;
|
|
||||||
import org.springframework.web.bind.annotation.PostMapping;
|
|
||||||
import org.springframework.web.bind.annotation.PutMapping;
|
|
||||||
import org.springframework.web.bind.annotation.RequestMapping;
|
|
||||||
import com.ruoyi.common.annotation.Anonymous;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 设置Anonymous注解允许匿名访问的url
|
|
||||||
*
|
|
||||||
* @author ruoyi
|
|
||||||
*/
|
|
||||||
@Configuration
|
|
||||||
public class PermitAllUrlProperties implements InitializingBean, ApplicationContextAware
|
|
||||||
{
|
|
||||||
private List<String> urls = new ArrayList<>();
|
|
||||||
|
|
||||||
private ApplicationContext applicationContext;
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void afterPropertiesSet() throws Exception
|
|
||||||
{
|
|
||||||
Map<String, Object> controllers = applicationContext.getBeansWithAnnotation(Controller.class);
|
|
||||||
for (Object bean : controllers.values())
|
|
||||||
{
|
|
||||||
Class<?> beanClass;
|
|
||||||
if (bean instanceof Advised)
|
|
||||||
{
|
|
||||||
beanClass = ((Advised) bean).getTargetSource().getTarget().getClass();
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
beanClass = bean.getClass();
|
|
||||||
}
|
|
||||||
// 处理类级别的匿名访问注解
|
|
||||||
if (beanClass.isAnnotationPresent(Anonymous.class))
|
|
||||||
{
|
|
||||||
RequestMapping baseMapping = beanClass.getAnnotation(RequestMapping.class);
|
|
||||||
if (Objects.nonNull(baseMapping))
|
|
||||||
{
|
|
||||||
String[] baseUrl = baseMapping.value();
|
|
||||||
for (String url : baseUrl)
|
|
||||||
{
|
|
||||||
urls.add(prefix(url) + "/*");
|
|
||||||
}
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理方法级别的匿名访问注解
|
|
||||||
Method[] methods = beanClass.getDeclaredMethods();
|
|
||||||
for (Method method : methods)
|
|
||||||
{
|
|
||||||
if (method.isAnnotationPresent(Anonymous.class))
|
|
||||||
{
|
|
||||||
RequestMapping baseMapping = beanClass.getAnnotation(RequestMapping.class);
|
|
||||||
String[] baseUrl = {};
|
|
||||||
if (Objects.nonNull(baseMapping))
|
|
||||||
{
|
|
||||||
baseUrl = baseMapping.value();
|
|
||||||
}
|
|
||||||
if (method.isAnnotationPresent(RequestMapping.class))
|
|
||||||
{
|
|
||||||
RequestMapping requestMapping = method.getAnnotation(RequestMapping.class);
|
|
||||||
String[] uri = requestMapping.value();
|
|
||||||
urls.addAll(rebuildUrl(baseUrl, uri));
|
|
||||||
}
|
|
||||||
else if (method.isAnnotationPresent(GetMapping.class))
|
|
||||||
{
|
|
||||||
GetMapping requestMapping = method.getAnnotation(GetMapping.class);
|
|
||||||
String[] uri = requestMapping.value();
|
|
||||||
urls.addAll(rebuildUrl(baseUrl, uri));
|
|
||||||
}
|
|
||||||
else if (method.isAnnotationPresent(PostMapping.class))
|
|
||||||
{
|
|
||||||
PostMapping requestMapping = method.getAnnotation(PostMapping.class);
|
|
||||||
String[] uri = requestMapping.value();
|
|
||||||
urls.addAll(rebuildUrl(baseUrl, uri));
|
|
||||||
}
|
|
||||||
else if (method.isAnnotationPresent(PutMapping.class))
|
|
||||||
{
|
|
||||||
PutMapping requestMapping = method.getAnnotation(PutMapping.class);
|
|
||||||
String[] uri = requestMapping.value();
|
|
||||||
urls.addAll(rebuildUrl(baseUrl, uri));
|
|
||||||
}
|
|
||||||
else if (method.isAnnotationPresent(DeleteMapping.class))
|
|
||||||
{
|
|
||||||
DeleteMapping requestMapping = method.getAnnotation(DeleteMapping.class);
|
|
||||||
String[] uri = requestMapping.value();
|
|
||||||
urls.addAll(rebuildUrl(baseUrl, uri));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private List<String> rebuildUrl(String[] bases, String[] uris)
|
|
||||||
{
|
|
||||||
List<String> urls = new ArrayList<>();
|
|
||||||
for (String base : bases)
|
|
||||||
{
|
|
||||||
if (uris.length > 0)
|
|
||||||
{
|
|
||||||
for (String uri : uris)
|
|
||||||
{
|
|
||||||
urls.add(prefix(base) + prefix(uri));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
urls.add(prefix(base));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return urls;
|
|
||||||
}
|
|
||||||
|
|
||||||
private String prefix(String seg)
|
|
||||||
{
|
|
||||||
return seg.startsWith("/") ? seg : "/" + seg;
|
|
||||||
}
|
|
||||||
|
|
||||||
@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;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue