spring boot整合mybatis多数据源、使用自定义数据源

无论是多数据源还是自定义数据源，大体流程都差不多。核心都是通过spring boot的自定义配置来实现的。

直接上代码。解释都在代码里。

package com.zb.test;

import javax.sql.DataSource;

import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.annotation.EnableTransactionManagement;

/**
 * 读取数据配置表
 * 
 * @author zb
 * @see springboot 默认只能读一个库，这里要读多个库,要自己写。
 */
// 说明这是一个配置文件，等同于spring.xml中的bean
@org.springframework.context.annotation.Configuration
@MapperScan(
		// 指定数据实现存储
		basePackages = { "com.zb.dao.testdata" },
		// 关键：指定Factory，否则多个数据源，代码不知道用哪个会报错
		sqlSessionFactoryRef = "sqlSessionFactory",
		// 关键：指定Template，否则多个数据源，代码不知道用哪个会报错
		sqlSessionTemplateRef = "sqlSessionTemplate")
// 开启事务
@EnableTransactionManagement(proxyTargetClass = true)
public class DataSourceConfigGamedata {

	@Bean(name = "dataSource")
	// prefix值必须是application.yml中对应属性的前缀
	@ConfigurationProperties(prefix = "spring.datasource.testdata")
	public DataSource dataSource() {
		DataSourceBuilder<?> builder = DataSourceBuilder.create();
		// 这里的场景适合用mybatis自带Unpooled连接池，特点：链接用时打开，用完即关闭
		builder.type(UnpooledDataSource.class);
		return builder.build();
	}

	@Bean(name = "sqlSessionFactory")
	public SqlSessionFactory sqlSessionFactory(@Qualifier("dataSource") DataSource dataSource) throws Exception {
		SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
		bean.setDataSource(dataSource);

		// 添加XML目录
		ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
		try {
			bean.setMapperLocations(resolver.getResources("classpath*:mapper/*.xml"));
			return bean.getObject();
		} catch (Exception e) {
			e.printStackTrace();
			throw new RuntimeException(e);
		}
	}

	@Bean(name = "sqlSessionTemplate")
	public SqlSessionTemplate sqlSessionTemplate(@Qualifier("sqlSessionFactory") SqlSessionFactory sqlSessionFactory)
			throws Exception {
		SqlSessionTemplate template = new SqlSessionTemplate(sqlSessionFactory); // 使用上面配置的Factory
		return template;
	}

	// 使用方式：@Transactional(transactionManager="gamedataTransactionManager")
	@Bean(name = "gamedataTransactionManager")
	public PlatformTransactionManager gamedataTransactionManager(@Qualifier("dataSource") DataSource oracleDataSource) {
		return new DataSourceTransactionManager(oracleDataSource);
	}
}

重点在于dataSource方法中的 builder.type(UnpooledDataSource.class);这里绑定的是什么数据源，就会使用什么连接池。比如我这里的使用的UnpooledDataSource绑定的是mybatis自带的一个连接池。这些数据源绑定的都是DataSource接口，自己实现一个其实也可。

这里还有一个关键点：在yml中要设置数据源的属性。需要设置什么属性去看对数据源的set方法。找到对应名字设置即可。

Bean名字多个数据源要用不同的名字。否则会冲突。@Qualifier中也要用对应Bean的名字。

@MapperScan中的（sqlSessionFactoryRef，sqlSessionTemplateRef）也要取对应Bean的名字。