springboot整合dynamic-datasource实现

一：依赖引入

<dependency>
   <groupId>com.baomidou</groupId>
   <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
   <version>3.5.1</version>
</dependency>

二：修改配置文件

spring:
  datasource:
    druid:
      stat-view-servlet:
        enabled: true
        login-username: admin
        login-password: 123456
    dynamic:
      p6spy: true
      lazy: true
      # 配置全局druid参数，请按需配置
      druid:
        initial-size: 5
        max-active: 8
        min-idle: 3
        max-wait: 1000
        validation-query: 'select 1'
      primary: master
      datasource:
        master:
          username: root
          password: "123456"
          url: jdbc:mysql://localhost:3306/schema_master?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=GMT%2B8&useAffectedRows=true
          driver-class-name: com.mysql.cj.jdbc.Driver
        slave_1:
          username: root
          password: "123456"
          url: jdbc:mysql://localhost:3306/schema_slave_1?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=GMT%2B8&useAffectedRows=true
          driver-class-name: com.mysql.cj.jdbc.Driver
          druid:
            initial-size: 6
        slave_2:
          username: root
          password: "123456"
          url: jdbc:mysql://localhost:3306/schema_slave_2?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=GMT%2B8&useAffectedRows=true
          driver-class-name: com.mysql.cj.jdbc.Driver
logging:
  level:
    com.baomidou: debug

三：自定义拦截器通过参数切换数据源

package com.yx.interceptor;

import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * 动态数据源拦截器
 *
 * @author yx
 */
@Slf4j
@Component
public class DynamicDatasourceInterceptor implements HandlerInterceptor {

    /**
     * 在请求处理之前进行调用（Controller方法调用之前）
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
        String requestURI = request.getRequestURI();
        log.info("经过多数据源Interceptor,当前路径是{}", requestURI);
        String headerDs = request.getHeader("ds");

        String dsKey = "master";
        if (StringUtils.isNotBlank(headerDs)) {
            if (headerDs.startsWith("master")) {
                dsKey = "master";
            } else if (headerDs.startsWith("slave_1")) {
                dsKey = "slave_1";
            } else {
                dsKey = "slave_2";
            }
        }
        DynamicDataSourceContextHolder.push(dsKey);
        return true;
    }

    /**
     * 请求处理之后进行调用，但是在视图被渲染之前（Controller方法调用之后）
     */
    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) {

    }

    /**
     * 在整个请求结束之后被调用，也就是在DispatcherServlet 渲染了对应的视图之后执行（主要是用于进行资源清理工作）
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
        DynamicDataSourceContextHolder.clear();
    }
}

四：注解切换数据源

@DS("slave_1")

五：项目demo示例

dynamic-datasource-demo