mybatis用spring的動態數據源實現讀寫分離

一、環境：

　　三個mysql數據庫。一個master，兩個slaver。master寫數據，slaver讀數據。

二、原理：

　　借助Spring的 AbstractRoutingDataSource 這個抽象實現。我們要實現 determineCurrentLookupKey()這個方法來動態的選擇使用哪個數據源操着數據庫

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {

    protected abstract Object determineCurrentLookupKey();

}

 

三、實現步驟：

1、添加spring，mybatis，mysql相關的pom依賴。

2、寫jdbc.properties，定義三個數據庫。

jdbc.driver=com.mysql.jdbc.Driver

#master庫
master.jdbc.url=jdbc:mysql://127.0.0.1:3306/master?characterEncoding=utf8
master.jdbc.user=root
master.jdbc.password=tiger
#slave 一 庫
slave.one.jdbc.url=jdbc:mysql://127.0.0.1:3306/slave-one?characterEncoding=utf8
slave.one.jdbc.user=root
slave.one.jdbc.password=tiger
#slave 二 庫
slave.two.jdbc.url=jdbc:mysql://127.0.0.1:3306/slave-two?characterEncoding=utf8
slave.two.jdbc.user=root
slave.two.jdbc.password=tiger

 

3、配置三個數據源，分別寫到三個配置文件中。

datasources-master.xml、datasource-slave-one.xml和datasource-slave-two.xml三個文件都一樣，這里就寫一個

<!--master數據源，支持讀寫-->
    <bean id="dataSourceMaster" class="com.alibaba.druid.pool.DruidDataSource"
          init-method="init" destroy-method="close">
        <property name="driverClassName" value="${jdbc.driver}"/>
        <property name="url" value="${master.jdbc.url}"/>
        <property name="username" value="${master.jdbc.user}"/>
        <property name="password" value="${master.jdbc.password}"/>

        <property name="filters" value="stat"/>

        <property name="maxActive" value="20"/>
        <property name="initialSize" value="1"/>
        <property name="maxWait" value="60000"/>
        <property name="minIdle" value="1"/>

        <property name="timeBetweenEvictionRunsMillis" value="60000"/>
        <property name="minEvictableIdleTimeMillis" value="300000"/>

        <property name="validationQuery" value="SELECT 'x'"/>
        <property name="testWhileIdle" value="true"/>
        <property name="testOnBorrow" value="false"/>
        <property name="testOnReturn" value="false"/>
    </bean>

 

4、寫spring的配置文件applicationContext.xml。

    <bean id="propertyConfigurer" class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">
        <property name="locations">
            <list>
                <value>classpath*:jdbc.properties</value>
            </list>
        </property>
        <property name="fileEncoding" value="UTF-8" />
        <property name= "ignoreResourceNotFound" value="false"/>
    </bean>

    <context:component-scan base-package="org.hope.lee"/>

    <aop:aspectj-autoproxy/>

    <!--spring的路由來管理數據源-->
    <bean id="dynamicDataSource" class="org.hope.lee.utils.DynamicDataSource">
        <property name="targetDataSources">
            <map>
                <entry value-ref="dataSourceMaster" key="db_master"/>
                <entry value-ref="dataSourceSlaveOne" key="db_slave_one"/>
                <entry value-ref="dataSourceSlaveTwo" key="db_slave_two"/>
            </map>
        </property>
    </bean>

    <!--spring-mybatis整合-->
    <bean id="dynamicsqlSessionFactory"
          class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="dynamicDataSource"/>
        <property name="mapperLocations">
            <array>
                <value>classpath:mappers/*Mapper.xml</value>
            </array>
        </property>
        <property name="configLocation" value="classpath:mybatis-config.xml"/>
        <property name="typeAliasesPackage" value="org.hope.lee.model"/>
    </bean>

    <!--自動掃描所有的Mapper接口與文件-->
    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="org.hope.lee.dao"></property>
        <property name="sqlSessionFactoryBeanName" value="dynamicsqlSessionFactory"></property>
    </bean>

    <!--配置事務-->
    <bean id="transactionManager"
          class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dynamicDataSource"/>
    </bean>
    <!--開啟注解事務-->
    <tx:annotation-driven transaction-manager="transactionManager"/>

    <import resource="classpath:datasource-master.xml"/>
    <import resource="classpath:datasource-slave-one.xml"/>
    <import resource="classpath:datasource-slave-two.xml"/>

 

5、新建DBContextHolder，DBType為動態設置數據庫的util

package org.hope.lee.utils;

public class DBType {
    public final static String DB_TYPE_MASTER = "db_master";

    public final static String DB_TYPE_SLAVE_ONE = "db_slave_one";

    public final static String DB_TYPE_SLAVE_TWO = "db_slave_two";
}

package org.hope.lee.utils;

public class DBContextHolder {

    private static ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    public static String getDBType() {
        String db = contextHolder.get();
        if(db == null) {
            db = DBType.DB_TYPE_MASTER; //默認是master庫
        }

        return db;
    }

    public static void setDBType(String dbType) {
        contextHolder.set(dbType);
    }

    public static void clearDBType() {
        contextHolder.remove();
    }

}

 

6、繼承Spring的 AbstractRoutingDataSource 來動態的進行數據庫路由

package org.hope.lee.utils;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DBContextHolder.getDBType();
    }
}

 

7、創建三個數據庫master、slave-one、slave-two。三個庫建同一張user表進行測試。

DROP TABLE IF EXISTS `user`;
CREATE TABLE `user` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `name` varchar(255) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8;

 

8、寫mybatis的dao層,service層,model層

package org.hope.lee.model;

public class User {
    private int id;
    private String name;

    setters()&getters()
}

 

package org.hope.lee.dao;

import org.hope.lee.model.User;
import org.springframework.stereotype.Repository;

@Repository
public interface UserMapper {
    void insert(User user);
    User selectOne(int id);
}

 

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">

<mapper namespace="org.hope.lee.dao.UserMapper">
    <resultMap id="usersResultMap" type="User">
        <id column="id" property="id" javaType="Integer" />
        <result column="name" property="name" />
    </resultMap>

    <insert id="insert" useGeneratedKeys="true" parameterType="User">
        INSERT INTO `user`(name) VALUES(#{name, jdbcType=VARCHAR});
    </insert>

    <select id="selectOne" resultMap="usersResultMap" parameterType="int" >
        SELECT id, name FROM `user` WHERE id=#{id, jdbcType=INTEGER}
    </select>
</mapper>

 

package org.hope.lee.service;

import org.hope.lee.dao.UserMapper;
import org.hope.lee.model.User;
import org.hope.lee.utils.DBContextHolder;
import org.hope.lee.utils.DBType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service("userService")
public class UserServiceImpl implements UserService {

    @Autowired
    private UserMapper userMapper;

    public void addUser(User user) {
        //設置數據庫
        DBContextHolder.setDBType(DBType.DB_TYPE_MASTER);
        userMapper.insert(user);
    }

    public User getUserById(int id) {
       //設置數據庫，單元測試的時候自己手動修改一下，看看效果       
　　　　DBContextHolder.setDBType(DBType.DB_TYPE_SLAVE_ONE);
       return userMapper.selectOne(id);
    }
}

 

9、單元測試。

import org.hope.lee.model.User;
import org.hope.lee.service.UserService;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath*:applicationContext.xml"})
public class UserServiceTest {

    @Autowired
    private UserService userService;

    @Test
    public void addUserTest() {
        User user = new User();
        user.setName("馬雲");

        userService.addUser(user);
    }

    @Test
    public void getUserOneTest() {
        int id = 1;
        User u = userService.getUserById(id);
        System.out.println(u.getName());
    }
}

 

10、在service層修改 DBContextHolder.setDBType()來看看效果。

 

四、遇到的問題：

　　1、遇到Spring的PropertyPlaceholderConfigurer不起效果，在數據源配置的${jdbc.driver}中獲取不到jdbc.properties中的值。

<bean id="propertyConfigurer" class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">
      <property name="locations">
          <list>
              <value>classpath*:jdbc.properties</value>
          </list>
      </property>
      <property name="fileEncoding" value="UTF-8" />
      <property name= "ignoreResourceNotFound" value="false"/>
</bean>

 

　　解決：

<!--
　　原來id的名稱是sqlSessionFactory,但是在spring里使用org.mybatis.spring.mapper.MapperScannerConfigurer
   進行自動掃描的時候，設置了sqlSessionFactory 的話，他會優先於PropertyPlaceholderConfigurer執行。
   從而導致PropertyPlaceholderConfigurer失效，
   這時在xml中用${url}、${username}、${password}等這樣之類的表達式，
   將無法獲取到properties文件里的內容。
-->
    <bean id="dynamicsqlSessionFactory"
          class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="dynamicDataSource"/>
        <property name="mapperLocations">
            <array>
                <value>classpath:mappers/*Mapper.xml</value>
            </array>
        </property>
        <property name="configLocation" value="classpath:mybatis-config.xml"/>
        <property name="typeAliasesPackage" value="org.hope.lee.model"/>
    </bean>

 六、還有一種方式是使用mysql自帶的replicationDriver來實現讀寫分離。大家自己也可以試試

　　　　　　http://blog.csdn.net/lixiucheng005/article/details/17391857

https://gitee.com/huayicompany/mybatis-learn/tree/master/separated-read-write

參考：

[1] 博客，http://blog.csdn.net/xtj332/article/details/43953699

[2] 博客，http://blog.csdn.net/keda8997110/article/details/16827215