JAVA项目实现多数据源动态切换

1、先引入动态数据源涉及的包,我这边用的是gradle,maven也有相关对应的

2、application.yml配置多数据源,或者properties,看自己习惯

3、创建切换数据源的AOP,我这边的规则是根据目录来切换,还有别的规则可以自行切换,主要是这个DynamicDataSourceContextHolder(dynamic-datasource-spring-boot-starter中自带类可直接使用),用来切换数据源。不想用自带的可以自己定义一个,有很多这种,这里是最简单的

package com.wxsdg.common.datasource;


import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

@Aspect
@Component
public class DynamicDataSourceAop {


    @Pointcut("execution(* com.wxsdg.mapper.ds..*.*(..))")
    public void pointcut() {}

    @Before("pointcut()")
    public void doBefore(JoinPoint joinPoint)
    {
        String path=joinPoint.getSignature().getDeclaringTypeName();
        String dataSourceKey = path.substring(path.indexOf("ds.")+3).split("\.")[0];
        if(dataSourceKey !=null){
            //设置为当前目录下的数据源名称
            DynamicDataSourceContextHolder.push(dataSourceKey);
        }
    }

    @After("pointcut()")
    public void after(JoinPoint point) {
        //清理掉当前设置的数据源
        DynamicDataSourceContextHolder.clear();
    }
}

4、根据目录创建一个mysql连接、oracle连接,具体代码我就不贴了,主要是第3步骤的AOP起主要作用

5、测试

5.1、插入两个数据库数据

5.2、请求访问MYSQL的数据库连接http://localhost:8088/query

5.3、请求访问ORACLE的数据库连接http://localhost:8088/query2

6 、完事了~~~~~~

经验分享 程序员 微信小程序 职场和发展