Fixed issue of can't switch datasource

Author: 何惠民

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

@@ -1,11 +1,10 @@
 package cn.com.hellowood.dynamicdatasource.configuration;
 
-import org.springframework.beans.factory.annotation.Qualifier;
+import org.mybatis.spring.SqlSessionFactoryBean;
 import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
 import org.springframework.boot.context.properties.ConfigurationProperties;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-import org.springframework.context.annotation.Primary;
 
 import javax.sql.DataSource;
 import java.util.HashMap;
@@ -14,21 +13,19 @@
 /**
  * Multiple DataSource Configurer
  *
- * @Date 2017-08-15 11:37
- * @Author HelloWood
- * @Email hellowoodes@gmail.com
+ * @author HelloWood
+ * @date 2017 -08-15 11:37
+ * @Email hellowoodes @gmail.com
  */
 @Configuration
 public class DataSourceConfigurer {
 
     /**
      * master DataSource
      *
-     * @return
+     * @return data source
      */
     @Bean("master")
-    @Qualifier("master")
-    @Primary
     @ConfigurationProperties(prefix = "application.server.db.master")
     public DataSource master() {
         return DataSourceBuilder.create().build();
@@ -37,53 +34,54 @@ public DataSource master() {
     /**
      * slave DataSource
      *
-     * @return
+     * @return data source
      */
     @Bean("slave")
-    @Qualifier("slave")
     @ConfigurationProperties(prefix = "application.server.db.slave")
     public DataSource slave() {
         return DataSourceBuilder.create().build();
     }
 
+    /**
+     * Dynamic data source.
+     *
+     * @return the data source
+     */
     @Bean("dynamicDataSource")
-    @Qualifier("dynamicDataSource")
-    public DataSource dataSource() {
+    public DataSource dynamicDataSource() {
         DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource();
-
-        dynamicRoutingDataSource.setDefaultTargetDataSource(slave());
-
         Map<Object, Object> dataSourceMap = new HashMap<>(2);
         dataSourceMap.put("master", master());
         dataSourceMap.put("slave", slave());
+
+        // Set master datasource as default
+        dynamicRoutingDataSource.setDefaultTargetDataSource(master());
+        // Set master and slave datasource as target datasource
         dynamicRoutingDataSource.setTargetDataSources(dataSourceMap);
-        DynamicDataSourceContextHolder.dataSourceIds.addAll(dataSourceMap.keySet());
+
+        // To put datasource keys into DataSourceContextHolder to judge if the datasource is exist
+        DynamicDataSourceContextHolder.dataSourceKeys.addAll(dataSourceMap.keySet());
         return dynamicRoutingDataSource;
     }
-//
-//    @Bean
-//    public PlatformTransactionManager transactionManager(DataSource dynamicDataSource) {
-//        return new DataSourceTransactionManager(dynamicDataSource);
-//    }
-//
-//    @Bean
-//    @ConfigurationProperties(prefix = "mybatis")
-//    public SqlSessionFactoryBean sqlSessionFactoryBean(DataSource dynamicDataSource) {
-//        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
-//        sqlSessionFactoryBean.setDataSource(dynamicDataSource);
-//        return sqlSessionFactoryBean;
-//    }
-//
-//    @Bean
-//    public SqlSessionFactory sqlSessionFactory() throws Exception {
-//        return sqlSessionFactoryBean(dataSource()).getObject();
-////        return sqlSessionFactoryBean.getObject();
-//    }
-//
-//    @Bean
-//    public SqlSessionTemplate sqlSessionTemplate(SqlSessionFactory sqlSessionFactory) {
-//        SqlSessionTemplate sqlSessionTemplate = new SqlSessionTemplate(sqlSessionFactory);
-//        return sqlSessionTemplate;
-//    }
+
+    /**
+     * Sql session factory bean.
+     * Here to config datasource for SqlSessionFactory
+     * <p>
+     * You need to add @{@code @ConfigurationProperties(prefix = "mybatis")}, if you are using *.xml file,
+     * the {@code 'mybatis.type-aliases-package'} and {@code 'mybatis.mapper-locations'} should be set in
+     * {@code 'application.properties'} file, or there will appear invalid bond statement exception
+     *
+     * @return the sql session factory bean
+     */
+    @Bean
+    @ConfigurationProperties(prefix = "mybatis")
+    public SqlSessionFactoryBean sqlSessionFactoryBean() {
+        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
+        // 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());
+        return sqlSessionFactoryBean;
+    }
 }
 

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

@@ -1,38 +1,39 @@
 package cn.com.hellowood.dynamicdatasource.configuration;
 
 
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
 import java.util.ArrayList;
 import java.util.List;
 
 /**
  * Multiple DataSource Context Holder
  *
- * @Date 2017-08-15 14:26
- * @Author HelloWood
- * @Email hellowoodes@gmail.com
+ * @author HelloWood
+ * @date 2017 -08-15 14:26
+ * @Email hellowoodes @gmail.com
  */
-//@Component
 public class DynamicDataSourceContextHolder {
 
-    private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class);
+//    private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class);
 
     /**
      * Maintain variable for every thread, to avoid effect other thread
      */
-    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
+    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>() {
+        @Override
+        protected String initialValue() {
+            return "master";
+        }
+    };
 
     /**
      * All DataSource List
      */
-    public static List<Object> dataSourceIds = new ArrayList<>();
+    public static List<Object> dataSourceKeys = new ArrayList<>();
 
     /**
      * To switch DataSource
      *
-     * @param key
+     * @param key the key
      */
     public static void setDataSourceKey(String key) {
         contextHolder.set(key);
@@ -41,7 +42,7 @@ public static void setDataSourceKey(String key) {
     /**
      * Get current DataSource
      *
-     * @return
+     * @return data source key
      */
     public static String getDataSourceKey() {
         return contextHolder.get();
@@ -57,11 +58,10 @@ public static void clearDataSourceKey() {
     /**
      * Check if give DataSource is in current DataSource list
      *
-     * @param key
-     * @return
+     * @param key the key
+     * @return boolean boolean
      */
     public static boolean containDataSourceKey(String key) {
-        logger.info("DataSourceIds is {}", dataSourceIds);
-        return dataSourceIds.contains(key);
+        return dataSourceKeys.contains(key);
     }
 }