Spring+SpringMvc+MybatisPlus+Aop（自定义注解）动态切换数据源

package com.hf.db.aspect; import com.hf.db.utils.db.DataSourceCheckUtils; import com.hf.db.utils.db.DynamicDataSourceHolder; 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.springframework.stereotype.Component; import java.lang.reflect.Method; /** * 功能描述: <br> * 〈 * Aop动态切换数据源的具体实现 * 〉 * * @className: ChooseDataSource * @author: hf * @version: 1.0.0 * @date: 2018/12/25 18:21 * <p> * History: * <author:> <time:> <version:> <desc:> */ @Aspect @Component public class ChooseDataSourceAspect { /** * 功能描述: <br> * 〈 * 配置接入点,切入controller层 * 〉 * * @className: ChooseDataSourceAspect * @author: hf * @version: 1.0.0 * @date: 2018/12/25 18:25 * @param: [] * @return: void * <p> * History: * <author:> <time:> <version:> <desc:> */ @Pointcut("execution(* com.hf.db.controller..*.*(..))") private void controllerAspect() { } /** * 功能描述: <br> * 〈 * 配置切面动态切换数据源 * 〉 * * @className: ChooseDataSourceAspect * @author: hf * @version: 1.0.0 * @date: 2018/12/25 18:25 * @param: [joinPoint 切点] * @return: java.lang.Object * <p> * History: * <author:> <time:> <version:> <desc:> */ @Around("controllerAspect()") public Object permission(ProceedingJoinPoint joinPoint) throws Throwable { Object target = joinPoint.getTarget(); Object[] args = joinPoint.getArgs(); Method method = getMethod(joinPoint, args); //获取数据库名称参数 ChooseDataSource chooseDataSource = method.getAnnotation(ChooseDataSource.class); if (chooseDataSource != null) { String dataSourceName = chooseDataSource.dataSourceName(); //检查数据库名称是否存在 if (DataSourceCheckUtils.check(dataSourceName)) { DynamicDataSourceHolder.putDataSourceName(dataSourceName); } else { DynamicDataSourceHolder.putDataSourceName("server_1"); } } return joinPoint.proceed(); } /** * 功能描述: <br> * 〈 * 获取切面信息 * 〉 * * @className: ChooseDataSourceAspect * @author: hf * @version: 1.0.0 * @date: 2018/12/7 15:28 * @param: [joinPoint 切点, args 参数] * @return: java.lang.reflect.Method * <p> * History: * <author:> <time:> <version:> <desc:> */ private Method getMethod(ProceedingJoinPoint joinPoint, Object[] args) throws NoSuchMethodException { String methodName = joinPoint.getSignature().getName(); Class clazz = joinPoint.getTarget().getClass(); Method[] methods = clazz.getMethods(); for (Method method : methods) { if (methodName.equals(method.getName())) { return method; } } return null; } }