From a8394df1176a9444a5cf6f84dac535f842e8c1b5 Mon Sep 17 00:00:00 2001
From: xiejun <xiejun@vci-tech.com>
Date: 星期三, 08 一月 2025 18:36:21 +0800
Subject: [PATCH] 产品型号集成，类型转换，人员组织加日志，申请接口引用码段值校验
---
Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/config/TenantDataSourceConfiguration.java | 179 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 179 insertions(+), 0 deletions(-)
diff --git a/Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/config/TenantDataSourceConfiguration.java b/Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/config/TenantDataSourceConfiguration.java
new file mode 100644
index 0000000..f3f8be8
--- /dev/null
+++ b/Source/BladeX-Tool/blade-starter-tenant/src/main/java/org/springblade/core/tenant/config/TenantDataSourceConfiguration.java
@@ -0,0 +1,179 @@
+/*
+ * Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * Neither the name of the dreamlu.net developer nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ * Author: Chill 搴勯獮 (smallchill@163.com)
+ */
+package org.springblade.core.tenant.config;
+
+import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure;
+import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
+import com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationAdvisor;
+import com.baomidou.dynamic.datasource.creator.DruidDataSourceCreator;
+import com.baomidou.dynamic.datasource.processor.DsHeaderProcessor;
+import com.baomidou.dynamic.datasource.processor.DsProcessor;
+import com.baomidou.dynamic.datasource.processor.DsSessionProcessor;
+import com.baomidou.dynamic.datasource.processor.DsSpelExpressionProcessor;
+import com.baomidou.dynamic.datasource.provider.DynamicDataSourceProvider;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceAutoConfiguration;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceCreatorAutoConfiguration;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
+import lombok.AllArgsConstructor;
+import lombok.RequiredArgsConstructor;
+import org.springblade.core.tenant.dynamic.*;
+import org.springframework.beans.factory.SmartInitializingSingleton;
+import org.springframework.beans.factory.config.BeanDefinition;
+import org.springframework.boot.autoconfigure.AutoConfiguration;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Import;
+import org.springframework.context.annotation.Primary;
+import org.springframework.context.annotation.Role;
+import org.springframework.core.annotation.Order;
+import org.springframework.jdbc.core.JdbcTemplate;
+
+import javax.sql.DataSource;
+
+import static org.springblade.core.tenant.constant.TenantBaseConstant.TENANT_DYNAMIC_DATASOURCE_PROP;
+import static org.springblade.core.tenant.constant.TenantBaseConstant.TENANT_DYNAMIC_GLOBAL_PROP;
+
+/**
+ * 澶氱鎴锋暟鎹簮閰嶇疆绫�
+ *
+ * @author Chill
+ */
+@RequiredArgsConstructor
+@EnableConfigurationProperties({DataSourceProperties.class, DynamicDataSourceProperties.class})
+@AutoConfiguration(before = {DruidDataSourceAutoConfigure.class, DynamicDataSourceAutoConfiguration.class})
+@Import(value = {DynamicDataSourceCreatorAutoConfiguration.class})
+@ConditionalOnProperty(value = TENANT_DYNAMIC_DATASOURCE_PROP, havingValue = "true")
+public class TenantDataSourceConfiguration {
+
+ @Bean
+ @Primary
+ public DynamicDataSourceProvider dynamicDataSourceProvider(DataSourceProperties dataSourceProperties, DynamicDataSourceProperties dynamicDataSourceProperties) {
+ String driverClassName = dataSourceProperties.getDriverClassName();
+ String url = dataSourceProperties.getUrl();
+ String username = dataSourceProperties.getUsername();
+ String password = dataSourceProperties.getPassword();
+ DataSourceProperty master = dynamicDataSourceProperties.getDatasource().get(dynamicDataSourceProperties.getPrimary());
+ if (master != null) {
+ driverClassName = master.getDriverClassName();
+ url = master.getUrl();
+ username = master.getUsername();
+ password = master.getPassword();
+ }
+ return new TenantDataSourceJdbcProvider(dynamicDataSourceProperties, driverClassName, url, username, password);
+ }
+
+ @Bean
+ @Primary
+ public DataSource dataSource(DynamicDataSourceProvider dynamicDataSourceProvider, DynamicDataSourceProperties dynamicDataSourceProperties) {
+ DynamicRoutingDataSource dataSource = new DynamicRoutingDataSource();
+ dataSource.setPrimary(dynamicDataSourceProperties.getPrimary());
+ dataSource.setStrict(dynamicDataSourceProperties.getStrict());
+ dataSource.setStrategy(dynamicDataSourceProperties.getStrategy());
+ dataSource.setProvider(dynamicDataSourceProvider);
+ dataSource.setP6spy(dynamicDataSourceProperties.getP6spy());
+ dataSource.setSeata(dynamicDataSourceProperties.getSeata());
+ return dataSource;
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ public TenantDataSourceAnnotationInterceptor tenantDataSourceAnnotationInterceptor(DsProcessor dsProcessor, DynamicDataSourceProperties dynamicDataSourceProperties) {
+ return new TenantDataSourceAnnotationInterceptor(dynamicDataSourceProperties.isAllowedPublicOnly(), dsProcessor);
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ @Role(value = BeanDefinition.ROLE_INFRASTRUCTURE)
+ public DynamicDataSourceAnnotationAdvisor dynamicDatasourceAnnotationAdvisor(TenantDataSourceAnnotationInterceptor tenantDataSourceAnnotationInterceptor, DynamicDataSourceProperties dynamicDataSourceProperties) {
+ DynamicDataSourceAnnotationAdvisor advisor = new DynamicDataSourceAnnotationAdvisor(tenantDataSourceAnnotationInterceptor);
+ advisor.setOrder(dynamicDataSourceProperties.getOrder());
+ return advisor;
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ @ConditionalOnProperty(value = TENANT_DYNAMIC_GLOBAL_PROP, havingValue = "true")
+ public TenantDataSourceGlobalInterceptor tenantDataSourceGlobalInterceptor() {
+ return new TenantDataSourceGlobalInterceptor();
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ @Role(value = BeanDefinition.ROLE_INFRASTRUCTURE)
+ @ConditionalOnProperty(value = TENANT_DYNAMIC_GLOBAL_PROP, havingValue = "true")
+ public TenantDataSourceGlobalAdvisor tenantDataSourceGlobalAdvisor(TenantDataSourceGlobalInterceptor tenantDataSourceGlobalInterceptor, DynamicDataSourceProperties dynamicDataSourceProperties) {
+ TenantDataSourceGlobalAdvisor advisor = new TenantDataSourceGlobalAdvisor(tenantDataSourceGlobalInterceptor);
+ advisor.setOrder(dynamicDataSourceProperties.getOrder() + 1);
+ return advisor;
+ }
+
+ @Bean
+ @ConditionalOnMissingBean
+ public DsProcessor dsProcessor() {
+ DsHeaderProcessor headerProcessor = new DsHeaderProcessor();
+ DsSessionProcessor sessionProcessor = new DsSessionProcessor();
+ DsTenantIdProcessor tenantIdProcessor = new DsTenantIdProcessor();
+ DsSpelExpressionProcessor spelExpressionProcessor = new DsSpelExpressionProcessor();
+ headerProcessor.setNextProcessor(sessionProcessor);
+ sessionProcessor.setNextProcessor(tenantIdProcessor);
+ tenantIdProcessor.setNextProcessor(spelExpressionProcessor);
+ return headerProcessor;
+ }
+
+ @Order
+ @AutoConfiguration
+ @AllArgsConstructor
+ @ConditionalOnProperty(value = TENANT_DYNAMIC_DATASOURCE_PROP, havingValue = "true")
+ public static class TenantDataSourceAnnotationConfiguration implements SmartInitializingSingleton {
+
+ private final TenantDataSourceAnnotationInterceptor tenantDataSourceAnnotationInterceptor;
+
+ private final DataSource dataSource;
+ private final DruidDataSourceCreator dataSourceCreator;
+ private final JdbcTemplate jdbcTemplate;
+
+ @Override
+ public void afterSingletonsInstantiated() {
+ TenantDataSourceHolder tenantDataSourceHolder = new TenantDataSourceHolder(dataSource, dataSourceCreator, jdbcTemplate);
+ tenantDataSourceAnnotationInterceptor.setHolder(tenantDataSourceHolder);
+ }
+ }
+
+ @Order
+ @AutoConfiguration
+ @AllArgsConstructor
+ @ConditionalOnProperty(value = TENANT_DYNAMIC_GLOBAL_PROP, havingValue = "true")
+ public static class TenantDataSourceGlobalConfiguration implements SmartInitializingSingleton {
+
+ private final TenantDataSourceGlobalInterceptor tenantDataSourceGlobalInterceptor;
+
+ private final DataSource dataSource;
+ private final DruidDataSourceCreator dataSourceCreator;
+ private final JdbcTemplate jdbcTemplate;
+
+ @Override
+ public void afterSingletonsInstantiated() {
+ TenantDataSourceHolder tenantDataSourceHolder = new TenantDataSourceHolder(dataSource, dataSourceCreator, jdbcTemplate);
+ tenantDataSourceGlobalInterceptor.setHolder(tenantDataSourceHolder);
+ }
+ }
+
+}
--
Gitblit v1.9.3