最近在做WebMagic的后台，遇到一个问题：后台用到了数据库，本来理想情况下是用Mysql，但是为了做到开箱即用，也整合了一个嵌入式 数据库H2。这里面就有个问题了，如何用一套代码，提供对Mysql和H2两种方案的支持？博主收集了一些资料，也调试了很久，终于找到一套可行方案，记 录下来。代码贴的有点多，主要是为了以后方便自己查找。

H2的使用

H2是一个嵌入式，纯Java实现的数据库，它各方面都要好于Java的sqlitejdbc。它可以使用内存模式，也可以使用磁盘模式。具体使用可以看攻略：

为MyBatis同时配置两套数据源

我们希望达到的效果是，不同的数据源使用不同的sql，并且这个切换最好只在配置中体现，与代码无关。所以我们选择xml的方式编写sql语句。

MyBatis Spring的使用

同时使用Mybatis-Spring插件，这样Mybatis可以将Mapper（也就是DAO）自动配置成Bean，非常方便。它的一个完整示例可以看这个项目： 。这里我配置如下：

配置Bean

配置Mapper

对应的DAO和配置文件如下：

• us.codecraft.webmagic.dao.DynamicClassDao:

public interface DynamicClassDao {    public int add(DynamicClass dynamicClass); }

• DynamicClassDao.xml

         
      
        insert into DynamicClass (`ClassName`,`SourceCode`,`AddTime`,`UpdateTime`) values (#{className},#{sourceCode},now(),now()) 
       
     

使用databaseIdProvider进行多个数据源的SQL切换

MyBatis支持根据不同的数据库名来进行SQL语句的切换。做法是初始化 SqlSessionFactoryBean 的时候，配置一个 databaseIdProvider 。

      
       
        
        
         sqlserver
         
        
         db2
         
        
         oracle
         
        
         mysql
         
        
         h2
         
        
       
      
      
       
      
      
       
       
       
     

然后在Mapper的xml里，把相应的语句加上 databaseId="xxx" 就可以了。

         
      
        insert into DynamicClass (`ClassName`,`SourceCode`,`AddTime`,`UpdateTime`) values (#{className},#{sourceCode},now(),now()) 
       
      
        insert into DynamicClass (`ClassName`,`SourceCode`,`AddTime`,`UpdateTime`) values (#{className},#{sourceCode},now(),now()) 
       
     

Spring Profile

Profile是Spring 3.1后新增的特性，简单来说，就是根据不同的环境，读取不同的配置。这些配置可以放在一起，但是单独生效。贴个代码吧，很容易说明问题了：

设置Profile有不同的方式。

在JUnit里面，使用注解@ActiveProfile即可。

@RunWith(SpringJUnit4ClassRunner.class)@ContextConfiguration(locations = {
      "classpath*:/config/spring/applicationContext*.xml"})@ActiveProfiles("test") @Transactional public abstract class AbstractTest { }

Web项目则是在web.xml里设置：

       
      
       spring.profiles.active
       
      
       product
       
     

如此配置，就可以达到在不同的环境使用不同bean的目的！

 

 

二、在springboot中，可以使用@ConfigurationProperties(prefix = "xxx") 在不同环境下，配置不同的数据源

参考资料

• H2数据库攻略
• spring+mybatis 多数据源整合
• 如何用Spring 3.1的Environment和Profile简化工作