Druid多数据源配置(Datasurce动态切换)

原创

已于 2023-12-20 10:06:06 修改 · 5.8k 阅读

26 ·

CC 4.0 BY-SA版权

文章标签：

#spring boot #spring #java

于 2021-07-29 15:11:49 首次发布

本文详细介绍了如何在Spring Boot应用中使用Druid实现多数据源配置，并演示了如何通过注解动态切换数据源。此外，还展示了如何将ShardingJDBC集成到已有数据源中，实现在second库表级别的分片操作。

Druid多数据源配置(Datasurce动态切换)

提示: 本材料只做个人学习参考,不作为系统的学习流程,请注意识别!!!

一. 直接上代码

多数据源切换自定义注解

package com.gateway.admin.datasources.annotation;

import java.lang.annotation.*;
/**
 * 多数据源注解
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
   
   
    String name() default "";
}

多数据源,切面处理类

package com.gateway.admin.datasources;

import com.gateway.admin.datasources.annotation.DataSource;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.Ordered;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * 多数据源，切面处理类
 */
@Aspect
@Component
public class DataSourceAspect implements Ordered {
   
   
    protected Logger logger = LoggerFactory.getLogger(getClass());
    
    /**
	 * 针对上面注解做切面拦截
 	 */
    @Pointcut("@annotation(com.gateway.admin.datasources.annotation.DataSource)")
    public void dataSourcePointCut() {
   
   }

    @Around("dataSourcePointCut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
   
   
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        DataSource ds = method.getAnnotation(DataSource.class);
        if(ds == null){
   
   
        	//如果没有注解,使用默认数据源
            DynamicDataSource.setDataSource(DataSourceNames.FIRST);
        }else {
   
   
        	//根据注解中设置的数据源名称,选择对应的数据源
            DynamicDataSource.setDataSource(ds.name());
            logger.debug("set datasource is " + ds.name());
        }

        try {
   
   
            return point.proceed();
        } finally {
   
   
        	//清除数据源配置
            DynamicDataSource.clearDataSource();
        }
    }

    @Override
    public int getOrder() {
   
   
        return 1;
    }
}

动态数据源类,需要继承:AbstractRoutingDataSource

package com.gateway.admin.datasources;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
import javax.sql.DataSource;
import java.util.Map;

/**
 * 动态数据源
 */
public class DynamicDataSource extends AbstractRoutingDataSource {
   
   
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
   
   
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        super.setTargetDataSources(targetDataSources);
        super.afterPropertiesSet();
    }

    @Override
    protected Object determineCurrentLookupKey() {
   
   
        return getDataSource();
    }

    public static void setDataSource