Spring Boot集成Druid实现多数据源的两种方式

package com.test.datasources.aspect; import java.lang.reflect.Method; 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 com.test.datasources.DataSourceNames; import com.test.datasources.DynamicDataSource; import com.test.datasources.annotation.DataSource; /** * 多数据源，切面处理类 */ @Aspect @Component public class DataSourceAspect implements Ordered { protected Logger logger = LoggerFactory.getLogger(getClass()); /* * @annotation匹配指定注解的方法 * @within匹配指定注解的类 * 注意：这里只拦截所注解的类，如果调用的是父类的方法，那么不会拦截，除非父类方法在子类中被覆盖。 */ @Pointcut("@annotation(com.test.datasources.annotation.DataSource) || @within(com.test.datasources.annotation.DataSource)") public void dataSourcePointCut() { } @Around("dataSourcePointCut()") public Object around(ProceedingJoinPoint point) throws Throwable { MethodSignature signature = (MethodSignature) point.getSignature(); Method method = signature.getMethod(); Class<?> dataClass = Class.forName(signature.getDeclaringTypeName()); DataSource dsMethod = method.getAnnotation(DataSource.class); DataSource dsClass = dataClass.getAnnotation(DataSource.class); if(dsMethod != null){ //方法优先，如果方法上存在注解，则优先使用方法上的注解 DynamicDataSource.setDataSource(dsMethod.value()); logger.debug("set datasource is " + dsMethod.value()); }else if(dsClass != null){ //其次类优先，如果类上存在注解，则使用类上的注解 DynamicDataSource.setDataSource(dsClass.value()); logger.debug("set datasource is " + dsClass.value()); }else{ //如果都不存在，则使用默认 DynamicDataSource.setDataSource(DataSourceNames.FIRST); logger.debug("set datasource is " + DataSourceNames.FIRST); } try { return point.proceed(); } finally { DynamicDataSource.clearDataSource(); logger.debug("clean datasource"); } } @Override public int getOrder() { return 1; } }