• 极客专栏正式上线!欢迎访问 https://www.jikewenku.com/topic.html
  • 极客专栏正式上线!欢迎访问 https://www.jikewenku.com/topic.html

Spring Profile 和 Mybatis 进行多个数据源的切换

技术杂谈 勤劳的小蚂蚁 2个月前 (01-30) 69次浏览 已收录 0个评论 扫描二维码

最近在做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”>
       <property name=”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”
 
<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″?>
       xsi:schemaLocation=”http://www.springframework.org/schema/beans
 
    <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>
        <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>
        <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的目的!

参考资料



丨极客文库, 版权所有丨如未注明 , 均为原创丨
本网站采用知识共享署名-非商业性使用-相同方式共享 3.0 中国大陆许可协议进行授权
转载请注明原文链接:Spring Profile 和 Mybatis 进行多个数据源的切换
喜欢 (0)
[247507792@qq.com]
分享 (0)
勤劳的小蚂蚁
关于作者:
温馨提示:本文来源于网络,转载文章皆标明了出处,如果您发现侵权文章,请及时向站长反馈删除。

您必须 登录 才能发表评论!

  • 精品技术教程
  • 编程资源分享
  • 问答交流社区
  • 极客文库知识库

客服QQ


QQ:2248886839


工作时间:09:00-23:00