SpringBoot中怎么通過自定義注解實現主備庫切換

今天就跟大家聊聊有關SpringBoot中怎么通過自定義注解實現主備庫切換，可能很多人都不太了解，為了讓大家更加了解，小編給大家總結了以下內容，希望大家根據這篇文章可以有所收獲。

一.通過AOP 自定義注解實現主庫到備庫的切換

1.1 自定義注解

自定義注解如下代碼所示

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 switch3Backup() default false;}

1.2 實現方法攔截器對自定義注解處理

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.switch3Backup()) {        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;  }}

1.3 配置OverallQueryConfiguration

在Spring Boot中裝配AOP Bean，實現掃描特定目錄下的注解，實現切面變成形成通知處理。示例代碼如下

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;  }}

1.4 如何使用注解從主庫到備庫的切換

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

看完上述內容，你們對SpringBoot中怎么通過自定義注解實現主備庫切換有進一步的了解嗎？如果還想了解更多知識或者相關內容，請關注億速云行業資訊頻道，感謝大家的支持。