从零开发分布式数据库中间件 二、构建MyBatis的读写分离数据库中间件

<h1>
    <span class="link_title"><a href="/happyheng/article/details/54628737">
    从零开发分布式数据库中间件 二、构建MyBatis的读写分离数据库中间件        
       
    </a>
    </span>

     
</h1>

  在上一节  从零开发分布式数据库中间件 一、读写分离的数据库中间件 中，我们讲了如何通过ThreadLocal来指定每次访问的数据源，并通过jdbc的连接方式来切换数据源，那么这一节我们使用我们常用的数据库持久层框架MyBatis来实现数据库读写分离。

一、数据源代理：

此类与上一节相似，即可以指定当前线程访问的数据源。

package com.happyheng.datasource;

/**
 * 数据源代理设置
 * Created by happyheng on 17/1/15.
 */
public class DataSourceProxy {

    private static ThreadLocal<DataSourceEnum> threadLocal = new ThreadLocal<>();

    public enum DataSourceEnum {
        MASTER,
        SLAVE
    }

    /**
     * 为当前线程设置数据源
     */
    public static void setDataSource(DataSourceEnum sourceEnum) {
        threadLocal.set(sourceEnum);
    }

    public static DataSourceEnum getDataSource() {
        return threadLocal.get();
    }

}

package com.happyheng.datasource;
/**


数据源代理设置


Created by happyheng on 17/1/15.

*/

public class DataSourceProxy {
private static ThreadLocal<DataSourceEnum> threadLocal = new ThreadLocal<>();
public enum DataSourceEnum {

MASTER,

SLAVE

}
/**

为当前线程设置数据源

*/

public static void setDataSource(DataSourceEnum sourceEnum) {

threadLocal.set(sourceEnum);

}

public static DataSourceEnum getDataSource() {

return threadLocal.get();

}


}

二、数据源Map：

  首先我们需要将我们的读写数据源都写入到配置文件中，并设置到继承了AbstractRoutingDataSource抽象类的子类中，接下来我们会讲解AbstractRoutingDataSource的作用：

<bean id="masterDataSource" class="org.apache.commons.dbcp.BasicDataSource"
      destroy-method="close">
    <property name="driverClassName" value="${master.driver}" />
    <property name="url" value="${master.dburl}" />
    <property name="username" value="${master.user}" />
    <property name="password" value="${master.password}" />
</bean>

<bean id="slaveDataSource1" class="org.apache.commons.dbcp.BasicDataSource"
      destroy-method="close">
    <property name="driverClassName" value="${slave1.driver}" />
    <property name="url" value="${slave1.dburl}" />
    <property name="username" value="${slave1.user}" />
    <property name="password" value="${slave1.password}" />
</bean>

<bean id="slaveDataSource2" class="org.apache.commons.dbcp.BasicDataSource"
      destroy-method="close">
    <property name="driverClassName" value="${slave2.driver}" />
    <property name="url" value="${slave2.dburl}" />
    <property name="username" value="${slave2.user}" />
    <property name="password" value="${slave2.password}" />
</bean>

<bean id="dataSource" class="com.happyheng.datasource.OptionalDataSource" >
    <!-- 通过key-value的形式来关联数据源 -->
    <property name="targetDataSources">
        <map>
            <entry key="masterDataSource" value-ref="masterDataSource" />
            <entry key="slaveDataSource1" value-ref="slaveDataSource1" />
            <entry key="slaveDataSource2" value-ref="slaveDataSource2" />
        </map>
    </property>
    <property name="defaultTargetDataSource" ref="masterDataSource" />
</bean>

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
    <property name="dataSource" ref="dataSource" />
    <property name="mapperLocations" value="classpath*:mybatis/**/*Mapper.xml"/>
</bean>

    <bean id="masterDataSource" class="org.apache.commons.dbcp.BasicDataSource"
          destroy-method="close">
        <property name="driverClassName" value="${master.driver}" />
        <property name="url" value="${master.dburl}" />
        <property name="username" value="${master.user}" />
        <property name="password" value="${master.password}" />
    </bean>
&lt;bean id="slaveDataSource1" class="org.apache.commons.dbcp.BasicDataSource"
      destroy-method="close"&gt;
    &lt;property name="driverClassName" value="${slave1.driver}" /&gt;
    &lt;property name="url" value="${slave1.dburl}" /&gt;
    &lt;property name="username" value="${slave1.user}" /&gt;
    &lt;property name="password" value="${slave1.password}" /&gt;
&lt;/bean&gt;

&lt;bean id="slaveDataSource2" class="org.apache.commons.dbcp.BasicDataSource"
      destroy-method="close"&gt;
    &lt;property name="driverClassName" value="${slave2.driver}" /&gt;
    &lt;property name="url" value="${slave2.dburl}" /&gt;
    &lt;property name="username" value="${slave2.user}" /&gt;
    &lt;property name="password" value="${slave2.password}" /&gt;
&lt;/bean&gt;

&lt;bean id="dataSource" class="com.happyheng.datasource.OptionalDataSource" &gt;
    &lt;!-- 通过key-value的形式来关联数据源 --&gt;
    &lt;property name="targetDataSources"&gt;
        &lt;map&gt;
            &lt;entry key="masterDataSource" value-ref="masterDataSource" /&gt;
            &lt;entry key="slaveDataSource1" value-ref="slaveDataSource1" /&gt;
            &lt;entry key="slaveDataSource2" value-ref="slaveDataSource2" /&gt;
        &lt;/map&gt;
    &lt;/property&gt;
    &lt;property name="defaultTargetDataSource" ref="masterDataSource" /&gt;
&lt;/bean&gt;

&lt;bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"&gt;
    &lt;property name="dataSource" ref="dataSource" /&gt;
    &lt;property name="mapperLocations" value="classpath*:mybatis/**/*Mapper.xml"/&gt;
&lt;/bean&gt;</pre><br>




二、AbstractRoutingDataSource数据源路由类：
  在MyBatis中，需从SqlSessionFactory中获取dao文件，而SqlSessionFactory即需要数据源，因为我们需要根据不同的情况来选定数据源，所以不能写死一个数据源，而是应该将数据源写入到AbstractRoutingDataSource中的map中，AbstractRoutingDataSource即能够根据不同的情况指定访问的数据源。
  还有需要注意的是，AbstractRoutingDataSource是一个抽象类，需要实现其determineCurrentLookupKey方法，来指定每次访问数据库的数据源。
下为继承了AbstractRoutingDataSource的OptionalDataSource类：
package com.happyheng.datasource;

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

/**
 * Created by happyheng on 17/1/10.
 */
public class OptionalDataSource extends AbstractRoutingDataSource {

    // 数据源
    private String masterDataSource = "masterDataSource";
    private String[] slaveDataSource = {"slaveDataSource1", "slaveDataSource2"};

    @Override
    protected Object determineCurrentLookupKey() {


        DataSourceProxy.DataSourceEnum dataSourceEnum = DataSourceProxy.getDataSource();

        if (dataSourceEnum == DataSourceProxy.DataSourceEnum.SLAVE) {

            double random = Math.random();
            int randomIndex = (int)(random * slaveDataSource.length);

            System.out.println("访问的是从数据库" + (randomIndex + 1));
            return slaveDataSource[randomIndex];
        } else {

            System.out.println("访问的是主数据库");
            return masterDataSource;
        }
    }
}
package com.happyheng.datasource;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**


Created by happyheng on 17/1/10.

*/

public class OptionalDataSource extends AbstractRoutingDataSource {
// 数据源

private String masterDataSource = "masterDataSource";

private String[] slaveDataSource = {"slaveDataSource1", "slaveDataSource2"};
@Override

protected Object determineCurrentLookupKey() {
 DataSourceProxy.DataSourceEnum dataSourceEnum = DataSourceProxy.getDataSource();

 if (dataSourceEnum == DataSourceProxy.DataSourceEnum.SLAVE) {

     double random = Math.random();
     int randomIndex = (int)(random * slaveDataSource.length);

     System.out.println("访问的是从数据库" + (randomIndex + 1));
     return slaveDataSource[randomIndex];
 } else {

     System.out.println("访问的是主数据库");
     return masterDataSource;
 }

}

}