在 Spring 引导自动配置中使用子 class 扩展数据源

Question

我正在尝试编写一个自动配置库，为任何DataSource添加功能。 我编写了一个子类，我将在这里调用CustomDataSource并覆盖DataSource的一些方法。

@Configuration
@ConditionalOnBean(DataSource.class)
@AutoConfigureAfter(DataSourceAutoConfiguration.class)
public class CustomDataSourceAutoConfiguration {

    private final DataSource dataSource;

    public CustomDataSourceAutoConfiguration(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    @Primary
    @Bean
    public CustomDataSource customDataSource() {
        return new CustomDataSource(dataSource);
    }
}

但我找不到让我做我想做的事的方法。 它总是会导致循环引用和异常：

BeanCurrentlyInCreationException: Error creating bean with name 'customDataSource': Requested bean is currently in creation: Is there an unresolvable circular reference?

有没有解决的办法？

Answer 1

我找到了一种通过实现BeanPostProcessor来解决此问题的方法：

public class DataSourcePostProcessor implements BeanPostProcessor {

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) {
        if (bean instanceof DataSource && !(bean instanceof CustomDataSource)) {
            return new CustomDataSource((DataSource) bean);
        } else {
            return bean;
        }
    }

}

postProcessAfterInitialization方法可以显式用于将 bean 包装在代理中，引用BeanPostProcessor文档：

[...] 使用代理包装 bean 的后处理器通常会实现postProcessAfterInitialization(java.lang.Object, java.lang.String) 。