快捷導(dǎo)航

Spring中的動(dòng)態(tài)數(shù)據(jù)源解讀

更新時(shí)間：2023年06月27日 09:16:24 作者：daliucheng

這篇文章主要介紹了關(guān)于Spring中的動(dòng)態(tài)數(shù)據(jù)源解讀，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

原理

平常在使用Mysql的時(shí)候是通過(guò)JDBC的，得給一個(gè)url，userName，和password，如下：

jdbc:mysql://localhost:3306/t_db1?useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=UTC

一個(gè)url對(duì)應(yīng)一個(gè)Connection對(duì)象，需要在url中指定需要連接的庫(kù)。

之后的Mysql的CRUD的操作都是通過(guò)Connection對(duì)象來(lái)做的。所以，動(dòng)態(tài)，就是在這個(gè)時(shí)候操作的，在獲取Connection的時(shí)候來(lái)通過(guò)指定的key來(lái)判斷要用哪個(gè)數(shù)據(jù)源。這得有一個(gè)前提，得先建立Connection，之后在獲取Connection的時(shí)候在判斷，

1.通過(guò)什么判斷呢？存放key和Connection的數(shù)據(jù)結(jié)構(gòu)是什么？

只要指定一個(gè)回調(diào)方法，key隨便，存放key和Connection的數(shù)據(jù)結(jié)構(gòu)是Map。在獲取Connection的時(shí)候，先確定Key，在獲取Connection就好了。

再看Spring

Java連接Mysql指定了接口，需要實(shí)現(xiàn)DataSource接口，Spring中提供了抽象類（AbstractDataSource）

解釋說(shuō)明

DriverManagerDataSource

這個(gè)沒(méi)有什么可說(shuō)，只是封裝了一下連接mysql需要用的一些屬性，比如userName,password,url,driver，就是就老一套的獲取Connection封裝了一下

AbstractRoutingDataSource

Spring提供的動(dòng)態(tài)數(shù)據(jù)源的抽象類

首先，它里面有一個(gè)Map，Map的key是Object（這是自定義的），V是Object（雖然是Object，代碼里面規(guī)定了，只能為String，和DataSource），其實(shí)它本質(zhì)就應(yīng)該是DataSource，這里是String的目的是為了可以通過(guò)DataSourceLookup來(lái)獲取數(shù)據(jù)源。（比如，可以在配置數(shù)據(jù)源的時(shí)候，直接配置數(shù)據(jù)源的bean的名字，然后自己寫(xiě)一個(gè)DataSourceLookup的實(shí)現(xiàn)類，從BeanFactory中獲取DataSource）。

還提供了一個(gè)determineCurrentLookupKey（）方法來(lái)判斷當(dāng)前數(shù)據(jù)源的key，其實(shí)就是map中的key。在最終獲取Connection的時(shí)候通過(guò)當(dāng)前的key獲取DataSource,在通過(guò)DataSource獲取真正的Connection。

代碼分析

1.屬性

2.實(shí)現(xiàn)接口

實(shí)現(xiàn)了InitializingBean那它肯定在afterPropertiesSet會(huì)做操作。等會(huì)看看

3.重要方法

afterPropertiesSet

獲取Connection之前確定Key

`determineCurrentLookupKey`方法是留給子類拓展的。

determineCurrentLookupKey怎么來(lái)確定key呢？

它是一個(gè)無(wú)參的方法，一般來(lái)說(shuō)，都是放在ThreadLocal中，在執(zhí)行sql操作之前，在對(duì)應(yīng)的ThreadLocal放這次需要的key，就可以在determineCurrentLookupKey中獲取ThreadLocal中的值，確定key。

<font color='red'>Spring提供了一個(gè)動(dòng)態(tài)數(shù)據(jù)源的實(shí)現(xiàn)類，`IsolationLevelDataSourceRouter`,key是事務(wù)的隔離級(jí)別。意思就是可以根據(jù)不同的隔離級(jí)別來(lái)選擇DataSource</font>

還有，一般都是用切面來(lái)操作ThreadLocal的

例子

我這里為了簡(jiǎn)單，就不寫(xiě)切面了。直接代碼cv。

1.配置類

package datasource.dynamic;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;
@Configuration
public class Config {
	@Bean
	public DriverManagerDataSource dataSource1(){
		String url = "jdbc:mysql://localhost:3306/t_db1?useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=UTC";
		String userName = "root";
		String password = "123456";
		return new DriverManagerDataSource(url,userName,password);
	}
	@Bean
	public DriverManagerDataSource dataSource2(){
		String url = "jdbc:mysql://localhost:3306/t_db2?useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=UTC";
		String userName = "root";
		String password = "123456";
		return new DriverManagerDataSource(url,userName,password);
	}
	@Bean
	public DriverManagerDataSource dataSource3(){
		String url = "jdbc:mysql://localhost:3306/t_db3?useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=UTC";
		String userName = "root";
		String password = "123456";
		return new DriverManagerDataSource(url,userName,password);
	}
	@Bean
	public MyDynamicDataSource dynamicDataSource(Map<String,DriverManagerDataSource> map){
		HashMap<Object, Object> original = new HashMap<>(map);
		MyDynamicDataSource myDynamicDataSource = new MyDynamicDataSource();
		myDynamicDataSource.setTargetDataSources(original);
		return myDynamicDataSource;
	}
	@Bean
	public JdbcTemplate jdbcTemplate(AbstractRoutingDataSource dataSource){
		return new JdbcTemplate(dataSource);
	}
}

配置了三個(gè)數(shù)據(jù)庫(kù)，也就是三個(gè)數(shù)據(jù)源，利用JdbcTemplate來(lái)快捷的執(zhí)行sql，JdbcTemplate需要指定用自己定義的動(dòng)態(tài)數(shù)據(jù)源（MyDynamicDataSource）。

此外，之前說(shuō)了，動(dòng)態(tài)數(shù)據(jù)源里面有個(gè)map，它得需要數(shù)據(jù)源，此外還需要key，這里用bean的名字做為key，創(chuàng)建MyDynamicDataSource之后，設(shè)置TargetDataSources。當(dāng)MyDynamicDataSource被Spring創(chuàng)建的時(shí)候，InitializingBean#afterPropertiesSet（）會(huì)通過(guò)DataSourceLookup轉(zhuǎn)換。

2.ThreadLocal

package datasource.dynamic;
import org.springframework.util.Assert;
public class DataSourceKeyHolder {
	private static final ThreadLocal<String> keyHolder = new ThreadLocal<>();
	public static void setKey(String key) {
		keyHolder.remove();
		keyHolder.set(key);
	}
	public static String getKey() {
		String s = keyHolder.get();
		Assert.notNull(s, "key 不能為空");
		return s;
	}
	public static void clear(){
		keyHolder.remove();
	}
}

3.動(dòng)態(tài)數(shù)據(jù)源實(shí)現(xiàn)類

package datasource.dynamic;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class MyDynamicDataSource extends AbstractRoutingDataSource {
	@Override
	protected Object determineCurrentLookupKey() {
		return DataSourceKeyHolder.getKey();
	}
}

只是從ThreadLocal中獲取當(dāng)前的key就好了。

4.實(shí)體對(duì)象

public class TestBean {
	private Integer id;
	private String name;
	private Double age;
	public Integer getId() {
		return id;
	}
	public void setId(Integer id) {
		this.id = id;
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public Double getAge() {
		return age;
	}
	public void setAge(Double age) {
		this.age = age;
	}
	@Override
	public String toString() {
		return "TestBean{" +
				"id=" + id +
				", name='" + name + '\'' +
				", age=" + age +
				'}';
	}
}

5.sql

-- auto-generated definition
create table t_test
(
    id   int auto_increment
        primary key,
    name text   null,
    age  double null
);

創(chuàng)建三個(gè)數(shù)據(jù)庫(kù)，在三個(gè)數(shù)據(jù)庫(kù)里面都創(chuàng)建這個(gè)表，填寫(xiě)點(diǎn)數(shù)據(jù)。

6.主要測(cè)試類

package datasource.dynamic;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.jdbc.core.BeanPropertyRowMapper;
import org.springframework.jdbc.core.JdbcTemplate;
public class MainTest {
	public static void main(String[] args) {
		try {
			AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Config.class);
			JdbcTemplate jdbcTemplate = context.getBean(JdbcTemplate.class); // 這里從頭到尾用的都是一個(gè)JdbcTemplate
			dataSource1(jdbcTemplate); // 動(dòng)態(tài)數(shù)據(jù)源1
			dataSource2(jdbcTemplate);// 動(dòng)態(tài)數(shù)據(jù)源2
			dataSource3(jdbcTemplate);// 動(dòng)態(tài)數(shù)據(jù)源3
		}catch (Exception e){
			e.printStackTrace();
		}
	}
	public static void dataSource1(	JdbcTemplate jdbcTemplate ){
		DataSourceKeyHolder.setKey("dataSource1");
		try {
			TestBean testBean = jdbcTemplate.queryForObject("select * from t_test where id=?", new BeanPropertyRowMapper<>(TestBean.class), 1);
			System.out.println(testBean);
		}finally {
			DataSourceKeyHolder.clear();
		}
	}
	public static void dataSource2(	JdbcTemplate jdbcTemplate ){
		DataSourceKeyHolder.setKey("dataSource2");
		try {
			TestBean testBean = jdbcTemplate.queryForObject("select * from t_test where id=?", new BeanPropertyRowMapper<>(TestBean.class), 1);
			System.out.println(testBean);
		}finally {
			DataSourceKeyHolder.clear();
		}
	}
	public static void dataSource3(	JdbcTemplate jdbcTemplate ){
		DataSourceKeyHolder.setKey("dataSource3");
		try {
			TestBean testBean = jdbcTemplate.queryForObject("select * from t_test where id=?", new BeanPropertyRowMapper<>(TestBean.class), 1);
			System.out.println(testBean);
		}finally {
			DataSourceKeyHolder.clear();
		}
	}
}

7.結(jié)果