ShardingConfig

src/main/java/com/sqx/config/RedisConfig.java

@@ -1,5 +1,6 @@
 package com.sqx.config;
 
+import com.sqx.common.utils.RedisUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.cache.CacheManager;
 import org.springframework.context.annotation.Bean;
@@ -70,4 +71,10 @@ public class RedisConfig {
     public ZSetOperations<String, Object> zSetOperations(RedisTemplate<String, Object> redisTemplate) {
         return redisTemplate.opsForZSet();
     }
+
+    @Bean
+    public RedisUtils redisUtils() {
+        // 根据实际情况初始化RedisUtils实例，可能需要传入相关配置参数，如Redis连接信息等
+        return new RedisUtils();
+    }
 }

src/main/java/com/sqx/config/ShardingConfig.java

@@ -0,0 +1,153 @@
+package com.sqx.config;
+
+import com.sqx.sharding.MasterSlaveRules;
+import com.sqx.sharding.ShardingDataBase;
+import com.zaxxer.hikari.HikariConfig;
+import com.zaxxer.hikari.HikariDataSource;
+import lombok.Data;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.shardingsphere.api.config.masterslave.MasterSlaveRuleConfiguration;
+import org.apache.shardingsphere.api.config.sharding.ShardingRuleConfiguration;
+import org.apache.shardingsphere.api.config.sharding.TableRuleConfiguration;
+import org.apache.shardingsphere.shardingjdbc.api.ShardingDataSourceFactory;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import javax.annotation.PostConstruct;
+import javax.sql.DataSource;
+import java.sql.SQLException;
+import java.util.*;
+import java.util.stream.Collectors;
+
+
+/**
+ * sharding-jdbc 配置文件
+ */
+@Slf4j
+@Data
+@Configuration
+@ConfigurationProperties(prefix = "spring.shardingsphere")
+public class ShardingConfig {
+
+    /**
+     * 读取数据源信息
+     */
+    private Map<String, ShardingDataBase> datasource;
+    /**
+     * 读写分离的配置
+     */
+    private Map<String, MasterSlaveRules> masterSlaveRules;
+    /**
+     * 显示sharding-jdbc的sql
+     */
+    private String showSql;
+
+    /**
+     * 中心库的节点
+     */
+    private String centerTablesDataNode;
+
+    @Value("${center-tables}")
+    private String tableNamesAsString;
+
+    /**
+     * 中心表,不进行分库操作
+     */
+//    @Value("${center-tables}")
+    private Set<String> centerTables;
+
+    @PostConstruct
+    public void init() {
+        // 使用换行符进行分割，然后去除空白字符，收集到Set集合中
+        centerTables = Arrays.stream(tableNamesAsString.split(","))
+                .map(String::trim)
+                .collect(Collectors.toSet());
+    }
+
+
+    /**
+     * 配置sharding-jdbc数据源
+     */
+    @Bean
+    public DataSource dataSource() throws SQLException {
+        ShardingRuleConfiguration shardingRuleConfig = new ShardingRuleConfiguration();
+
+        // 配置数据库主从
+        shardingRuleConfig.setMasterSlaveRuleConfigs(masterSlaveRuleConfigs());
+
+        // 配置表的切分策略
+        shardingRuleConfig.setTableRuleConfigs(addTableRuleConfigs());
+
+        // 配置是否显示sql
+//        Properties props = new Properties();
+//        props.put("sql.show", true);
+
+        // 配置数据源
+        Map<String, DataSource> dataSourceMap = getShardingDataBase();
+        return ShardingDataSourceFactory.createDataSource(dataSourceMap, shardingRuleConfig, null);
+//        return ShardingDataSourceFactory.createDataSource(dataSourceMap, shardingRuleConfig, props);
+    }
+
+    /**
+     * 配置
+     */
+    private Set<TableRuleConfiguration> addTableRuleConfigs() {
+        Set<TableRuleConfiguration> sets = new HashSet<>();
+        for (String centerTable : centerTables) {
+            TableRuleConfiguration tableRuleConfig = new TableRuleConfiguration(centerTable,
+                    String.format(centerTablesDataNode, centerTable));
+            sets.add(tableRuleConfig);
+        }
+
+//        // 定义区域表的分库规则
+//        InlineShardingStrategyConfiguration databaseShardingStrategyConfig = new InlineShardingStrategyConfiguration(
+//                regionTablesShardingDatabaseColumn, regionTablesShardingDatabaseAlgorithm);
+//        for (String regionTable : regionTables) {
+//            TableRuleConfiguration tableRuleConfig = new TableRuleConfiguration(regionTable, String.format(regionTablesDataNode, regionTable));
+//            tableRuleConfig.setDatabaseShardingStrategyConfig(databaseShardingStrategyConfig);
+//            sets.add(tableRuleConfig);
+//        }
+        return sets;
+    }
+
+    /**
+     * 配置数据源
+     */
+    private Map<String, DataSource> getShardingDataBase() {
+        String testQuery = "SELECT 1";
+        Map<String, DataSource> map = new HashMap<>();
+        datasource.forEach((datasourceName, shardingDataBase) -> {
+            HikariConfig config = new HikariConfig();
+            config.setDriverClassName(shardingDataBase.getDriverClassName());
+            config.setJdbcUrl(shardingDataBase.getJdbcUrl());
+            config.setUsername(shardingDataBase.getUsername());
+            config.setPassword(shardingDataBase.getPassword());
+            config.setPoolName(datasourceName);
+            config.setMinimumIdle(shardingDataBase.getMinimumIdle());
+            config.setMaximumPoolSize(shardingDataBase.getMaximumPoolSize());
+            config.setIdleTimeout(shardingDataBase.getIdleTimeout());
+            config.setMaxLifetime(shardingDataBase.getMaxLifetime());
+            config.setConnectionTimeout(shardingDataBase.getConnectionTimeout());
+            config.setConnectionTestQuery(testQuery);
+            map.put(datasourceName, new HikariDataSource(config));
+        });
+        return map;
+    }
+
+    /**
+     * 配置读写分离
+     */
+    private Set<MasterSlaveRuleConfiguration> masterSlaveRuleConfigs() {
+        Set<MasterSlaveRuleConfiguration> sets = new HashSet<>();
+        masterSlaveRules.forEach((databaseName, masterSlaveRules) -> {
+            MasterSlaveRuleConfiguration masterSlaveRuleConfig = new MasterSlaveRuleConfiguration(databaseName,
+                    masterSlaveRules.getMasterDataSourceName(), masterSlaveRules.getSlaveDataSourceNames());
+            System.out.println("写入"+masterSlaveRules.getMasterDataSourceName());
+            System.out.println("读出"+masterSlaveRules.getSlaveDataSourceNames());
+            sets.add(masterSlaveRuleConfig);
+        });
+        return sets;
+    }
+}

src/main/java/com/sqx/config/ShardingDataSourcePoolMetricsAutoConfiguration.java

@@ -0,0 +1,67 @@
+package com.sqx.config;
+
+import com.zaxxer.hikari.HikariDataSource;
+import com.zaxxer.hikari.metrics.micrometer.MicrometerMetricsTrackerFactory;
+import io.micrometer.core.instrument.MeterRegistry;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.shardingsphere.shardingjdbc.jdbc.core.datasource.ShardingDataSource;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.actuate.autoconfigure.metrics.MetricsAutoConfiguration;
+import org.springframework.boot.actuate.autoconfigure.metrics.export.simple.SimpleMetricsExportAutoConfiguration;
+import org.springframework.boot.autoconfigure.AutoConfigureAfter;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
+import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
+import org.springframework.boot.jdbc.DataSourceUnwrapper;
+import org.springframework.context.annotation.Configuration;
+
+import javax.sql.DataSource;
+import java.util.Collection;
+
+/**
+ * @author wxkj0012
+ */
+@Configuration(proxyBeanMethods = false)
+@AutoConfigureAfter({MetricsAutoConfiguration.class, DataSourceAutoConfiguration.class,
+        SimpleMetricsExportAutoConfiguration.class})
+@ConditionalOnClass({DataSource.class, MeterRegistry.class})
+@ConditionalOnBean({DataSource.class, MeterRegistry.class})
+@Slf4j
+public class ShardingDataSourcePoolMetricsAutoConfiguration {
+
+    @Configuration(proxyBeanMethods = false)
+    @ConditionalOnClass({ShardingDataSource.class, HikariDataSource.class})
+    static class ShardingDataSourceMetricsConfiguration {
+
+        private final MeterRegistry registry;
+
+        ShardingDataSourceMetricsConfiguration(MeterRegistry registry) {
+            this.registry = registry;
+        }
+
+        @Autowired
+        void bindMetricsRegistryToHikariDataSources(Collection<DataSource> dataSources) {
+            for (DataSource dataSource : dataSources) {
+                ShardingDataSource shardingDataSource = DataSourceUnwrapper.unwrap(dataSource, ShardingDataSource.class);
+                Collection<DataSource> shardDataSources = shardingDataSource.getDataSourceMap().values();
+                for (DataSource ds : shardDataSources) {
+                    HikariDataSource hikariDataSource = DataSourceUnwrapper.unwrap(ds, HikariDataSource.class);
+                    if (hikariDataSource != null) {
+                        bindMetricsRegistryToHikariDataSource(hikariDataSource);
+                    }
+                }
+            }
+        }
+
+        private void bindMetricsRegistryToHikariDataSource(HikariDataSource hikari) {
+            if (hikari.getMetricRegistry() == null && hikari.getMetricsTrackerFactory() == null) {
+                try {
+                    hikari.setMetricsTrackerFactory(new MicrometerMetricsTrackerFactory(this.registry));
+                } catch (Exception ex) {
+                    log.warn("Failed to bind Hikari metrics: {}}", ex.getMessage());
+                }
+            }
+        }
+
+    }
+}

src/main/java/com/sqx/config/SwaggerConfig.java

@@ -1,58 +1,85 @@
-package com.sqx.config;
-
-import io.swagger.annotations.ApiOperation;
-import org.springframework.beans.factory.annotation.Value;
-import org.springframework.context.annotation.Bean;
-import org.springframework.context.annotation.Configuration;
-import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
-import springfox.documentation.builders.ApiInfoBuilder;
-import springfox.documentation.builders.PathSelectors;
-import springfox.documentation.builders.RequestHandlerSelectors;
-import springfox.documentation.service.ApiInfo;
-import springfox.documentation.service.ApiKey;
-import springfox.documentation.spi.DocumentationType;
-import springfox.documentation.spring.web.plugins.Docket;
-import springfox.documentation.swagger2.annotations.EnableSwagger2;
-
-import java.util.List;
-
-import static com.google.common.collect.Lists.newArrayList;
-
-@Configuration
-@EnableSwagger2
-public class SwaggerConfig implements WebMvcConfigurer {
-
-    @Value("${swagger.enabled}")
-    private boolean enabled;
-
-    @Bean
-    public Docket createRestApi() {
-        return new Docket(DocumentationType.SWAGGER_2)
-                .enable(enabled)
-            .apiInfo(apiInfo())
-            .select()
-            //加了ApiOperation注解的类，才生成接口文档
-            .apis(RequestHandlerSelectors.withMethodAnnotation(ApiOperation.class))
-            //包下的类，才生成接口文档
-            //.apis(RequestHandlerSelectors.basePackage("com.sqx.controller"))
-            .paths(PathSelectors.any())
-            .build()
-            .securitySchemes(security());
-    }
-
-    private ApiInfo apiInfo() {
-        return new ApiInfoBuilder()
-            .title("")
-            .description("sqx-fast文档")
-            .termsOfServiceUrl("")
-            .version("3.0.0")
-            .build();
-    }
-
-    private List<ApiKey> security() {
-        return newArrayList(
-            new ApiKey("token", "token", "header")
-        );
-    }
-
-}
+//package com.sqx.config;
+//
+//import io.swagger.annotations.Api;
+//import io.swagger.annotations.ApiOperation;
+//import org.springframework.beans.factory.annotation.Value;
+//import org.springframework.context.annotation.Bean;
+//import org.springframework.context.annotation.Configuration;
+//import org.springframework.context.annotation.Profile;
+//import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+//import springfox.documentation.builders.ApiInfoBuilder;
+//import springfox.documentation.builders.PathSelectors;
+//import springfox.documentation.builders.RequestHandlerSelectors;
+//import springfox.documentation.service.ApiInfo;
+//import springfox.documentation.service.ApiKey;
+//import springfox.documentation.spi.DocumentationType;
+//import springfox.documentation.spring.web.plugins.Docket;
+//import springfox.documentation.swagger2.annotations.EnableSwagger2;
+//
+//import java.util.List;
+//
+//import static com.google.common.collect.Lists.newArrayList;
+//
+////@Configuration
+////@EnableSwagger2
+////public class SwaggerConfig implements WebMvcConfigurer {
+////
+////    @Value("${swagger.enabled}")
+////    private boolean enabled;
+////
+////    @Bean
+////    public Docket createRestApi() {
+////        return new Docket(DocumentationType.SWAGGER_2)
+////                .enable(enabled)
+////            .apiInfo(apiInfo())
+////            .select()
+////            //加了ApiOperation注解的类，才生成接口文档
+////            .apis(RequestHandlerSelectors.withMethodAnnotation(ApiOperation.class))
+////            //包下的类，才生成接口文档
+////            //.apis(RequestHandlerSelectors.basePackage("com.sqx.controller"))
+////            .paths(PathSelectors.any())
+////            .build()
+////            .securitySchemes(security());
+////    }
+////
+////    private ApiInfo apiInfo() {
+////        return new ApiInfoBuilder()
+////            .title("")
+////            .description("sqx-fast文档")
+////            .termsOfServiceUrl("")
+////            .version("3.0.0")
+////            .build();
+////    }
+////
+////    private List<ApiKey> security() {
+////        return newArrayList(
+////            new ApiKey("token", "token", "header")
+////        );
+////    }
+////
+////}
+//
+//@Configuration
+//@EnableSwagger2
+//@Profile({"local", "dev"})
+//public class SwaggerConfig {
+//
+//    @Bean
+//    public Docket api() {
+//        return new Docket(DocumentationType.SWAGGER_2)
+//                .apiInfo(apiInfo())
+//                .select()
+//                .apis(RequestHandlerSelectors.withClassAnnotation(Api.class))
+//                .apis(RequestHandlerSelectors.withMethodAnnotation(ApiOperation.class))
+//                .paths(PathSelectors.any())
+//                .build();
+//    }
+//
+//    private ApiInfo apiInfo() {
+//        return new ApiInfoBuilder()
+//                .title("悠车位后台API接口文档")
+//                .description("悠车位服务端后台API接口文档")
+//                .version("1.0")
+//                .build();
+//    }
+//}