Spring Boot教程（三十二）多數據源配置與使用（1）

以前在介紹使用JdbcTemplate和Spring-data-jpa時，都使用了單數據源。在單數據源的狀況下，Spring Boot的配置很是簡單，只須要在application.properties文件中配置鏈接參數便可。可是每每隨着業務量發展，咱們一般會進行數據庫拆分或是引入其餘數據庫，從而咱們須要配置多個數據源，下面基於以前的JdbcTemplate和Spring-data-jpa例子分別介紹兩種多數據源的配置方式。

多數據源配置

建立一個Spring配置類，定義兩個DataSource用來讀取application.properties中的不一樣配置。以下例子中，主數據源配置爲spring.datasource.primary開頭的配置，第二數據源配置爲spring.datasource.secondary開頭的配置。

@Configuration
public class DataSourceConfig {

    @Bean(name = "primaryDataSource")
    @Qualifier("primaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondaryDataSource")
    @Qualifier("secondaryDataSource")
    @Primary
    @ConfigurationProperties(prefix="spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create().build();
    }

}

對應的application.properties配置以下：

spring.datasource.primary.url=jdbc:mysql://localhost:3306/test1
spring.datasource.primary.username=root
spring.datasource.primary.password=root
spring.datasource.primary.driver-class-name=com.mysql.jdbc.Driver

spring.datasource.secondary.url=jdbc:mysql://localhost:3306/test2
spring.datasource.secondary.username=root
spring.datasource.secondary.password=root
spring.datasource.secondary.driver-class-name=com.mysql.jdbc.Driver

　　

JdbcTemplate支持

對JdbcTemplate的支持比較簡單，只須要爲其注入對應的datasource便可，以下例子，在建立JdbcTemplate的時候分別注入名爲primaryDataSource和secondaryDataSource的數據源來區分不一樣的JdbcTemplate。

@Bean(name = "primaryJdbcTemplate")
public JdbcTemplate primaryJdbcTemplate(
        @Qualifier("primaryDataSource") DataSource dataSource) {
    return new JdbcTemplate(dataSource);
}

@Bean(name = "secondaryJdbcTemplate")
public JdbcTemplate secondaryJdbcTemplate(
        @Qualifier("secondaryDataSource") DataSource dataSource) {
    return new JdbcTemplate(dataSource);
}

接下來經過測試用例來演示如何使用這兩個針對不一樣數據源的JdbcTemplate。

@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(Application.class)
public class ApplicationTests {

	@Autowired
	@Qualifier("primaryJdbcTemplate")
	protected JdbcTemplate jdbcTemplate1;

	@Autowired
	@Qualifier("secondaryJdbcTemplate")
	protected JdbcTemplate jdbcTemplate2;

	@Before
	public void setUp() {
		jdbcTemplate1.update("DELETE  FROM  USER ");
		jdbcTemplate2.update("DELETE  FROM  USER ");
	}

	@Test
	public void test() throws Exception {

		// 往第一個數據源中插入兩條數據
		jdbcTemplate1.update("insert into user(id,name,age) values(?, ?, ?)", 1, "aaa", 20);
		jdbcTemplate1.update("insert into user(id,name,age) values(?, ?, ?)", 2, "bbb", 30);

		// 往第二個數據源中插入一條數據，若插入的是第一個數據源，則會主鍵衝突報錯
		jdbcTemplate2.update("insert into user(id,name,age) values(?, ?, ?)", 1, "aaa", 20);

		// 查一下第一個數據源中是否有兩條數據，驗證插入是否成功
		Assert.assertEquals("2", jdbcTemplate1.queryForObject("select count(1) from user", String.class));

		// 查一下第一個數據源中是否有兩條數據，驗證插入是否成功
		Assert.assertEquals("1", jdbcTemplate2.queryForObject("select count(1) from user", String.class));

	}


}

源碼來源