我遇到的问题是这样的。项目使用 Spring + Hibernate + proxool 实现数据库连接管理和访问。 需求是实现多数据源的动态配置和加载使用。 思路是:
1. 用一个类 AdvancedDataSourceInitizer 实现ApplicationListener 接口,当 ContextRefreshEvent 事件被发布时, 自动从数据库中读取数据库配置,转化为 ProxoolDataSource 对象,并存入到一个 Map<dataSourceName, ProxoolDataSource> 中;
package opstools.moonmm.support.listener;
import java.util.List;
import java.util.Map;
import javax.sql.DataSource;
import opstools.framework.datasource.MultiDataSource;
import opstools.moonmm.clusterconfig.entity.ClusterConfig;
import opstools.moonmm.clusterconfig.service.ClusterConfigService;
import opstools.moonmm.support.utils.DBUtil;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.jdbc.datasource.lookup.MapDataSourceLookup;
public class AdvancedDataSourceInitializer implements ApplicationListener, ApplicationContextAware {
private String desiredEventClassName;
protected ApplicationContext applicationContext;
public void onApplicationEvent(ApplicationEvent event) {
if (shouldStart(event)) {
Map<String, DataSource> cachedMap = (Map<String, DataSource>)applicationContext.getBean("dataSources");
ClusterConfigService clusterConfigService = (ClusterConfigService)applicationContext.getBean("clusterConfigService");
List<ClusterConfig> cclist = clusterConfigService.getAllClusterConfigInstances();
DBUtil.addCachedDatasources(cachedMap, cclist);
MapDataSourceLookup dsLookup = (MapDataSourceLookup) applicationContext.getBean("dataSourceLookup");
dsLookup.setDataSources(cachedMap);
MultiDataSource mds = (MultiDataSource) applicationContext.getBean("dataSource");
mds.setTargetDataSources(cachedMap);
mds.afterPropertiesSet();
}
}
protected Class<?> getDesiredType() {
try {
return Class.forName(desiredEventClassName);
} catch (ClassNotFoundException e) {
throw new RuntimeException(e);
}
}
public String getDesiredEventClassName() {
return desiredEventClassName;
}
public void setDesiredEventClassName(String desiredEventClassName) {
this.desiredEventClassName = desiredEventClassName;
}
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = applicationContext;
}
protected boolean shouldStart(ApplicationEvent event){
Class<?> clazz = getDesiredType();
return clazz.isInstance(event);
}
}
DBUtil.java : 用于将数据库配置转化为 ProxoolDataSource 对象, 归入连接池管理
package opstools.moonmm.support.utils;
import java.util.List;
import java.util.Map;
import javax.sql.DataSource;
import opstools.moonmm.clusterconfig.entity.ClusterConfig;
import opstools.moonmm.monitorconfig.entity.MonitorConfig;
import org.logicalcobwebs.proxool.ProxoolDataSource;
public class DBUtil {
private DBUtil(