|
|
@@ -3,6 +3,7 @@ package cn.iocoder.yudao.framework.tenant.config;
|
|
|
import cn.iocoder.yudao.framework.common.enums.WebFilterOrderEnum;
|
|
|
import cn.iocoder.yudao.framework.mybatis.core.util.MyBatisUtils;
|
|
|
import cn.iocoder.yudao.framework.redis.config.YudaoCacheProperties;
|
|
|
+import cn.iocoder.yudao.framework.tenant.core.aop.TenantIgnore;
|
|
|
import cn.iocoder.yudao.framework.tenant.core.aop.TenantIgnoreAspect;
|
|
|
import cn.iocoder.yudao.framework.tenant.core.db.TenantDatabaseInterceptor;
|
|
|
import cn.iocoder.yudao.framework.tenant.core.job.TenantJobAspect;
|
|
|
@@ -19,11 +20,13 @@ import cn.iocoder.yudao.framework.web.core.handler.GlobalExceptionHandler;
|
|
|
import cn.iocoder.yudao.module.system.api.tenant.TenantApi;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
|
|
|
+import jakarta.annotation.Resource;
|
|
|
import org.springframework.boot.autoconfigure.AutoConfiguration;
|
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
|
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
|
|
import org.springframework.boot.web.servlet.FilterRegistrationBean;
|
|
|
+import org.springframework.context.ApplicationContext;
|
|
|
import org.springframework.context.annotation.Bean;
|
|
|
import org.springframework.context.annotation.Primary;
|
|
|
import org.springframework.data.redis.cache.BatchStrategies;
|
|
|
@@ -32,14 +35,24 @@ import org.springframework.data.redis.cache.RedisCacheManager;
|
|
|
import org.springframework.data.redis.cache.RedisCacheWriter;
|
|
|
import org.springframework.data.redis.connection.RedisConnectionFactory;
|
|
|
import org.springframework.data.redis.core.RedisTemplate;
|
|
|
+import org.springframework.web.method.HandlerMethod;
|
|
|
+import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
|
|
|
+import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;
|
|
|
+import org.springframework.web.util.pattern.PathPattern;
|
|
|
|
|
|
+import java.util.Map;
|
|
|
import java.util.Objects;
|
|
|
|
|
|
+import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
|
|
|
+
|
|
|
@AutoConfiguration
|
|
|
@ConditionalOnProperty(prefix = "yudao.tenant", value = "enable", matchIfMissing = true) // 允许使用 yudao.tenant.enable=false 禁用多租户
|
|
|
@EnableConfigurationProperties(TenantProperties.class)
|
|
|
public class YudaoTenantAutoConfiguration {
|
|
|
|
|
|
+ @Resource
|
|
|
+ private ApplicationContext applicationContext;
|
|
|
+
|
|
|
@Bean
|
|
|
public TenantFrameworkService tenantFrameworkService(TenantApi tenantApi) {
|
|
|
return new TenantFrameworkServiceImpl(tenantApi);
|
|
|
@@ -67,13 +80,41 @@ public class YudaoTenantAutoConfiguration {
|
|
|
// ========== WEB ==========
|
|
|
|
|
|
@Bean
|
|
|
- public FilterRegistrationBean<TenantContextWebFilter> tenantContextWebFilter() {
|
|
|
+ public FilterRegistrationBean<TenantContextWebFilter> tenantContextWebFilter(TenantProperties tenantProperties) {
|
|
|
FilterRegistrationBean<TenantContextWebFilter> registrationBean = new FilterRegistrationBean<>();
|
|
|
registrationBean.setFilter(new TenantContextWebFilter());
|
|
|
registrationBean.setOrder(WebFilterOrderEnum.TENANT_CONTEXT_FILTER);
|
|
|
+ addIgnoreUrls(tenantProperties);
|
|
|
return registrationBean;
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 如果 Controller 接口上，有 {@link TenantIgnore} 注解，那么添加到忽略的 URL 中
|
|
|
+ *
|
|
|
+ * @param tenantProperties 租户配置
|
|
|
+ */
|
|
|
+ private void addIgnoreUrls(TenantProperties tenantProperties) {
|
|
|
+ // 获得接口对应的 HandlerMethod 集合
|
|
|
+ RequestMappingHandlerMapping requestMappingHandlerMapping = (RequestMappingHandlerMapping)
|
|
|
+ applicationContext.getBean("requestMappingHandlerMapping");
|
|
|
+ Map<RequestMappingInfo, HandlerMethod> handlerMethodMap = requestMappingHandlerMapping.getHandlerMethods();
|
|
|
+ // 获得有 @TenantIgnore 注解的接口
|
|
|
+ for (Map.Entry<RequestMappingInfo, HandlerMethod> entry : handlerMethodMap.entrySet()) {
|
|
|
+ HandlerMethod handlerMethod = entry.getValue();
|
|
|
+ if (!handlerMethod.hasMethodAnnotation(TenantIgnore.class)) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+ // 添加到忽略的 URL 中
|
|
|
+ if (entry.getKey().getPatternsCondition() != null) {
|
|
|
+ tenantProperties.getIgnoreUrls().addAll(entry.getKey().getPatternsCondition().getPatterns());
|
|
|
+ }
|
|
|
+ if (entry.getKey().getPathPatternsCondition() != null) {
|
|
|
+ tenantProperties.getIgnoreUrls().addAll(
|
|
|
+ convertList(entry.getKey().getPathPatternsCondition().getPatterns(), PathPattern::getPatternString));
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
// ========== Security ==========
|
|
|
|
|
|
@Bean
|
YunaiV