Mybatis插件机制

news2024/11/30 0:37:58

什么是插件机制

插件插件，就是能在执行某个方法之前加入一些功能代码，有啥方法能够实现呢？当然是动态代理了，为啥要使用动态代理应为他是为了写框架扩展性必备的东西。只要定义一些接口或者类就行使用jdk自带的或者CGLIB之类的动态代理库完成方法的织入。

学习之前需要掌握的知识点

1、动态代理 2、注解 3、反射 4、责任链的设计模式

反射调用对象

public class Invocation {

  private final Object target; //目标对象
  private final Method method; //目标方法
  private final Object[] args; //目标参数

  public Object proceed() throws InvocationTargetException, IllegalAccessException {
    return method.invoke(target, args); //反射调用目标方法
  }
}

这里需要注意 proceed这个方法，因为这个方法的设计得很重要。

Interceptor 接口设计

public interface Interceptor {
  /**
   * 拦截执行方法
   * @param invocation
   * @return
   * @throws Throwable
   */
  Object intercept(Invocation invocation) throws Throwable;

  default Object plugin(Object target) {
    return Plugin.wrap(target, this); //对目标对象进行动态代理，this为拦截器
  }

  default void setProperties(Properties properties) {/*properties 属性注射方法*/
    // NOP
  }

}

责任链模式的设计

public class InterceptorChain {

  private final List<Interceptor> interceptors = new ArrayList<>(); //所有的执行器链

  public Object pluginAll(Object target) {//target 目标对象
    for (Interceptor interceptor : interceptors) {
      target = interceptor.plugin(target); //进行代理
    }
    return target;
  }

  public void addInterceptor(Interceptor interceptor) {
    interceptors.add(interceptor);
  }

  public List<Interceptor> getInterceptors() {
    return Collections.unmodifiableList(interceptors);
  }

}

注解设计

我们插件机制最细维度是希望能够为某个具体的方法进行功能上的增强。在java语法中，一个方法的标识需要有所属的Class对象，已经方法名称，已经方法形参类型。所以我们可以设计一个注解来说明要在那个类对象的那个方法进行插件机制的功能增强。

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target({})
public @interface Signature {
  /**
   * Returns the java type.
   *
   * @return the java type
   */
  Class<?> type(); //类对象

  /**
   * Returns the method name.
   *
   * @return the method name
   */
  String method();//方法

  /**
   * Returns java types for method argument.
   * @return java types for method argument
   */
  Class<?>[] args(); //参数的类对象数组
}

当然为了支持一个拦截器能够插入多个方法，我们需要再设计一个注解也就是支持 @Signature 注解的数组形式。

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface Intercepts {
  /**
   * Returns method signatures to intercept.
   *
   * @return method signatures
   */
  Signature[] value(); //数组形式
}

解析注解

注解提供了配置说明，但是具体这个配置说明的功能还需要我们去编写代码去实现。

提取注解 getSignatureMap

private static Map<Class<?>, Set<Method>> getSignatureMap(Interceptor interceptor) {
    Intercepts interceptsAnnotation = interceptor.getClass().getAnnotation(Intercepts.class); //获取注解
    // issue #251
    if (interceptsAnnotation == null) { //没有发现注解
      throw new PluginException("No @Intercepts annotation was found in interceptor " + interceptor.getClass().getName());
    }
    Signature[] sigs = interceptsAnnotation.value(); //获取注解的value
    Map<Class<?>, Set<Method>> signatureMap = new HashMap<>(); //创建存储对象
    for (Signature sig : sigs) { //遍历数组
      Set<Method> methods = MapUtil.computeIfAbsent(signatureMap, sig.type(), k -> new HashSet<>()); //判断这个类对象的Set<Method>是否存在，不存在就新建，存在就用原来的
      try {
        Method method = sig.type().getMethod(sig.method(), sig.args()); //获取目标方法
        methods.add(method); //塞入目标方法
      } catch (NoSuchMethodException e) {
        throw new PluginException("Could not find method on " + sig.type() + " named " + sig.method() + ". Cause: " + e, e); //方法找不到异常
      }
    }
    return signatureMap; //注解信息，提取完毕
  }

这样我们就获得了该拦截器插件能对那个类的那些方法进行增强的信息。那么接下来就是需要判断增强的目标对象是否在插件的类和方法声明内。

/**
   * 
   * @param type 增强的目标方法
   * @param signatureMap 增强允许的类和方法列表
   * @return 增强的接口
   */
  private static Class<?>[] getAllInterfaces(Class<?> type, Map<Class<?>, Set<Method>> signatureMap) {
    Set<Class<?>> interfaces = new HashSet<>(); //用来存放实现的接口， 因为采用的是jdk的动态代理
    while (type != null) {
      for (Class<?> c : type.getInterfaces()) { //获取当前类的所有接口
        if (signatureMap.containsKey(c)) { //接口是否在提取出来的注解信息中存在
          interfaces.add(c);
        }
      }
      type = type.getSuperclass(); //获取父类
    }
    return interfaces.toArray(new Class<?>[0]);
  }

进行代理

public static Object wrap(Object target, Interceptor interceptor) {
    Map<Class<?>, Set<Method>> signatureMap = getSignatureMap(interceptor); //提取Intercepts注解信息完毕
    Class<?> type = target.getClass(); //获取目标类类对象
    Class<?>[] interfaces = getAllInterfaces(type, signatureMap); //获取type 在signatrueMap中所有的接口集合
    if (interfaces.length > 0) {//说明有接口， 进行动态代理
      return Proxy.newProxyInstance(
          type.getClassLoader(), //当前类的类加载器
          interfaces, //接口
          new Plugin(target, interceptor, signatureMap)); //invocationhandler
    }
    return target;
  }

Invocationhandler 精髓实现

 public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    try {
      Set<Method> methods = signatureMap.get(method.getDeclaringClass()); //获取目标方法
      if (methods != null && methods.contains(method)) { //当前方法是否是在拦截中
        return interceptor.intercept(new Invocation(target, method, args)); //调用拦截器
      }
      return method.invoke(target, args); //不在拦截中直接调用
    } catch (Exception e) {
      throw ExceptionUtil.unwrapThrowable(e);
    }
  }