| | |
|---|
| | | package com.ruoyi.framework.config; |
|---|
| | | |
|---|
| | | import java.io.IOException; |
|---|
| | | import java.util.HashMap; |
|---|
| | | import java.util.Map; |
|---|
| | | import javax.servlet.Filter; |
|---|
| | | import javax.servlet.FilterChain; |
|---|
| | | import javax.servlet.ServletException; |
|---|
| | | import javax.servlet.ServletRequest; |
|---|
| | | import javax.servlet.ServletResponse; |
|---|
| | | import javax.sql.DataSource; |
|---|
| | | import com.alibaba.druid.spring.boot.autoconfigure.properties.DruidStatProperties; |
|---|
| | | import com.alibaba.druid.util.Utils; |
|---|
| | | import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
|---|
| | | import org.springframework.boot.context.properties.ConfigurationProperties; |
|---|
| | | import org.springframework.boot.web.servlet.FilterRegistrationBean; |
|---|
| | | import org.springframework.context.annotation.Bean; |
|---|
| | | import org.springframework.context.annotation.Configuration; |
|---|
| | | import org.springframework.context.annotation.Primary; |
|---|
| | | import com.alibaba.druid.pool.DruidDataSource; |
|---|
| | | import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder; |
|---|
| | | import com.alibaba.druid.spring.boot.autoconfigure.properties.DruidStatProperties; |
|---|
| | | import com.alibaba.druid.util.Utils; |
|---|
| | | import com.ruoyi.common.enums.DataSourceType; |
|---|
| | | import com.ruoyi.common.utils.spring.SpringUtils; |
|---|
| | | import com.ruoyi.framework.config.properties.DruidProperties; |
|---|
| | | import com.ruoyi.framework.datasource.DynamicDataSource; |
|---|
| | | |
|---|
| | | import javax.servlet.*; |
|---|
| | | import java.io.IOException; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * druid 配置多数据源 |
|---|
| | | * |
|---|
| | | * |
|---|
| | | * @author ruoyi |
|---|
| | | */ |
|---|
| | | @Configuration |
|---|
| | | public class DruidConfig |
|---|
| | | { |
|---|
| | | @Bean |
|---|
| | | @ConfigurationProperties("spring.datasource.druid.master") |
|---|
| | | public DataSource masterDataSource(DruidProperties druidProperties) |
|---|
| | | { |
|---|
| | | DruidDataSource dataSource = DruidDataSourceBuilder.create().build(); |
|---|
| | | return druidProperties.dataSource(dataSource); |
|---|
| | | } |
|---|
| | | |
|---|
| | | @Bean |
|---|
| | | @ConfigurationProperties("spring.datasource.druid.slave") |
|---|
| | | @ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true") |
|---|
| | | public DataSource slaveDataSource(DruidProperties druidProperties) |
|---|
| | | { |
|---|
| | | DruidDataSource dataSource = DruidDataSourceBuilder.create().build(); |
|---|
| | | return druidProperties.dataSource(dataSource); |
|---|
| | | } |
|---|
| | | |
|---|
| | | @Bean(name = "dynamicDataSource") |
|---|
| | | @Primary |
|---|
| | | public DynamicDataSource dataSource(DataSource masterDataSource) |
|---|
| | | { |
|---|
| | | Map<Object, Object> targetDataSources = new HashMap<>(); |
|---|
| | | targetDataSources.put(DataSourceType.MASTER.name(), masterDataSource); |
|---|
| | | setDataSource(targetDataSources, DataSourceType.SLAVE.name(), "slaveDataSource"); |
|---|
| | | return new DynamicDataSource(masterDataSource, targetDataSources); |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 设置数据源 |
|---|
| | | * |
|---|
| | | * @param targetDataSources 备选数据源集合 |
|---|
| | | * @param sourceName 数据源名称 |
|---|
| | | * @param beanName bean名称 |
|---|
| | | */ |
|---|
| | | public void setDataSource(Map<Object, Object> targetDataSources, String sourceName, String beanName) |
|---|
| | | { |
|---|
| | | try |
|---|
| | | { |
|---|
| | | DataSource dataSource = SpringUtils.getBean(beanName); |
|---|
| | | targetDataSources.put(sourceName, dataSource); |
|---|
| | | } |
|---|
| | | catch (Exception e) |
|---|
| | | { |
|---|
| | | } |
|---|
| | | } |
|---|
| | | public class DruidConfig { |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 去除监控页面底部的广告 |
|---|
