当使用AbstractRoutingDataSource作为数据源并尝试在运行时动态更改数据源时,可能会遇到动态更改无效的问题。这是因为AbstractRoutingDataSource的determineCurrentLookupKey()方法只会在初始化时被调用一次。
解决该问题的方法是,重写AbstractRoutingDataSource的determineCurrentLookupKey()方法,并使用ThreadLocal来存储当前数据源的标识。可以使用以下示例代码来实现:
public class DynamicDataSource extends AbstractRoutingDataSource {
private static final ThreadLocal dataSourceKey = new ThreadLocal<>();
public static void setCurrentDataSource(String dataSource) {
dataSourceKey.set(dataSource);
}
public static void clearCurrentDataSource() {
dataSourceKey.remove();
}
@Override
protected Object determineCurrentLookupKey() {
return dataSourceKey.get();
}
}
在调用数据库操作之前,需要先调用DynamicDataSource.setCurrentDataSource()方法来设置当前使用的数据源。例如:
DynamicDataSource.setCurrentDataSource("dataSource1"); // 进行数据库操作 DynamicDataSource.clearCurrentDataSource();
当然,也可以根据不同的需求,将数据源标识存储在其他地方,如访问控制上下文中。
上一篇:Abstractroomdatabaseimplementationnotfounddaggerhilt
下一篇:AbstractRoutingDataSource实现中,TransactionSynchronizationManager未返回正确的值。