feat: sharding jdbc 数据源集成到 dynamic-datasource

Author: cadecode

common/plugin/datasource/src/main/java/com/github/cadecode/uniboot/common/plugin/datasource/config/ShardingJdbcConfig.java

@@ -0,0 +1,65 @@
+package com.github.cadecode.uniboot.common.plugin.datasource.config;
+
+import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
+import com.baomidou.dynamic.datasource.provider.AbstractDataSourceProvider;
+import com.baomidou.dynamic.datasource.provider.DynamicDataSourceProvider;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceAutoConfiguration;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
+import lombok.RequiredArgsConstructor;
+import org.apache.shardingsphere.driver.jdbc.adapter.AbstractDataSourceAdapter;
+import org.springframework.boot.autoconfigure.AutoConfigureBefore;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Lazy;
+import org.springframework.context.annotation.Primary;
+
+import javax.annotation.Resource;
+import javax.sql.DataSource;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Sharding JDBC 配置
+ *
+ * @author Cade Li
+ * @since 2023/11/29
+ */
+@RequiredArgsConstructor
+@ConditionalOnProperty(name = "spring.shardingsphere.enabled", havingValue = "true")
+@AutoConfigureBefore(DynamicDataSourceAutoConfiguration.class)
+@Configuration
+public class ShardingJdbcConfig {
+
+    public static final String SHARDING_DATA_SOURCE_NAME = "sharding";
+
+    private final DynamicDataSourceProperties properties;
+
+    @Lazy
+    @Resource
+    private AbstractDataSourceAdapter shardingSphereDataSource;
+
+    @Bean
+    public DynamicDataSourceProvider dynamicDataSourceProvider() {
+        return new AbstractDataSourceProvider() {
+            @Override
+            public Map<String, DataSource> loadDataSources() {
+                HashMap<String, DataSource> dataSourceMap = new HashMap<>();
+                dataSourceMap.put(SHARDING_DATA_SOURCE_NAME, shardingSphereDataSource);
+                return dataSourceMap;
+            }
+        };
+    }
+
+    @Primary
+    @Bean
+    public DataSource dataSource() {
+        DynamicRoutingDataSource dataSource = new DynamicRoutingDataSource();
+        dataSource.setPrimary(properties.getPrimary());
+        dataSource.setStrict(properties.getStrict());
+        dataSource.setStrategy(properties.getStrategy());
+        dataSource.setP6spy(properties.getP6spy());
+        dataSource.setSeata(properties.getSeata());
+        return dataSource;
+    }
+}