try to optimize according to the hint

Author: qwjzka

src/main/java/org/apache/ibatis/plugin/Invocation.java

@@ -35,14 +35,16 @@ public class Invocation {
   private final Object target;
   private final Method method;
   private final Object[] args;
+  private final List<Interceptor> interceptors;
 
-  public Invocation(Object target, Method method, Object[] args) {
+  public Invocation(Object target, Method method, Object[] args, List<Interceptor> interceptors) {
     if (!targetClasses.contains(method.getDeclaringClass())) {
       throw new IllegalArgumentException("Method '" + method + "' is not supported as a plugin target.");
     }
     this.target = target;
     this.method = method;
     this.args = args;
+    this.interceptors = interceptors;
   }
 
   public Object getTarget() {
@@ -57,8 +59,14 @@ public Object[] getArgs() {
     return args;
   }
 
-  public Object proceed() throws InvocationTargetException, IllegalAccessException {
-    return method.invoke(target, args);
+  public Object proceed() throws Throwable {
+    int intSize = interceptors.size();
+    if ((intSize --)== 0) {
+      return method.invoke(target, args);
+    }
+    Interceptor interceptor = interceptors.get(intSize);
+    return interceptor.intercept(this);
+
   }
 
 }

src/main/java/org/apache/ibatis/plugin/Plugin.java

@@ -18,52 +18,72 @@
 import java.lang.reflect.InvocationHandler;
 import java.lang.reflect.Method;
 import java.lang.reflect.Proxy;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Map;
 import java.util.Set;
-
 import org.apache.ibatis.reflection.ExceptionUtil;
-import org.apache.ibatis.util.MapUtil;
+
 
 /**
  * @author Clinton Begin
  */
 public class Plugin implements InvocationHandler {
 
   private final Object target;
-  private final Interceptor interceptor;
-  private final Map<Class<?>, Set<Method>> signatureMap;
+  private final Map<Method, List<Interceptor>> interceptorMap;
 
-  private Plugin(Object target, Interceptor interceptor, Map<Class<?>, Set<Method>> signatureMap) {
+  private Plugin(Object target, Map<Method, List<Interceptor>> interceptorMap) {
     this.target = target;
-    this.interceptor = interceptor;
-    this.signatureMap = signatureMap;
+    this.interceptorMap = interceptorMap;
+  }
+
+  public Map<Method, List<Interceptor>> getInterceptorMap() {
+    return interceptorMap;
   }
 
   public static Object wrap(Object target, Interceptor interceptor) {
     Map<Class<?>, Set<Method>> signatureMap = getSignatureMap(interceptor);
     Class<?> type = target.getClass();
     Class<?>[] interfaces = getAllInterfaces(type, signatureMap);
     if (interfaces.length > 0) {
-      return Proxy.newProxyInstance(type.getClassLoader(), interfaces, new Plugin(target, interceptor, signatureMap));
-    }
+      if (Proxy.isProxyClass(target.getClass())) {
+        InvocationHandler invocationHandler = Proxy.getInvocationHandler(target);
+        if (invocationHandler instanceof Plugin) {
+          Map<Method, List<Interceptor>> interceptorMap = ((Plugin) invocationHandler).getInterceptorMap();
+          mapping(interceptor, signatureMap, interceptorMap);
+          return target;
+        }
+      }
+
+      Map<Method, List<Interceptor>> interceptorMap = new HashMap<>();
+      mapping(interceptor, signatureMap, interceptorMap);
+      return Proxy.newProxyInstance(type.getClassLoader(), interfaces, new Plugin(target, interceptorMap));}
     return target;
   }
 
   @Override
   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));
-      }
+      List<Interceptor> interceptors = interceptorMap.get(method);
+      if (interceptors != null) {
+        return new Invocation(target, method, args, interceptors).proceed();}
       return method.invoke(target, args);
     } catch (Exception e) {
       throw ExceptionUtil.unwrapThrowable(e);
     }
   }
 
+  private static void mapping(Interceptor interceptor, Map<Class<?>, Set<Method>> signatureMap, Map<Method, List<Interceptor>> interceptorMap) {
+    for (Set<Method> methods : signatureMap.values()) {
+      for (Method method : methods) {
+        interceptorMap.computeIfAbsent(method, (key) -> new ArrayList<>()).add(interceptor);
+      }
+    }
+  }
+
   private static Map<Class<?>, Set<Method>> getSignatureMap(Interceptor interceptor) {
     Intercepts interceptsAnnotation = interceptor.getClass().getAnnotation(Intercepts.class);
     // issue #251
@@ -74,7 +94,7 @@ private static Map<Class<?>, Set<Method>> getSignatureMap(Interceptor intercepto
     Signature[] sigs = interceptsAnnotation.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> methods = signatureMap.computeIfAbsent(sig.type(), k -> new HashSet<>());
       try {
         Method method = sig.type().getMethod(sig.method(), sig.args());
         methods.add(method);