gpt4 book ai didi

Spring AbstractRoutingDatasource 动态数据源的实例讲解

转载 作者:qq735679552 更新时间:2022-09-29 22:32:09 32 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章Spring AbstractRoutingDatasource 动态数据源的实例讲解由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

Spring AbstractRoutingDatasource 动态数据源

数据源上下文

AbstractRoutingDatasource 需要信息决定路由到那个数据源,该信息我们称为上下文。可以市任何对象,下面示例我们使用枚举 ClientDatabase 作为上下文:

?
1
2
3
public enum ClientDatabase {
     CLIENT_A, CLIENT_B
}

值得注意的是,实际中上下文可以是任何有意义的领域对象。如 Environment 定义上下文,在这种上下文中枚举可以包括 PRODUCTION, DEVELOPMENT, and TESTING. 。

上下文持有者 ( ContextHolder )

上下文持有者实现是存储当前上下文的容器,如 ThreadLocal 对象。除了包括上下文引用外,还应该包括set , get , clear 静态方法,AbstractRoutingDatasource 将查询 ContextHolder 获得 上下文,然后使用上下文查找实际的数据源.

这里最重要的是使用 ThreadLocal ,让上下文绑定只当前执行线程。使用这种方法就是为了保障当数据访问逻辑跨多个数据源使用事务:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
public class ClientDatabaseContextHolder {
     private static ThreadLocal<ClientDatabase> CONTEXT  = new ThreadLocal<>();
     public static void set(ClientDatabase clientDatabase) {
         Assert.notNull(clientDatabase, "clientDatabase cannot be null" );
         CONTEXT.set(clientDatabase);
     }
     public static ClientDatabase getClientDatabase() {
         return CONTEXT.get();
     }
     public static void clear() {
         CONTEXT.remove();
     }
}

数据源路由器

下面定义 ClientDataSourceRouter 继承 AbstractRoutingDataSource. 重写方法 determineCurrentLookupKey ,查询 ClientDatabaseContextHolder 获得当前 上下文。AbstractRoutingDataSource 帮我们处理了其他逻辑,仅了解它透明地返回了合适的数据源:

?
1
2
3
4
5
6
public class ClientDataSourceRouter extends AbstractRoutingDataSource {
     @Override
     protected Object determineCurrentLookupKey() {
         return ClientDatabaseContextHolder.getClientDatabase();
     }
}

配置

下面利用Map上下文存储DataSource对象,用于配置 AbstractRoutingDataSource 。当然也可以指定缺省数据源,当没有设定上下文时选择缺省数据源.

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
@Configuration
public class RoutingTestConfiguration {
     @Bean
     public ClientService clientService() {
         return new ClientService( new ClientDao(clientDatasource()));
     }
 
     @Bean
     public DataSource clientDatasource() {
         Map<Object, Object> targetDataSources = new HashMap<>();
         DataSource clientADatasource = clientADatasource();
         DataSource clientBDatasource = clientBDatasource();
         targetDataSources.put(ClientDatabase.CLIENT_A, clientADatasource);
         targetDataSources.put(ClientDatabase.CLIENT_B, clientBDatasource);
         ClientDataSourceRouter clientRoutingDatasource  = new ClientDataSourceRouter();
         clientRoutingDatasource.setTargetDataSources(targetDataSources);
         clientRoutingDatasource.setDefaultTargetDataSource(clientADatasource);       
         return clientRoutingDatasource;
     }
     // ...
}

我们可以使用不同方式的数据源,但典型场景时运行时创建,或使用JNDI方式查找。下面定义示例的Dao类,它依赖动态数据源进行实例化:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
public class ClientDao {
     private static final String SQL_GET_CLIENT_NAME = "select name from client" ;
     private final JdbcTemplate jdbcTemplate;
     public ClientDao(DataSource datasource) {
         this .jdbcTemplate = new JdbcTemplate(datasource);
     }
     public String getClientName() {
         return this .jdbcTemplate.query(SQL_GET_CLIENT_NAME, rowMapper).get( 0 );
     }
     private static RowMapper<String> rowMapper = (rs, rowNum) -> {
         return rs.getString( "name" );
     };
}

使用动态数据源

实际使用时,首先设置上下文,然后再执行业务操作。我们定义业务层,其中业务方法包括上下文作为参数。在执行业务之前设置上下文,调用之后清除上下文:

?
1
2
3
4
5
6
7
8
9
10
11
12
public class ClientService {
     private final ClientDao clientDao;
     public ClientService(ClientDao clientDao) {
         this .clientDao = clientDao;
     }
     public String getClientName(ClientDatabase clientDb) {
         ClientDatabaseContextHolder.set(clientDb);
         String clientName = this .clientDao.getClientName();
         ClientDatabaseContextHolder.clear();
         return clientName;
     }
}

总结

本文通过示例解释如何使用 Spring AbstractRoutingDataSource 实现 动态数据源切换.

以上为个人经验,希望能给大家一个参考,也希望大家多多支持我.

原文链接:https://blog.csdn.net/neweastsun/article/details/118281324 。

最后此篇关于Spring AbstractRoutingDatasource 动态数据源的实例讲解的文章就讲到这里了,如果你想了解更多关于Spring AbstractRoutingDatasource 动态数据源的实例讲解的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

32 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com