快捷導(dǎo)航

Spring Boot中自定義注解結(jié)合AOP實(shí)現(xiàn)主備庫切換問題

更新時間：2019年08月02日 09:50:03 作者：alterem

這篇文章主要介紹了Spring Boot中自定義注解+AOP實(shí)現(xiàn)主備庫切換的相關(guān)知識,本篇文章的場景是做調(diào)度中心和監(jiān)控中心時的需求，后端使用TDDL實(shí)現(xiàn)分表分庫，需要的朋友可以參考下

摘要:本篇文章的場景是做調(diào)度中心和監(jiān)控中心時的需求，后端使用TDDL實(shí)現(xiàn)分表分庫，需求:實(shí)現(xiàn)關(guān)鍵業(yè)務(wù)的查詢監(jiān)控,當(dāng)用Mybatis查詢數(shù)據(jù)時需要從主庫切換到備庫或者直接連到備庫上查詢,從而減小主庫的壓力，在本篇文章中主要記錄在Spring Boot中通過自定義注解結(jié)合AOP實(shí)現(xiàn)直接連接備庫查詢。

一.通過AOP 自定義注解實(shí)現(xiàn)主庫到備庫的切換

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

1.2 實(shí)現(xiàn)方法攔截器對自定義注解處理

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;
/**
 * 處理走備庫邏輯的注解
 */
@Component
public 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() + "方法調(diào)用失敗，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，實(shí)現(xiàn)掃描特定目錄下的注解，實(shí)現(xiàn)切面變成形成通知處理。示例代碼如下

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;
@Configuration
public 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;
  }
  //設(shè)置默認(rèn)的aop配置對應(yīng)的是原來的<aop:advisor>
  @Bean
  public Advisor druidAdvisor() {
    DefaultPointcutAdvisor defaultPointcutAdvisor = new DefaultPointcutAdvisor();
    defaultPointcutAdvisor.setPointcut(jdkRegexpMethodPointcut());
    defaultPointcutAdvisor.setAdvice(overallQueryInterceptor());
    return defaultPointcutAdvisor;
  }
}

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

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

總結(jié)

以上所述是小編給大家介紹的Spring Boot中自定義注解結(jié)合AOP實(shí)現(xiàn)主備庫切換,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！

您可能感興趣的文章: