Spring Boot配置多數(shù)據(jù)源并應(yīng)用于JdbcTemplate

管理

編輯

刪除

CRMEB 2025-01-17 09:17:24

暫不處理

多數(shù)據(jù)源配置

創(chuàng)建一個(gè)Spring配置類，定義兩個(gè)DataSource用來讀取application.properties中的不同配置。如下例子中，主數(shù)據(jù)源配置為spring.datasource.primary開頭的配置，第二數(shù)據(jù)源配置為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();
    }

}

對(duì)應(yīng)的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支持

對(duì)JdbcTemplate的支持比較簡單，只需要為其注入對(duì)應(yīng)的datasource即可，如下例子，在創(chuàng)建JdbcTemplate的時(shí)候分別注入名為primaryDataSource和secondaryDataSource的數(shù)據(jù)源來區(qū)分不同的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);
    }

接下來通過測試用例來演示如何使用這兩個(gè)針對(duì)不同數(shù)據(jù)源的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 {

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

		// 往第二個(gè)數(shù)據(jù)源中插入一條數(shù)據(jù)，若插入的是第一個(gè)數(shù)據(jù)源，則會(huì)主鍵沖突報(bào)錯(cuò)
		jdbcTemplate2.update("insert into user(id,name,age) values(?, ?, ?)", 1, "aaa", 20);

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

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

	}


}