springboot多数据源配置及切换

注:本文的多数据源配置及切换的实现方法是,在框架中封装,具体项目中配置及使用,也适用于多模块项目

配置文件数据源读取

通过springboot的Envioment和Binder对象进行读取,无需手动声明DataSource的Bean

yml数据源配置格式如下:

spring:
  datasource:
    master:
      type: com.alibaba.druid.pool.DruidDataSource
      driverClassName: com.mysql.cj.jdbc.Driver
      url: jdbc:mysql://localhost:3306/main?
useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&serverTimezone=Asia/Shanghai
      username: root
      password: 11111
    cluster:
      - key: db1
        type: com.alibaba.druid.pool.DruidDataSource
        driverClassName: com.mysql.cj.jdbc.Driver
        url: jdbc:mysql://localhost:3306/haopanframetest_db1?
useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&serverTimezone=Asia/Shanghai
        username: root
        password: 11111
      - key: db2
        type: com.alibaba.druid.pool.DruidDataSource
        driverClassName: com.mysql.cj.jdbc.Driver
        url: jdbc:mysql://localhost:3306/haopanframetest_db2?
useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&serverTimezone=Asia/Shanghai
        username: root
        password: 11111

master为主数据库必须配置,cluster下的为从库,选择性配置

获取配置文件信息代码如下

    @Autowired
    private Environment env;
    @Autowired
    private ApplicationContext applicationContext;
    private Binder binder;

    binder = Binder.get(env);
    List<Map> configs = binder.bind("spring.datasource.cluster", Bindable.listOf(Map.class)).get();
            for (int i = 0; i < configs.size(); i++) {
                config = configs.get(i);
                String key = ConvertOp.convert2String(config.get("key"));
                String type = ConvertOp.convert2String(config.get("type"));
                String driverClassName = ConvertOp.convert2String(config.get("driverClassName"));
                String url = ConvertOp.convert2String(config.get("url"));
                String username = ConvertOp.convert2String(config.get("username"));
                String password = ConvertOp.convert2String(config.get("password"));
            }    

动态加入数据源

定义获取数据源的Service,具体项目中进行实现

public interface ExtraDataSourceService {
    List<DataSourceModel> getExtraDataSourc();
}

获取对应Service的所有实现类进行调用

    private List<DataSourceModel> getExtraDataSource(){
        List<DataSourceModel> dataSourceModelList = new ArrayList<>();
        Map<String, ExtraDataSourceService> res =
 applicationContext.getBeansOfType(ExtraDataSourceService.class);
        for (Map.Entry en :res.entrySet()) {
            ExtraDataSourceService service = (ExtraDataSourceService)en.getValue();
            dataSourceModelList.addAll(service.getExtraDataSourc());
        }
        return dataSourceModelList;
    }

通过代码进行数据源注册

主要是用过继承类AbstractRoutingDataSource,重写setTargetDataSources/setDefaultTargetDataSource方法

    // 创建数据源
    public boolean createDataSource(String key, String driveClass, String url, String username, String password, String databasetype) {
        try {
            try { // 排除连接不上的错误
                Class.forName(driveClass);
                DriverManager.getConnection(url, username, password);// 相当于连接数据库
            } catch (Exception e) {
                return false;
            }
            @SuppressWarnings("resource")
            DruidDataSource druidDataSource = new DruidDataSource();
            druidDataSource.setName(key);
            druidDataSource.setDriverClassName(driveClass);
            druidDataSource.setUrl(url);
            druidDataSource.setUsername(username);
            druidDataSource.setPassword(password);
            druidDataSource.setInitialSize(1); //初始化时建立物理连接的个数。初始化发生在显示调用init方法,或者第一次getConnection时
            druidDataSource.setMaxActive(20); //最大连接池数量
            druidDataSource.setMaxWait(60000); //获取连接时最大等待时间,单位毫秒。当链接数已经达到了最大链接数的时候,应用如果还要获取链接就会出现等待的现象,等待链接释放并回到链接池,如果等待的时间过长就应该踢掉这个等待,不然应用很可能出现雪崩现象
            druidDataSource.setMinIdle(5); //最小连接池数量
            String validationQuery = "select 1 from dual";
            druidDataSource.setTestOnBorrow(true); //申请连接时执行validationQuery检测连接是否有效,这里建议配置为TRUE,防止取到的连接不可用
            druidDataSource.setTestWhileIdle(true);//建议配置为true,不影响性能,并且保证安全性。申请连接的时候检测,如果空闲时间大于timeBetweenEvictionRunsMillis,执行validationQuery检测连接是否有效。
            druidDataSource.setValidationQuery(validationQuery); //用来检测连接是否有效的sql,要求是一个查询语句。如果validationQuery为null,testOnBorrow、testOnReturn、testWhileIdle都不会起作用。
            druidDataSource.setFilters("stat");//属性类型是字符串,通过别名的方式配置扩展插件,常用的插件有:监控统计用的filter:stat日志用的filter:log4j防御sql注入的filter:wall
            druidDataSource.setTimeBetweenEvictionRunsMillis(60000); //配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒
            druidDataSource.setMinEvictableIdleTimeMillis(180000); //配置一个连接在池中最小生存的时间,单位是毫秒,这里配置为3分钟180000
            druidDataSource.setKeepAlive(true); //打开druid.keepAlive之后,当连接池空闲时,池中的minIdle数量以内的连接,空闲时间超过minEvictableIdleTimeMillis,则会执行keepAlive操作,即执行druid.validationQuery指定的查询SQL,一般为select * from dual,只要minEvictableIdleTimeMillis设置的小于防火墙切断连接时间,就可以保证当连接空闲时自动做保活检测,不会被防火墙切断
            druidDataSource.setRemoveAbandoned(true); //是否移除泄露的连接/超过时间限制是否回收。
            druidDataSource.setRemoveAbandonedTimeout(3600); //泄露连接的定义时间(要超过最大事务的处理时间);单位为秒。这里配置为1小时
            druidDataSource.setLogAbandoned(true); //移除泄露连接发生是是否记录日志
            druidDataSource.init();
            this.dynamicTargetDataSources.put(key, druidDataSource);
            setTargetDataSources(this.dynamicTargetDataSources);// 将map赋值给父类的TargetDataSources
            super.afterPropertiesSet();// 将TargetDataSources中的连接信息放入resolvedDataSources管理
            log.info(key+"数据源初始化成功");
            //log.info(key+"数据源的概况:"+druidDataSource.dump());
            return true;
        } catch (Exception e) {
            log.error(e + "");
            return false;
        }
    }

通过切面注解统一切换

定义注解

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.TYPE, ElementType.PARAMETER})
@Documented
public @interface TargetDataSource {
    String value() default "master"; //该值即key值
}

定义基于线程的切换类

public class DBContextHolder {
    private static Logger log = LoggerFactory.getLogger(DBContextHolder.class);
    // 对当前线程的操作-线程安全的
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    // 调用此方法,切换数据源
    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
        log.info("已切换到数据源:{}",dataSource);
    }

    // 获取数据源
    public static String getDataSource() {
        return contextHolder.get();
    }

    // 删除数据源
    public static void clearDataSource() {
        contextHolder.remove();
        log.info("已切换到主数据源");
    }

}

定义切面

方法的注解优先级高于类注解,一般用于Service的实现类

@Aspect
@Component
@Order(Ordered.HIGHEST_PRECEDENCE)
public class DruidDBAspect {
    private static Logger logger = LoggerFactory.getLogger(DruidDBAspect.class);
    @Autowired
    private DynamicDataSource dynamicDataSource;

    /**
     * 切面点 指定注解
     * */
    @Pointcut("@annotation(com.haopan.frame.common.annotation.TargetDataSource) " +
            "|| @within(com.haopan.frame.common.annotation.TargetDataSource)")
    public void dataSourcePointCut() {

    }

    /**
     * 拦截方法指定为 dataSourcePointCut
     * */
    @Around("dataSourcePointCut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Class targetClass = point.getTarget().getClass();
        Method method = signature.getMethod();

        TargetDataSource targetDataSource = (TargetDataSource)targetClass.getAnnotation(TargetDataSource.class);
        TargetDataSource methodDataSource = method.getAnnotation(TargetDataSource.class);
        if(targetDataSource != null || methodDataSource != null){
            String value;
            if(methodDataSource != null){
                value = methodDataSource.value();
            }else {
                value = targetDataSource.value();
            }
            DBContextHolder.setDataSource(value);
            logger.info("DB切换成功,切换至{}",value);
        }

        try {
            return point.proceed();
        } finally {
            logger.info("清除DB切换");
            DBContextHolder.clearDataSource();
        }
    }
}

分库切换

开发过程中某个库的某个表做了拆分操作,相同的某一次数据库操作可能对应到不同的库,需要对方法级别进行精确拦截,可以定义一个业务层面的切面,规定每个方法必须第一个参数为dbName,根据具体业务找到对应的库传参

    @Around("dataSourcePointCut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Class targetClass = point.getTarget().getClass();
        Method method = signature.getMethod();

        ProjectDataSource targetDataSource = 
(ProjectDataSource)targetClass.getAnnotation(ProjectDataSource.class);
        ProjectDataSource methodDataSource = method.getAnnotation(ProjectDataSource.class);
        String value = "";
        if(targetDataSource != null || methodDataSource != null){
            //获取方法定义参数
            DefaultParameterNameDiscoverer discover = new DefaultParameterNameDiscoverer();
            String[] parameterNames = discover.getParameterNames(method);
            //获取传入目标方法的参数
            Object[] args = point.getArgs();
            for (int i=0;i<parameterNames.length;i++){
                String pName = parameterNames[i];
                if(pName.toLowerCase().equals("dbname")){
                    value = ConvertOp.convert2String(args[i]);
                }
            }
            if(!StringUtil.isEmpty(value)){
                DBContextHolder.setDataSource(value);
                logger.info("DB切换成功,切换至{}",value);
            }
        }

        try {
            return point.proceed();
        } finally {
            if(!StringUtil.isEmpty(value)){
                logger.info("清除DB切换");
                DBContextHolder.clearDataSource();
            }

        }
    }

源码资源:https://download.csdn.net/download/u013407099/12870951

  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
### 回答1: Spring Boot支持多数据源配置切换。具体步骤如下: 1. 在application.properties或application.yml文件中配置多个数据源的连接信息。 2. 创建多个数据源配置类,分别继承org.springframework.boot.autoconfigure.jdbc.DataSourceProperties类,并使用@Configuration注解进行标注。 3. 创建多个数据源的实例,分别使用@Bean注解进行标注,并在方法中使用@ConfigurationProperties注解将对应的配置类注入。 4. 创建一个数据源切换切面类,使用@Aspect注解进行标注,并在方法中使用@Around注解进行标注。 5. 在切面类中获取当前线程的数据源标识,并根据标识切换数据源。 6. 在需要使用不同数据源的地方,使用@Qualifier注解指定对应的数据源实例。 7. 在需要切换数据源的地方,使用ThreadLocal或者AOP等方式设置当前线程的数据源标识。 以上就是Spring Boot多数据源配置切换的基本步骤。 ### 回答2: SpringBoot 是当前非常流行的一种 Java Web 框架,它提供了便捷的配置和开发方式,同时提供了极为方便的多数据源配置切换,这正是 SpringBoot 的特别之处。 当应用需要访问多个数据库时,SpringBoot 提供了两种方式进行多数据源配置:程序化配置和自动配置。 程序化配置 通过实现 `org.springframework.jdbc.datasource.AbstractDataSource` 接口,开发者可以自定义多个数据源,可以自行配置数据源的属性,如连接信息、数据库账户等。该方式下开发者需要在代码中显式地进行数据源切换,可以通过 `ThreadLocal` 存储当前线程使用的数据源配置多数据源的步骤如下: 1. 实现数据源配置类,该类需要继承 `org.springframework.jdbc.datasource.AbstractDataSource` 接口。 2. 通过使用注解的方式来指定不同数据源所对应的 bean。 3. 在代码中通过 `DataSourceContextHolder` 类来切换数据源。 自动配置 SpringBoot 2.x 已将多数据源自动配置纳入到官方的扩展包中。开发者需要在 application.yml 或 application.properties 文件中配置多个数据源SpringBoot 自动从配置文件中读取多个数据源的信息并将它们注入到 DataSource 类中。SpringBoot 自动配置多数据源的步骤如下: 1. 在 application.yml 文件中配置多个数据源的连接信息等属性。 2. 在配置类中通过 `@Bean` 注解数据源声明为 SpringBoot 的自动配置类。 3. 使用 JPA 或者 MyBatis 等框架时,指定不同的事物管理器和数据源。 通过以上两种方式,我们可以轻松地实现多数据源配置切换。在开发实践中,应用场景也会影响到多数据源的选择。例如,如果应用需要访问多个数据库来实现分表分库的功能,应配置多个数据源并通过程序化的方式切换数据源。如果两个数据源的使用场景不一样,或者是正式和测试环境的切换等情况,则应使用自动配置方式。 总结而言,SpringBoot 提供了既方便又可靠的多数据源配置切换方式,方便开发者快速开发出高效易维护的应用程序。 ### 回答3: 在开发中,可能需要使用多个数据源,比如一个项目使用MySQL,另一个使用Oracle等。Spring Boot提供了方便的多数据源配置切换方法。 一、配置多数据源 1.在application.properties文件中分别配置每个数据源的相关信息,例如: spring.datasource.first.url=jdbc:mysql://localhost:3306/firstdb spring.datasource.first.username=root spring.datasource.first.password=root spring.datasource.second.url=jdbc:mysql://localhost:3306/seconddb spring.datasource.second.username=root spring.datasource.second.password=root 2.为每个数据源创建DataSource Bean @Configuration public class DataSourceConfig { @Bean(name = "firstDataSource") @ConfigurationProperties(prefix = "spring.datasource.first") public DataSource firstDataSource() { return DataSourceBuilder.create().build(); } @Bean(name = "secondDataSource") @ConfigurationProperties(prefix = "spring.datasource.second") public DataSource secondDataSource() { return DataSourceBuilder.create().build(); } } 3.为每个数据源配置JdbcTemplate @Configuration public class JdbcTemplateConfig { @Autowired @Qualifier("firstDataSource") private DataSource firstDataSource; @Autowired @Qualifier("secondDataSource") private DataSource secondDataSource; @Bean(name = "firstJdbcTemplate") public JdbcTemplate firstJdbcTemplate() { return new JdbcTemplate(firstDataSource); } @Bean(name = "secondJdbcTemplate") public JdbcTemplate secondJdbcTemplate() { return new JdbcTemplate(secondDataSource); } } 二、切换数据源 1.创建一个自定义注解,例如: @Target({ElementType.METHOD, ElementType.TYPE}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface DataSourceSwitch { DataSourceType value(); } 2.创建一个DataSourceType枚举类,用于指定数据源类型,例如: public enum DataSourceType { FIRST, SECOND } 3.创建一个切面类,用于切换数据源,例如: @Aspect @Component public class DataSourceAspect { @Pointcut("@annotation(ds)") public void dataSourcePointCut(DataSourceSwitch ds) { } @Around("dataSourcePointCut(ds)") public Object around(ProceedingJoinPoint point, DataSourceSwitch ds) throws Throwable { DataSourceContextHolder.setDataSourceType(ds.value()); try { return point.proceed(); } finally { DataSourceContextHolder.clearDataSourceType(); } } } 4.在需要使用不同数据源的方法或类上添加自定义注解,例如: @Service public class UserServiceImpl implements UserService { @Autowired private JdbcTemplate firstJdbcTemplate; @Autowired private JdbcTemplate secondJdbcTemplate; @Override @DataSourceSwitch(DataSourceType.FIRST) public void addUser(User user) { String sql = "insert into user(id, name) values(?, ?)"; Object[] params = new Object[]{user.getId(), user.getName()}; firstJdbcTemplate.update(sql, params); } @Override @DataSourceSwitch(DataSourceType.SECOND) public User getUser(String id) { String sql = "select * from user where id=?"; Object[] params = new Object[]{id}; return secondJdbcTemplate.queryForObject(sql, params, new UserMapper()); } } 这样,当调用addUser方法时,会使用firstJdbcTemplate对应的数据源,而调用getUser方法时,会使用secondJdbcTemplate对应的数据源。 总之,Spring Boot提供的多数据源配置切换方法非常方便,能够极大地提高开发效率。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值