Spring Boot中自定义注解结合AOP实现主备库切换问题

import java.lang.annotation.ElementType;import java.lang.annotation.Retention;import java.lang.annotation.RetentionPolicy;import java.lang.annotation.Target;@Retention(RetentionPolicy.RUNTIME)@Target(ElementType.METHOD)public @interface SwitchDataBase {  boolean switch2Backup() default false;}

import java.lang.reflect.Method;import java.util.Arrays;import org.aopalliance.intercept.MethodInterceptor;import org.aopalliance.intercept.MethodInvocation;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.stereotype.Component;/** * 处理走备库逻辑的注解 */@Componentpublic class SwitchDataBaseInterceptor implements MethodInterceptor {  private final Logger log = LoggerFactory.getLogger(SwitchDataBaseInterceptor.class);  @Override  public Object invoke(MethodInvocation invocation) throws Throwable {    Method method = invocation.getMethod();    SwitchDataBase annotation = getAnnotation(method);    if (annotation == null) {      return invocation.proceed();    }    Object val = null;    if(!ThreadLocalMap.containsKey(GroupDataSourceRouteHelper.DATASOURCE_INDEX)) {      if (annotation.switch2Backup()) {        log.info("query back up DB, method: " + method.getName());        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(1, true);      } else {        log.info("query primary DB, method: " + method.getName());        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0, true);      }    }    try {      val = invocation.proceed();    } catch (Exception e) {      log.info(method.getDeclaringClass().getName() + "." +          invocation.getMethod().getName() + "方法调用失败，arguments：" +          Arrays.toString(invocation.getArguments()));      throw new RuntimeException(e);    } finally {      GroupDataSourceRouteHelper.removeGroupDataSourceIndex();    }    return val;  }  /**   * 找方法上面声明的注解   */  private SwitchDataBase getAnnotation(Method method) {    if (method.isAnnotationPresent(SwitchDataBase.class)) {      return method.getAnnotation(SwitchDataBase.class);    }    return null;  }}

import com.wdk.wms.annotation.SwitchDataBaseInterceptor;import org.springframework.aop.Advisor;import org.springframework.aop.support.DefaultPointcutAdvisor;import org.springframework.aop.support.JdkRegexpMethodPointcut;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;@Configurationpublic class SwitchDataBaseConfiguration {  @Bean(name = "overallQueryInterceptor")  public SwitchDataBaseInterceptor overallQueryInterceptor() {    return new SwitchDataBaseInterceptor();  }  //添加aop的pointcut  @Bean(name = "jdkRegexpMethodPointcut")  public JdkRegexpMethodPointcut jdkRegexpMethodPointcut() {    JdkRegexpMethodPointcut jdkRegexpMethodPointcut = new JdkRegexpMethodPointcut();    jdkRegexpMethodPointcut.setPatterns("com.wdk.wms.mapper.*");    return jdkRegexpMethodPointcut;  }  //设置默认的aop配置对应的是原来的<aop:advisor>  @Bean  public Advisor druidAdvisor() {    DefaultPointcutAdvisor defaultPointcutAdvisor = new DefaultPointcutAdvisor();    defaultPointcutAdvisor.setPointcut(jdkRegexpMethodPointcut());    defaultPointcutAdvisor.setAdvice(overallQueryInterceptor());    return defaultPointcutAdvisor;  }}

@SwitchDataBase(switch2Backup = true)List<ConsumerNoticeMsg> listByTemplateOver3(@Param("templates") List<Integer> templates);