使用Mybatis怎么實現一個攔截器

這篇文章給大家介紹使用Mybatis怎么實現一個攔截器，內容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

//服務員的接口
public interface Waiter {
  void serve();
}
//服務員的實現
public class WaiterImpl implements Waiter {
  @Override
  public void serve() {
    System.out.println("服務中...");
  }
}
//需要代理的對象處理器
class WaitInvocationHandler implements InvocationHandler {
  private Waiter waiter;

  public WaitInvocationHandler(Waiter waiter1) {
    waiter = waiter1;
  }

  @Override
  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    System.out.println("你好");
    Object invoke = method.invoke(waiter, args);
    System.out.println("再見");
    return invoke;
  }
}
public class App {
  //普通的實現
  @Test
  public void fun() {
    Waiter waiter = new WaiterImpl();
    waiter.serve();
  }

  @Test
  public void fun1() {
    Waiter a = new WaiterImpl();
    ClassLoader classLoader = getClass().getClassLoader();
    Class[] classes = {Waiter.class};
    //生成代理對象
    Waiter waiterproxy = (Waiter) Proxy.newProxyInstance(classLoader, classes, new WaitInvocationHandler(a));
    waiterproxy.serve();
  }
}

攔截器

V1

我現在要在函數執行前后記錄日志操作，考慮需要在代理方法那里定義個攔截器的接口，如下代碼所示

//攔截器 V1 版本
public interface MyInterceptorV1 {
  void interceptor();
}
//代理對象變成這個
public class TargetProxyV1 implements InvocationHandler {

  private Target target;

  private MyInterceptorV1 myInterceptor;

  public TargetProxyV1(Target target, MyInterceptorV1 myInterceptor) {
    this.target = target;
    this.myInterceptor = myInterceptor;
  }

  @Override
  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    myInterceptor.interceptor();
    return method.invoke(target, args);
  }
}

這是最簡單的版本，但是我們很多時候需要攔截參數等根據參數判斷攔不攔截等，代碼更新如下

public interface MyInterceptorV1 {
  void interceptor(Method method, Object[] args);
}

@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
  myInterceptor.interceptor(method, args);
  return method.invoke(target, args);
}

V2

似乎上面的方案很完美

廢話肯定不完美，不然怎么會有這段

沒錯，第二段代碼并不是很優雅，有方法參數重復，可以考慮將三者抽出來，直接在攔截器的實現里寫上 method.invoke(target, args); ，如下代碼所示

@Getter
@Setter
@AllArgsConstructor
public class MyInvocation {
  private Object target;
  private Method method;
  private Object[] args;

  public Object proceed() throws InvocationTargetException, IllegalAccessException {
    return method.invoke(target, args);
  }
}
//沒錯這就是 V2 版本
public interface MyInterceptorV2 {
  Object interceptor(MyInvocation invocation) throws Throwable;
}

關于使用Mybatis怎么實現一個攔截器就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。