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

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

Spring AbstractRoutingDatasource 动态数据源

数据源上下文

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

public enum ClientDatabase {

CLIENT_A, CLIENT_B

}

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

上下文持有者 ( ContextHolder )

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

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

public class ClientDatabaseContextHolder {

private static ThreadLocal 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 帮我们处理了其他逻辑，仅了解它透明地返回了合适的数据源：

public class ClientDataSourceRouter extends AbstractRoutinhttp://gDataSource {

@Override

protected Object determineCurrentLookupKey() {

return ClientDatabaseContextHolder.getClientDatabase();

}

}

配置

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

@Configuration

public class RoutingTestConfiguration {

@Bean

public ClientService clientService() {

return new ClientService(new ClientDao(clientDatasource()));

}

@Bean

public DataSource clientDatasource() {

Map

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类，它依赖动态数据源进行实例化：

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 rowMapper = (rs, rowNum) -> {

return rs.getString("name");

};

}

使用动态数据源

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

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 实现 动态数据源切换。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。