ibatis简单读写分离方案

每个公司有年头的公司，都会有一些有年头的项目，需要有年头的人来维护。

场景，我们数据库操作读写频繁，生产采用oracle RAC集群，同时实时同步一个只读备库。

今天要维护的项目是内部管理系统，因为业务原因和一切从简原则，我们的应用直联生产主库。扛过单日100w笔的交易，数据库也是比较平稳。

近期业务部门需要频繁下载大量数据，考虑到会影响生产库IO，打算单独把下载功能走只读备库。现有内管项目采用strust1 + ibatis（这是多么老的搭档啊），那么我们需要修改ibatis实现可控的读写分离。

ibatis配置文件有两个，sql-map-config.xml配置数据源和sqlmap信息；dao.xml集成sql-map-config.xml同时指定dao和实现类的映射。

DaoManager，从名字上也可以看出，它一个dao管理类，管理数据源，dao等等。

# 获取daoManager
Resources.setCharset(Charset.forName("UTF-8"));
Reader reader = Resources.getResourceAsReader("com/test/dao/dao.xml");
DaoManager daoManager = DaoManagerBuilder.buildDaoManager(reader, props);

我打算在这个地方做做文章，想办法把dataSource区分主库和只读库。通过跟踪代码发现，DaoManagerBuilder.buildDaoManager(reader, props)有一个props可以传入，目测可以动态替换xml的配置参数。

通过查阅年久失修的API文档，发现该方法的注释写着：

public static DaoManager buildDaoManager(java.io.Reader reader,
                                         java.util.Properties props)
                                  throws DaoException

Builds a DAO Manager from the reader supplied.

Parameters:
reader - A Reader instance that reads a DAO configuration file (dao.xml).
props - A Properties instance that will be used for parameter substitution in the DAO configuration file (dao.xml).

**重点*：该props对象将被用作替换dao.xml里的参数*

Returns:
A configured DaoManager instance

Throws:
DaoException

意思就是说，我们可以在xml里面通过${}定义自己的变量，通过props定义变量的值，从而实现动态的修改。

如下，我们先自定义dataSource的主要变量：

<dataSource type="DBCP">
    <property name="JDBC.Driver" value="${JDBC.Driver}" />
    <property name="JDBC.ConnectionURL" value="${JDBC.ConnectionURL}" />
    <property name="JDBC.Username" value="${JDBC.Username}" />
    <property name="JDBC.Password" value="${JDBC.Password}" />
</dataSource>

然后修改获取daoManager的方法：

private static DaoManager buildDaoManager(String prefix) {
    try {
        Resources.setCharset(Charset.forName("UTF-8"));
        Reader reader = Resources.getResourceAsReader("com/test/dao/dao.xml");
        Properties p = new Properties();
        p.setProperty("JDBC.Driver", getValue(prefix + "driver"));
        p.setProperty("JDBC.ConnectionURL", getValue(prefix + "url"));
        p.setProperty("JDBC.Username", getValue(prefix + "username"));
        p.setProperty("JDBC.Password", getValue(prefix + "password"));
        return DaoManagerBuilder.buildDaoManager(reader, p);
    } catch (Exception e) {
        logger.error("DaoConfig happen Exception", e);
        return null;
    }
}


----

/**
 * 在jdbc.properties加载拼接需要的参数
 * @param key
 * @return
 * @throws NullPointerException
 * @throws IOException
 */
private static String getValue(String key) throws NullPointerException, IOException {
    if (jdbc_props.isEmpty()) {
        jdbc_props.load(DaoConfig.class.getClassLoader().getResourceAsStream(jdbc_config));
    }
    String value = jdbc_props.getProperty(key);
    if (Strings.isNullOrEmpty(value)) {
        throw new NullPointerException("jdbc配置：" + key + " 为空");
    }
    return "${" + value.trim() + "}";
}

其中，prefix用来区分配置文件的类型，包括：master. 和 readonly. ，在这里我用jdbc.properties存放数据库连接信息，大概如下：

# master
master.driver=oracle.jdbc.driver.OracleDriver
master.url=jdbc:oracle:thin:@yourip:1521/yourSID
master.username=your username
master.password=your pwd


# read db
readonly.driver=oracle.jdbc.driver.OracleDriver
readonly.url=jdbc:oracle:thin:@yourip:1521/yourSID
readonly.username=your username
readonly.password=your pwd

然后，我通过ThreaLocal存储当前线程是否为下载（只读）操作，给service返回不同的daoManager，以此达到目的。

    public static DaoManager getDaoManager() {
        if (DBStatus.isReadOnly() && null != readOnlyDaoManager) {
            return readOnlyDaoManager;
        } else {
            return daoManager;
        }
    }

---

Spring + MyBatis 实现手动控制读写分离，可自定义类扩展AbstractRoutingDataSource类时重写determineCurrentLookupKey()方法来实现数据源切换功能。
可供参考：https://o7planning.org/en/10869/using-multiple-datasources-with-spring-boot-and-routingdatasource