spring：如何在运行时管理用BeanFactoryPostProcessor创建的数据源？

Question

我正在用Spring Data JPA做一个项目。

我已经设法用BeanFactoryPostProcessor动态创建了datasource，并在使用AbstracTroutingDataSource登录时切换到所需的datasource。

现在我想在运行时做的是：

1. 使用BeanFactoryPostProcessor获取动态数据源映射
2. 创建新的数据源
3. 将最近创建的数据源与其他数据源一起放在映射中

SpringContext-jpa.xml

<?xml version="1.0" encoding="UTF-8"?> 
<beans xmlns="http://www.springframework.org/schema/beans"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xmlns:aop="http://www.springframework.org/schema/aop"
xmlns:context="http://www.springframework.org/schema/context"
xmlns:p="http://www.springframework.org/schema/p"
xmlns:tx="http://www.springframework.org/schema/tx"
xmlns:jpa="http://www.springframework.org/schema/data/jpa"
............
>
<!-- 
...
...
Spring Data JPA config 
...
...
-->

<!--    Parent abstract Datasource -->
<bean id="BasicdsCargestWeb" class="org.apache.commons.dbcp.BasicDataSource">
    <property name="driverClassName" value="${jdbc.driverClassName}"></property>
    <property name="username" value="${jdbc.username}"></property>
    <property name="password" value="${jdbc.password}"></property>
</bean>

<!--  Generic Datasource   -->
<bean id="dsCargestWeb" class="com.cargest.custom.CargestRoutingDataSource">
    <property name="targetDataSources">
        <map>

        </map>
    </property>
    <property name="defaultTargetDataSource" ref="cargestnet1ds" />
</bean>

</beans>

DataSourCeregisteringBeanFactoryPostProcessor.java

@Component 
class DatasourceRegisteringBeanFactoryPostProcessor implements BeanFactoryPostProcessor { 

html" target="_blank">public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) {

    // this is my list of Datasources 
    List<Database> dbs = new ArrayList<Database>();

    /*
     * Hidden code, here i get my list of Datasources 
     */

    BeanDefinitionRegistry factory = (BeanDefinitionRegistry) beanFactory;
    BeanDefinitionBuilder datasourceDefinitionBuilder;

    for (Database db : dbs) {
        datasourceDefinitionBuilder = BeanDefinitionBuilder
                .childBeanDefinition("BasicdsCargestWeb") 
                .addPropertyValue("url", db.getUrl()+db.getName()+"?autoReconnect=true");

        factory.registerBeanDefinition("cargestnet"+db.getId()+"ds",
                datasourceDefinitionBuilder.getBeanDefinition());
    }


    // Configure the dataSource bean properties 
    MutablePropertyValues mpv = factory.getBeanDefinition("dsCargestWeb").getPropertyValues();

    // Here you can set the default dataSource 
    mpv.removePropertyValue("defaultTargetDataSource");
    mpv.addPropertyValue("defaultTargetDataSource", 
        new RuntimeBeanReference("cargestnet1ds")); 

    // Set the targetDataSource properties map with the list of connections 
    ManagedMap<String, RuntimeBeanReference> mm = (ManagedMap<String, RuntimeBeanReference>) mpv.getPropertyValue("targetDataSources").getValue();
    System.out.println("list size "+mm.size());

    mm.clear();

    for (Database db : dbs) {
         mm.put(db.getId().toString(), new RuntimeBeanReference("cargestnet"+db.getId()+"ds"));
    }
} 
} 

问题是BeanFactoryPostProcessor类正在使用ConfigurableListableBeanFactory作为BeanFactory。

我需要从另一个类（在运行时）访问相同的beanFactory，以便修改我的数据源映射(DscarGestWeb-->TargetDatasources-->map)。

谢谢

Answer 1

通过RuntimeBeanReference和ManagedSet，可以将运行时集设置为BeanDefine。这是正确的使用方法：

ManagedSet<RuntimeBeanReference> processorSet = new ManagedSet<RuntimeBeanReference>();
    for (String processorBeanName : processorBeanNames) {
        processorSet.add(new RuntimeBeanReference(processorBeanName));
    }
beanDefine.getPropertyValues().addPropertyValue(new PropertyValue(key, value));

Answer 2

我认为您可以尝试以下架构：

• 工厂bean创建数据源的初始映射
• abstractroutingDataSource实现保存DataSources的映射，并在最初注入上述工厂bean

这样，如果以后需要添加新的DataSource，可以从AbstracTroutingDataSource实现获取映射并将其直接添加到其中，或者在AbstracTroutingDataSource实现中放置一个方法以添加新的DataSource。

作为上述的变体，您可以创建一个类似的abstractroutingDataSource实现，以及另一个注入它的bean，它在其init-method中计算DataSources映射，并在路由数据源中对其进行传输。