最近在做WebMagic的後臺，遇到一個問題：後臺用到了數據庫，本來理想情況下是用Mysql，但是為了做到開箱即用，也整合了一個嵌入式數據庫H2。這裏面就有個問題了，如何用一套代碼，提供對Mysql和H2兩種方案的支持？博主收集了一些資料，也調試了很久，終於找到一套可行方案，記錄下來。代碼貼的有點多，主要是為了以後方便自己查找。

H2的使用

H2是一個嵌入式，純Java實現的數據庫，它各方面都要好於Java的sqlitejdbc。它可以使用內存模式，也可以使用磁盤模式。具體使用可以看攻略：

http://www.cnblogs.com/gao241/p/3480472.html

為MyBatis同時配置兩套數據源

我們希望達到的效果是，不同的數據源使用不同的sql，並且這個切換最好只在配置中體現，與代碼無關。所以我們選擇xml的方式編寫sql語句。

MyBatis Spring的使用
同時使用Mybatis-Spring插件，這樣Mybatis可以將Mapper（也就是DAO）自動配置成Bean，非常方便。它的一個完整示例可以看這個項目：https://github.com/mybatis/jpetstore-6。這裏我配置如下：

配置Bean

<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
<property name="basePackage" value="us.codecraft.webmagic.dao" />

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

us.codecraft.webmagic.dao.DynamicClassDao:

public interface DynamicClassDao {

public int add(DynamicClass dynamicClass);

}
DynamicClassDao.xml

<mapper namespace="us.codecraft.webmagic.dao.DynamicClassDao">

<insert id="add" parameterType="us.codecraft.webmagic.model.DynamicClass">
  insert into DynamicClass (`ClassName`,`SourceCode`,`AddTime`,`UpdateTime`)
  values (#{className},#{sourceCode},now(),now())
</insert>

</mapper>
使用databaseIdProvider進行多個數據源的SQL切換
MyBatis支持根據不同的數據庫名來進行SQL語句的切換。做法是初始化SqlSessionFactoryBean的時候，配置一個databaseIdProvider。

<bean id="vendorProperties" class="org.springframework.beans.factory.config.PropertiesFactoryBean">
<property name="properties">
<props>
<prop key="SQL Server">sqlserver</prop>
<prop key="DB2">db2</prop>
<prop key="Oracle">oracle</prop>
<prop key="MySQL">mysql</prop>
<prop key="H2">h2</prop>
</props>
</property>
</bean>

<bean id="databaseIdProvider" class="org.apache.ibatis.mapping.VendorDatabaseIdProvider">
<property name="properties" ref="vendorProperties"/>
</bean>

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource" />
<property name="databaseIdProvider" ref="databaseIdProvider" />
<property name="mapperLocations" value="classpath*:/config/mapper/*/.xml" />
</bean>
然後在Mapper的xml裏，把相應的語句加上databaseId="xxx"就可以了。

<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">

<mapper namespace="us.codecraft.webmagic.dao.DynamicClassDao">

<insert id="add" parameterType="us.codecraft.webmagic.model.DynamicClass" databaseId="mysql">
  insert into DynamicClass (`ClassName`,`SourceCode`,`AddTime`,`UpdateTime`)
  values (#{className},#{sourceCode},now(),now())
</insert>

<insert id="add" parameterType="us.codecraft.webmagic.model.DynamicClass" databaseId="h2">
  insert into DynamicClass (`ClassName`,`SourceCode`,`AddTime`,`UpdateTime`)
  values (#{className},#{sourceCode},now(),now())
</insert>

</mapper>

Spring Profile

Profile是Spring 3.1後新增的特性，簡單來說，就是根據不同的環境，讀取不同的配置。這些配置可以放在一起，但是單獨生效。貼個代碼吧，很容易說明問題了：

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:jdbc="http://www.springframework.org/schema/jdbc"
xsi:schemaLocation="http://www.springframework.org/schema/beans
http://www.springframework.org/schema/beans/spring-beans-4.0.xsd http://www.springframework.org/schema/jdbc http://www.springframework.org/schema/jdbc/spring-jdbc.xsd">

<bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource"
      destroy-method="close">
    <property name="driverClassName" value="com.mysql.jdbc.Driver"/>
    <property name="url" value="jdbc:mysql://127.0.0.1:3306/WebMagic?characterEncoding=UTF-8"/>
    <property name="username" value="webmagic"/>
    <property name="password" value="webmagic"/>
</bean>

<beans profile="test">
    <bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource"
          destroy-method="close">
        <property name="driverClassName" value="org.h2.Driver"/>
        <property name="url" value="jdbc:h2:mem:WebMagic;DB_CLOSE_DELAY=-1"/>
    </bean>
    <!--Refer to https://github.com/springside/springside4/wiki/H2-Database -->
    <jdbc:initialize-database data-source="dataSource" ignore-failures="ALL">
        <jdbc:script location="classpath:sql/h2/schema.sql" />
        <!--<jdbc:script location="classpath:data/h2/import-data.sql" encoding="UTF-8"/>-->
    </jdbc:initialize-database>
</beans>

<beans profile="standalone">
    <bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource"
          destroy-method="close">
        <property name="driverClassName" value="org.h2.Driver"/>
        <property name="url" value="jdbc:h2:file:~/.h2/WebMagic;AUTO_SERVER=TRUE"/>
    </bean>
    <!--Refer to https://github.com/springside/springside4/wiki/H2-Database -->
    <jdbc:initialize-database data-source="dataSource" ignore-failures="ALL">
        <jdbc:script location="classpath:sql/h2/schema.sql" />
        <!--<jdbc:script location="classpath:data/h2/import-data.sql" encoding="UTF-8"/>-->
    </jdbc:initialize-database>
</beans>

</beans>
設置Profile有不同的方式。

在JUnit裏面，使用註解@ActiveProfile即可。

@RunWith(SpringJUnit4Cla***unner.class)
@ContextConfiguration(locations = {"classpath:/config/spring/applicationContext.xml"})@ActiveProfiles("test")
br/>@ActiveProfiles("test")
public abstract class AbstractTest {
}
Web項目則是在web.xml裏設置：

<init-param>
<param-name>spring.profiles.active</param-name>
<param-value>product</param-value>
</init-param>

總結：

如此配置，就可以達到在不同的環境使用不同bean的目的！
參考資料
H2數據庫攻略 http://www.cnblogs.com/gao241/p/3480472.html
spring+mybatis 多數據源整合 http://blog.csdn.net/fhx007/article/details/12530735
如何用Spring 3.1的Environment和Profile簡化工作 http://www.importnew.com/1099.html

Spring Profile和Mybatis進行多個數據源（H2和Mysql）的切換