我正在尝试使用AbstractRoutingDataSource创建动态数据源(基于会话),但是spring boot会在启动时继续调用AbstractRoutingDataSource的determineCurrentLookupKey()函数。
@ConfigurationProperties(prefix = "datasource.int")
@Bean
public DataSource internal() {
return DataSourceBuilder.create().build();
}
@ConfigurationProperties(prefix = "datasource.ext")
@Bean
public DataSource external() {
return DataSourceBuilder.create().build();
}
@Bean(name = "dataSource")
@Qualifier("dataSource")
public DynamicRoutingDataSourceResolver dataSource() {
DynamicRoutingDataSourceResolver resolver = new DynamicRoutingDataSourceResolver();
DataSource internal = internal();
DataSource external = external();
Map<Object, Object> dataSources = new HashMap<>();
dataSources.put(env.getRequiredProperty("app.param.int"), internal);
dataSources.put(env.getRequiredProperty("app.param.ext"), external);
resolver.setDefaultTargetDataSource(internal);
resolver.setTargetDataSources(dataSources);
resolver.afterPropertiesSet();
return resolver;
}
public class DynamicRoutingDataSourceResolver extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
Object o = request.getSession().getAttribute(Constant.DATASOURCE);
return o;
}
}
它会抛出NullPointerException,因为RequestAttributes为null。 我试图将@Primary放入其中一个数据源,但是当我尝试执行查询时,没有触发AbstractRoutingDataSource determineCurrentLookupKey()。
我正在使用弹簧数据jpa。
答案 0 :(得分:0)
您不必在此处指定任何形式的@Primary
。
只需创建并将所有数据源分配到AbstractRoutingDataSource
即可。如果有一个数据源是默认数据源,也可以使用#setDefaultTargetDataSource
进行设置,就像您的代码所示。
您的案例中的NullPointerException
只是您的代码需要防范的内容。如果您发现解析程序中不存在请求数据,只需返回数据源查找键的null
值即可。
AbstractRoutingDataSource
,或者指定的键未通过查找映射定义为数据源,则 null
使用定义的默认数据源。