Change datasource to Hikari

Author: 何惠民

README.md

@@ -7,6 +7,7 @@
 - [druid](https://github.com/helloworlde/SpringBoot-DynamicDataSource/tree/druid): 通过切面和注解方式实现的使用 Druid 连接池的动态数据源切换
 - [aspect_dao](https://github.com/helloworlde/SpringBoot-DynamicDataSource/tree/aspect_dao): 通过切面实现的 DAO 层的动态数据源切换
 - [roundrobin](https://github.com/helloworlde/SpringBoot-DynamicDataSource/tree/roundrobin): 通过切面使用轮询方式实现的只读数据源负载均衡
+- [hikari](https://github.com/helloworlde/SpringBoot-DynamicDataSource/tree/hikari): 升级到SpringBoot 2.0版本 数据源使用 Hikari
 
 > 以上分支都是基于 dev 分支修改或扩充而来，基本涵盖了常用的多数据源动态切换的方式，基本的原理都一样，都是通过切面根据不同的条件在执行数据库操作前切换数据源
 
@@ -28,10 +29,9 @@
 ## 添加依赖
 ```groovy
 dependencies {
-    compile('org.mybatis.spring.boot:mybatis-spring-boot-starter:1.3.1')
+    compile('org.mybatis.spring.boot:mybatis-spring-boot-starter:1.3.2')
     compile('org.springframework.boot:spring-boot-starter-web')
     compile('org.springframework.boot:spring-boot-starter-aop')
-    compile('com.alibaba:druid-spring-boot-starter:1.1.6')
     runtime('mysql:mysql-connector-java')
     testCompile('org.springframework.boot:spring-boot-starter-test')
 }
@@ -83,88 +83,40 @@ INSERT INTO product_slave_gamma.product (name, price) VALUES('slaveGamma', '1');
 - application.properties
 
 ```properties
+spring.datasource.type=com.zaxxer.hikari.HikariDataSource
 # Master datasource config
-spring.datasource.druid.master.name=master
-spring.datasource.druid.master.driver-class-name=com.mysql.jdbc.Driver
-spring.datasource.druid.master.url=jdbc:mysql://localhost/product_master?useSSL=false
-spring.datasource.druid.master.port=3306
-spring.datasource.druid.master.username=root
-spring.datasource.druid.master.password=123456
+spring.datasource.hikari.master.name=master
+spring.datasource.hikari.master.driver-class-name=com.mysql.jdbc.Driver
+spring.datasource.hikari.master.jdbc-url=jdbc:mysql://localhost/product_master?useSSL=false
+spring.datasource.hikari.master.port=3306
+spring.datasource.hikari.master.username=root
+spring.datasource.hikari.master.password=123456
 
 # SlaveAlpha datasource config
-spring.datasource.druid.slave-alpha.name=SlaveAlpha
-spring.datasource.druid.slave-alpha.driver-class-name=com.mysql.jdbc.Driver
-spring.datasource.druid.slave-alpha.url=jdbc:mysql://localhost/product_slave_alpha?useSSL=false
-spring.datasource.druid.slave-alpha.port=3306
-spring.datasource.druid.slave-alpha.username=root
-spring.datasource.druid.slave-alpha.password=123456
+spring.datasource.hikari.slave-alpha.name=SlaveAlpha
+spring.datasource.hikari.slave-alpha.driver-class-name=com.mysql.jdbc.Driver
+spring.datasource.hikari.slave-alpha.jdbc-url=jdbc:mysql://localhost/product_slave_alpha?useSSL=false
+spring.datasource.hikari.slave-alpha.port=3306
+spring.datasource.hikari.slave-alpha.username=root
+spring.datasource.hikari.slave-alpha.password=123456
 
 # SlaveBeta datasource config
-spring.datasource.druid.slave-beta.name=SlaveBeta
-spring.datasource.druid.slave-beta.driver-class-name=com.mysql.jdbc.Driver
-spring.datasource.druid.slave-beta.url=jdbc:mysql://localhost/product_slave_beta?useSSL=false
-spring.datasource.druid.slave-beta.port=3306
-spring.datasource.druid.slave-beta.username=root
-spring.datasource.druid.slave-beta.password=123456
+spring.datasource.hikari.slave-beta.name=SlaveBeta
+spring.datasource.hikari.slave-beta.driver-class-name=com.mysql.jdbc.Driver
+spring.datasource.hikari.slave-beta.jdbc-url=jdbc:mysql://localhost/product_slave_beta?useSSL=false
+spring.datasource.hikari.slave-beta.port=3306
+spring.datasource.hikari.slave-beta.username=root
+spring.datasource.hikari.slave-beta.password=123456
 
 # SlaveGamma datasource config
-spring.datasource.druid.slave-gamma.name=SlaveGamma
-spring.datasource.druid.slave-gamma.driver-class-name=com.mysql.jdbc.Driver
-spring.datasource.druid.slave-gamma.url=jdbc:mysql://localhost/product_slave_gamma?useSSL=false
-spring.datasource.druid.slave-gamma.port=3306
-spring.datasource.druid.slave-gamma.username=root
-spring.datasource.druid.slave-gamma.password=123456
-
-# Druid dataSource config
-spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
-spring.datasource.druid.initial-size=5
-spring.datasource.druid.max-active=20
-spring.datasource.druid.min-idle=5
-spring.datasource.druid.max-wait=60000
-spring.datasource.druid.pool-prepared-statements=false
-spring.datasource.druid.validation-query=SELECT 1
-spring.datasource.druid.validation-query-timeout=30000
-spring.datasource.druid.test-on-borrow=false
-spring.datasource.druid.test-on-return=false
-spring.datasource.druid.test-while-idle=true
-#spring.datasource.druid.time-between-eviction-runs-millis=
-#spring.datasource.druid.min-evictable-idle-time-millis=
-#spring.datasource.druid.max-evictable-idle-time-millis=10000
-
-# Druid stat filter config
-spring.datasource.druid.filters=stat,wall,log4j
-spring.datasource.druid.web-stat-filter.enabled=true
-spring.datasource.druid.web-stat-filter.url-pattern=/*
-spring.datasource.druid.web-stat-filter.exclusions=*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*
-spring.datasource.druid.web-stat-filter.session-stat-enable=true
-spring.datasource.druid.web-stat-filter.session-stat-max-count=10
-spring.datasource.druid.web-stat-filter.principal-session-name=user
-#spring.datasource.druid.web-stat-filter.principal-cookie-name=
-spring.datasource.druid.web-stat-filter.profile-enable=true
-spring.datasource.druid.filter.stat.db-type=mysql
-spring.datasource.druid.filter.stat.log-slow-sql=true
-spring.datasource.druid.filter.stat.slow-sql-millis=1000
-spring.datasource.druid.filter.stat.merge-sql=true
-spring.datasource.druid.filter.wall.enabled=true
-spring.datasource.druid.filter.wall.config.delete-allow=true
-spring.datasource.druid.filter.wall.config.drop-table-allow=false
-spring.datasource.druid.filter.slf4j.enabled=true
-# Druid manage page config
-spring.datasource.druid.stat-view-servlet.enabled=true
-spring.datasource.druid.stat-view-servlet.url-pattern=/druid/*
-spring.datasource.druid.stat-view-servlet.reset-enable=true
-spring.datasource.druid.stat-view-servlet.login-username=admin
-spring.datasource.druid.stat-view-servlet.login-password=admin
-#spring.datasource.druid.stat-view-servlet.allow=
-#spring.datasource.druid.stat-view-servlet.deny=
-spring.datasource.druid.use-global-data-source-stat=true
-# Druid AOP config
-spring.datasource.druid.aop-patterns=cn.com.hellowood.dynamicdatasource.service.*
-spring.aop.proxy-target-class=true
+spring.datasource.hikari.slave-gamma.name=SlaveGamma
+spring.datasource.hikari.slave-gamma.driver-class-name=com.mysql.jdbc.Driver
+spring.datasource.hikari.slave-gamma.jdbc-url=jdbc:mysql://localhost/product_slave_gamma?useSSL=false
+spring.datasource.hikari.slave-gamma.port=3306
+spring.datasource.hikari.slave-gamma.username=root
+spring.datasource.hikari.slave-gamma.password=123456
 
-# MyBatis config
-mybatis.type-aliases-package=cn.com.hellowood.dynamicdatasource.mapper
-mybatis.mapper-locations=mappers/**Mapper.xml
+spring.aop.proxy-target-class=true
 server.port=9999
 ```
 
@@ -215,7 +167,7 @@ public class DynamicRoutingDataSource extends AbstractRoutingDataSource {
 package cn.com.hellowood.dynamicdatasource.configuration;
 
 import org.mybatis.spring.SqlSessionFactoryBean;
-import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
+import org.springframework.boot.jdbc.DataSourceBuilder;
 import org.springframework.boot.context.properties.ConfigurationProperties;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -238,9 +190,9 @@ public class DataSourceConfigurer {
      */
     @Bean("master")
     @Primary
-    @ConfigurationProperties(prefix = "spring.datasource.druid.master")
+    @ConfigurationProperties(prefix = "spring.datasource.hikari.master")
     public DataSource master() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -249,9 +201,9 @@ public class DataSourceConfigurer {
      * @return the data source
      */
     @Bean("slaveAlpha")
-    @ConfigurationProperties(prefix = "spring.datasource.druid.slave-alpha")
+    @ConfigurationProperties(prefix = "spring.datasource.hikari.slave-alpha")
     public DataSource slaveAlpha() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -260,9 +212,9 @@ public class DataSourceConfigurer {
      * @return the data source
      */
     @Bean("slaveBeta")
-    @ConfigurationProperties(prefix = "spring.datasource.druid.slave-beta")
+    @ConfigurationProperties(prefix = "spring.datasource.hikari.slave-beta")
     public DataSource slaveBeta() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -273,7 +225,7 @@ public class DataSourceConfigurer {
     @Bean("slaveGamma")
     @ConfigurationProperties(prefix = "spring.datasource.druid.slave-gamma")
     public DataSource slaveGamma() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -315,6 +267,10 @@ public class DataSourceConfigurer {
     @ConfigurationProperties(prefix = "mybatis")
     public SqlSessionFactoryBean sqlSessionFactoryBean() {
         SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
+        // 配置 MyBatis
+        sqlSessionFactoryBean.setTypeAliasesPackage("cn.com.hellowood.dynamicdatasource.mapper");
+        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("mappers/**Mapper.xml"));
+
         // 配置数据源，此处配置为关键配置，如果没有将 dynamicDataSource 作为数据源则不能实现切换
         sqlSessionFactoryBean.setDataSource(dynamicDataSource());
         return sqlSessionFactoryBean;
@@ -346,18 +302,11 @@ import org.slf4j.LoggerFactory;
 
 import java.util.ArrayList;
 import java.util.List;
-import java.util.concurrent.locks.Lock;
-import java.util.concurrent.locks.ReentrantLock;
 
 public class DynamicDataSourceContextHolder {
 
     private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class);
 
-    /**
-     * 用于在切换数据源时保证不会被其他线程修改
-     */
-    private static Lock lock = new ReentrantLock();
-
     /**
      * 用于轮循的计数器
      */
@@ -399,7 +348,6 @@ public class DynamicDataSourceContextHolder {
      * 当使用只读数据源时通过轮循方式选择要使用的数据源
      */
     public static void useSlaveDataSource() {
-        lock.lock();
 
         try {
             int datasourceKeyIndex = counter % slaveDataSourceKeys.size();
@@ -409,9 +357,7 @@ public class DynamicDataSourceContextHolder {
             logger.error("Switch slave datasource failed, error message is {}", e.getMessage());
             useMasterDataSource();
             e.printStackTrace();
-        } finally {
-            lock.unlock();
-        }
+        } 
     }
 
     /**

build.gradle

@@ -1,6 +1,6 @@
 buildscript {
     ext {
-        springBootVersion = '1.5.9.RELEASE'
+        springBootVersion = '2.0.1.RELEASE'
     }
     repositories {
         maven { url 'http://maven.aliyun.com/nexus/content/groups/public/' }
@@ -14,6 +14,7 @@ buildscript {
 apply plugin: 'java'
 apply plugin: 'eclipse'
 apply plugin: 'org.springframework.boot'
+apply plugin: 'io.spring.dependency-management'
 
 group = 'cn.com.hellowood'
 version = '0.0.1-SNAPSHOT'
@@ -26,10 +27,10 @@ repositories {
 
 
 dependencies {
-    compile('org.mybatis.spring.boot:mybatis-spring-boot-starter:1.3.1')
+    compile('org.springframework.boot:spring-boot-starter-parent:2.0.1.RELEASE')
+    compile('org.mybatis.spring.boot:mybatis-spring-boot-starter:1.3.2')
     compile('org.springframework.boot:spring-boot-starter-web')
     compile('org.springframework.boot:spring-boot-starter-aop')
-    compile('com.alibaba:druid-spring-boot-starter:1.1.6')
     runtime('mysql:mysql-connector-java')
     testCompile('org.springframework.boot:spring-boot-starter-test')
 }

gradle/wrapper/gradle-wrapper.properties

@@ -1,5 +1,6 @@
+#Fri May 11 12:58:07 CST 2018
 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
 zipStoreBase=GRADLE_USER_HOME
 zipStorePath=wrapper/dists
-distributionUrl=https\://services.gradle.org/distributions/gradle-3.5.1-bin.zip
+distributionUrl=https\://rdc-public-software.oss-cn-hangzhou.aliyuncs.com/gradle-4.1-bin.zip

src/main/java/cn/com/hellowood/dynamicdatasource/configuration/DataSourceConfigurer.java

@@ -1,16 +1,18 @@
 package cn.com.hellowood.dynamicdatasource.configuration;
 
 import cn.com.hellowood.dynamicdatasource.common.DataSourceKey;
-import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
 import org.mybatis.spring.SqlSessionFactoryBean;
 import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.jdbc.DataSourceBuilder;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.context.annotation.Primary;
+import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
 import org.springframework.jdbc.datasource.DataSourceTransactionManager;
 import org.springframework.transaction.PlatformTransactionManager;
 
 import javax.sql.DataSource;
+import java.io.IOException;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -31,9 +33,9 @@ public class DataSourceConfigurer {
      */
     @Bean("master")
     @Primary
-    @ConfigurationProperties(prefix = "spring.datasource.druid.master")
+    @ConfigurationProperties(prefix = "spring.datasource.hikari.master")
     public DataSource master() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -42,9 +44,9 @@ public DataSource master() {
      * @return the data source
      */
     @Bean("slaveAlpha")
-    @ConfigurationProperties(prefix = "spring.datasource.druid.slave-alpha")
+    @ConfigurationProperties(prefix = "spring.datasource.hikari.slave-alpha")
     public DataSource slaveAlpha() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -53,9 +55,9 @@ public DataSource slaveAlpha() {
      * @return the data source
      */
     @Bean("slaveBeta")
-    @ConfigurationProperties(prefix = "spring.datasource.druid.slave-beta")
+    @ConfigurationProperties(prefix = "spring.datasource.hikari.slave-beta")
     public DataSource slaveBeta() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -64,9 +66,9 @@ public DataSource slaveBeta() {
      * @return the data source
      */
     @Bean("slaveGamma")
-    @ConfigurationProperties(prefix = "spring.datasource.druid.slave-gamma")
+    @ConfigurationProperties(prefix = "spring.datasource.hikari.slave-gamma")
     public DataSource slaveGamma() {
-        return DruidDataSourceBuilder.create().build();
+        return DataSourceBuilder.create().build();
     }
 
     /**
@@ -109,8 +111,11 @@ public DataSource dynamicDataSource() {
      */
     @Bean
     @ConfigurationProperties(prefix = "mybatis")
-    public SqlSessionFactoryBean sqlSessionFactoryBean() {
+    public SqlSessionFactoryBean sqlSessionFactoryBean() throws IOException {
         SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
+        // Here to config mybatis
+        sqlSessionFactoryBean.setTypeAliasesPackage("cn.com.hellowood.dynamicdatasource.mapper");
+        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("mappers/**Mapper.xml"));
         // Here is very important, if don't config this, will can't switch datasource
         // put all datasource into SqlSessionFactoryBean, then will autoconfig SqlSessionFactory
         sqlSessionFactoryBean.setDataSource(dynamicDataSource());

src/main/java/cn/com/hellowood/dynamicdatasource/configuration/DynamicDataSourceContextHolder.java

@@ -7,8 +7,6 @@
 
 import java.util.ArrayList;
 import java.util.List;
-import java.util.concurrent.locks.Lock;
-import java.util.concurrent.locks.ReentrantLock;
 
 /**
  * Multiple DataSource Context Holder
@@ -21,8 +19,6 @@ public class DynamicDataSourceContextHolder {
 
     private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class);
 
-    private static Lock lock = new ReentrantLock();
-
     private static int counter = 0;
 
     /**
@@ -61,8 +57,6 @@ public static void useMasterDataSource() {
      * Use slave data source.
      */
     public static void useSlaveDataSource() {
-        lock.lock();
-
         try {
             int datasourceKeyIndex = counter % slaveDataSourceKeys.size();
             CONTEXT_HOLDER.set(String.valueOf(slaveDataSourceKeys.get(datasourceKeyIndex)));
@@ -71,8 +65,6 @@ public static void useSlaveDataSource() {
             logger.error("Switch slave datasource failed, error message is {}", e.getMessage());
             useMasterDataSource();
             e.printStackTrace();
-        } finally {
-            lock.unlock();
         }
     }