使用Spring Profile和Mybatis進行多個數據源(H2和Mysql)的切換
轉載自:http://my.oschina.net/flashsword/blog/209872
最近在做WebMagic的後臺,遇到一個問題:後臺用到了資料庫,本來理想情況下是用Mysql,但是為了做到開箱即用,也整合了一個嵌入式資料庫H2。這裡面就有個問題了,如何用一套程式碼,提供對Mysql和H2兩種方案的支援?博主收集了一些資料,也除錯了很久,終於找到一套可行方案,記錄下來。程式碼貼的有點多,主要是為了以後方便自己查詢。
H2的使用
H2是一個嵌入式,純Java實現的資料庫,它各方面都要好於Java的sqlitejdbc。它可以使用記憶體模式,也可以使用磁碟模式。具體使用可以看攻略:
為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"> <propertyname="basePackage" value="us.codecraft.webmagic.dao" /> </bean> <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"> <property name="dataSource" ref="dataSource" /> <property name="mapperLocations" value="classpath*:/config/mapper/**/*.xml"/> </bean>
配置Mapper
對應的DAO和配置檔案如下:
- 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(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath*:/config/spring/applicationContext*.xml"})
@ActiveProfiles("test")
@Transactional
public abstract class AbstractTest {
}
Web專案則是在web.xml裡設定:
<init-param> <param-name>spring.profiles.active</param-name> <param-value>product</param-value> </init-param>
如此配置,就可以達到在不同的環境使用不同bean的目的!