接 SpringBoot 动态配置数据源-进阶-可视化动态配置数据源-1
配置文件修改后,需要使配置生效
采用springcloud 配置 jar : spring-cloud-starter-config + spring-boot-starter-actuator
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-actuator</artifactId> </dependency> <dependency> <groupId>org.springframework.cloud</groupId> <artifactId>spring-cloud-starter-config</artifactId> <version>2.0.5.RELEASE</version> </dependency>
application.yml 添加配置
management: endpoints: web: exposure: include: refresh
配置参数刷新
@Component @Getter @RefreshScope public class ApplicationConfig { @Value("${XXX.XXX}") private String XXX; @Autowired private Environment environment; }
动态数据源- AbstractRoutingDataSource(每执行一次数据库,动态获取DataSource)
public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { return DynamicDataSourceContextHolder.getDataSourceType(); } /** * 动态更新自定义数据源 * @param defaultDataSource * @param customDataSources */ public void updateTargetDataSource(DataSource defaultDataSource,Map<String,DataSource> slaveDataSources){ Map<Object,Object> customDS=new HashMap<Object, Object>(); customDS.putAll(slaveDataSources); setTargetDataSources(customDS); setDefaultTargetDataSource(defaultDataSource); afterPropertiesSet(); } }
动态数据源通知-DynamicDataSourceAspect
@Aspect @Order(-1)//保证在@Transactional之前执行 @Component public class DynamicDataSourceAspect { //改变数据源 public void changeDataSource( String targetDataSource) { if (StringUtil.isNotEmpty(targetDataSource) && DynamicDataSourceContextHolder.isContainsDataSource(targetDataSource)) { DynamicDataSourceContextHolder.setDataSourceType(targetDataSource); } } public void clearDataSource( String targetDataSource) { DynamicDataSourceContextHolder.clearDataSourceType(); } }
动态数据源上下文管理 -DynamicDataSourceContextHolder
public class DynamicDataSourceContextHolder { //存放当前线程使用的数据源类型信息 private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>(); //存放数据源id public static Set<String> dataSourceIds = new HashSet<>(); //设置数据源 public static void setDataSourceType(String dataSourceType) { contextHolder.set(dataSourceType); } //获取数据源 public static String getDataSourceType() { return contextHolder.get(); } //清除数据源 public static void clearDataSourceType() { contextHolder.remove(); } //判断当前数据源是否存在 public static boolean isContainsDataSource(String dataSourceId) { return dataSourceIds.contains(dataSourceId); } }
注册动态数据源-初始化数据源和提供了执行动态切换数据源的工具类- EnvironmentAware(获取配置文件配置的属性值)
@Slf4j public class DynamicDataSourceRegister implements ImportBeanDefinitionRegistrar, EnvironmentAware { @Override public void setEnvironment(Environment environment) { DynamicDataSourceRegisterUtil.initDefaultDataSource(environment); DynamicDataSourceRegisterUtil.initSlaveDataSources(environment); } /** * 初始化注册数据源 BeanDefinition * @param annotationMetadata * @param beanDefinitionRegistry */ @Override public void registerBeanDefinitions(AnnotationMetadata annotationMetadata, BeanDefinitionRegistry beanDefinitionRegistry) { Map<Object, Object> targetDataSources = new HashMap<Object, Object>(); //添加默认数据源 targetDataSources.put(DynamicDataSourceRegisterUtil.DEFAULT_TARGET_DATA_SOURCE, DynamicDataSourceRegisterUtil.defaultDataSource); DynamicDataSourceContextHolder.dataSourceIds.add(DynamicDataSourceRegisterUtil.DEFAULT_TARGET_DATA_SOURCE); //添加其他数据源 targetDataSources.putAll(DynamicDataSourceRegisterUtil.slaveDataSources); DynamicDataSourceContextHolder.dataSourceIds.addAll(DynamicDataSourceRegisterUtil.slaveDataSources.keySet()); //创建DynamicDataSource GenericBeanDefinition beanDefinition = new GenericBeanDefinition(); beanDefinition.setBeanClass(DynamicDataSource.class); beanDefinition.setSynthetic(true); MutablePropertyValues mpv = beanDefinition.getPropertyValues(); mpv.addPropertyValue(DynamicDataSourceRegisterUtil.DEFAULT_TARGET_DATA_SOURCE, DynamicDataSourceRegisterUtil.defaultDataSource); mpv.addPropertyValue(DynamicDataSourceRegisterUtil.TARGET_DATA_SOURCES, targetDataSources); //注册 - BeanDefinitionRegistry beanDefinitionRegistry.registerBeanDefinition( DynamicDataSourceRegisterUtil.DATA_SOURCE, beanDefinition); log.info("Dynamic DataSource Registry"); } }
数据源注册工具类-DynamicDataSourceRegisterUtil 配合DynamicDataSourceRegister
@Slf4j public class DynamicDataSourceRegisterUtil { private static final ConversionService conversionService = new DefaultConversionService(); public final static String DEFAULT_TARGET_DATA_SOURCE ="defaultTargetDataSource"; //默认数据源 public static DataSource defaultDataSource; //用户自定义数据源 public static Map<String, DataSource> slaveDataSources = new HashMap<>(); public final static String TARGET_DATA_SOURCES ="targetDataSources"; public final static String DATA_SOURCE ="dataSource"; /** * 创建DataSource * @param dataSourceMap * @return */ public static DataSource buildDataSource(Map<String, Object> dataSourceMap) { DataSource dataSource=null; try { dataSource= DruidDataSourceFactory.createDataSource(dataSourceMap); if(dataSource instanceof DruidDataSource){ //注意:这一设置是为解决Druid 在获取连接时由于连接配置出错会一直等待获取连接,比较重要 ((DruidDataSource) dataSource).setBreakAfterAcquireFailure(true); } return dataSource; } catch (Exception e) { e.printStackTrace(); } return null; } /** * 初始化主数据源 */ public static void initDefaultDataSource(Environment env) { // 读取主数据源 Map<String, Object> dsMap = new HashMap<>(); dsMap.put("driver", env.getProperty("spring.datasource.driver-class-name")); dsMap.put("url", env.getProperty("spring.datasource.url")); dsMap.put("username", env.getProperty("spring.datasource.username")); dsMap.put("password", env.getProperty("spring.datasource.password")); defaultDataSource = buildDataSource(dsMap); } /** * 关闭老的数据源 */ private static void closeOldCustomDataSources(){ if(slaveDataSources!=null&&slaveDataSources.size()>0){ for (String key:slaveDataSources.keySet()){ DataSource dataSource =slaveDataSources.get(key); if(dataSource instanceof DruidDataSource){ ((DruidDataSource)dataSource).close(); log.info("closed datasource "+key); } } } if(slaveDataSources!=null){ slaveDataSources.clear(); } } /** * 初始化更多数据源 * */ public static void initSlaveDataSources(Environment env) { //初始化之前要先将老的数据源关闭 closeOldCustomDataSources(); // 读取配置文件获取更多数据源,也可以通过defaultDataSource读取数据库获取更多数据源 String dsPrefixs = env.getProperty("slave.datasource.names"); for (String dsPrefix : dsPrefixs.split(",")) {// 多个数据源 Map<String, Object> dsMap = new HashMap<>(); dsMap.put("driver", env.getProperty("slave.datasource." + dsPrefix + ".driver-class-name")); dsMap.put("url", env.getProperty("slave.datasource." + dsPrefix + ".url")); dsMap.put("username", env.getProperty("slave.datasource." + dsPrefix + ".username")); dsMap.put("password", env.getProperty("slave.datasource." + dsPrefix + ".password")); DataSource dataSource= buildDataSource(dsMap); slaveDataSources.put(dsPrefix, dataSource); } } /** * 更新配置文件,并热加载到Environment中 */ public synchronized static void refreshDataSoureProperties(Environment environment) { //将属性持久化到配置文件 refreshDataSource(environment); } /** * 更新配置之后要更新DynamicDataSource * @param environment */ private static void refreshDataSource(Environment environment) { initSlaveDataSources(environment); DynamicDataSource dynamicDataSource =ApplicationContextUtil.getBean(DATA_SOURCE); dynamicDataSource.updateTargetDataSource(defaultDataSource,slaveDataSources); DynamicDataSourceContextHolder.dataSourceIds.clear(); DynamicDataSourceContextHolder.dataSourceIds.add(DEFAULT_TARGET_DATA_SOURCE); DynamicDataSourceContextHolder.dataSourceIds.addAll(slaveDataSources.keySet()); } }
springboot 启动Application 增加注解引入
@Import({DynamicDataSourceRegister.class})
调用:
页面调用接口修改Yaml文件配置项后
//将配置增加到上下文环境中-spring-cloud 包含的接口 POST: http://127.0.0.1:8080/XXX/actuator/refresh //返回后,调用自己的refresh-API 将数据源加载 @PostMapping("/refresh") public JSONObject refresh(){ //数据源更新 DynamicDataSourceRegisterUtil.refreshDataSoureProperties(configCommon.getEnvironment()); //TODO 其他业务代码 JSONObject result = new JSONObject(); result.put("code",200); result.put("msg","初始化配置成功"); return result; }
也可以自己封装一下 Java-http调用 /actuator/refresh,再更新数据源配置
完结
-----------------------20191230----------------------------------
原文地址:https://www.cnblogs.com/feecy/p/12108702.html
时间: 2024-10-12 21:26:15