这是我对DataSource的实现
public class DataSourceManager implements DataSource { private Map<String, DataSource> dataSources = new HashMap<>(); private DataSource dataSource; public DataSourceManager() { } public DataSourceManager(DataSource dataSource) { this.dataSource = dataSource; } public void add(String name, DataSource dataSource) { dataSources.put(name, dataSource); } public void switchDataSource(String name) { dataSource = dataSources.get(name); } @Override public PrintWriter getLogWriter() throws SQLException { return dataSource.getLogWriter(); } @Override public void setLogWriter(PrintWriter out) throws SQLException { dataSource.setLogWriter(out); } @Override public void setLoginTimeout(int seconds) throws SQLException { dataSource.setLoginTimeout(seconds); } @Override public int getLoginTimeout() throws SQLException { return dataSource.getLoginTimeout(); } @Override public Logger getParentLogger() throws SQLFeatureNotSupportedException { return dataSource.getParentLogger(); } @Override public <T> T unwrap(Class<T> iface) throws SQLException { return dataSource.unwrap(iface); } @Override public boolean isWrapperFor(Class<?> iface) throws SQLException { return dataSource.isWrapperFor(iface); } @Override public Connection getConnection() throws SQLException { return dataSource.getConnection(); } @Override public Connection getConnection(String username, String password) throws SQLException { return dataSource.getConnection(username, password); }}这是我的配置
@Configurationpublic class DataSourceConfig { @Autowired private Environment env; public DataSource makeDataSource(String name) { return DataSourceBuilder.create() .driverClassName(env.getProperty("spring.datasource." + name + ".driver-class-name")) .url(env.getProperty("spring.datasource." + name + ".url")).build(); } @Bean public DataSource dataSource() { DataSourceManager dataSourceManager = new DataSourceManager(); dataSourceManager.add("test1", makeDataSource("test1")); dataSourceManager.add("test2", makeDataSource("test2")); dataSourceManager.switchDataSource("test1"); return dataSourceManager; }}这是application.yml
spring: jpa: hibernate: ddl-auto: create properties: hibernate: dialect: org.hibernate.dialect.H2Dialect datasource: test1: name: test2 url: jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE driver-class-name: org.h2.Driver username: h2 password: h2 test2: name: test1 url: jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE driver-class-name: org.h2.Driver username: h2 password: h2
