插件Pointcut定义

Author: FuriousPws002

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

@@ -0,0 +1,35 @@
+package org.apache.ibatis.plugin;
+
+import java.lang.reflect.Method;
+
+/**
+ * @author furious 2024/4/18
+ */
+public class Invocation {
+
+    private final Object target;
+    private final Method method;
+    private final Object[] args;
+
+    public Invocation(Object target, Method method, Object[] args) {
+        this.target = target;
+        this.method = method;
+        this.args = args;
+    }
+
+    public <T> T getTarget() {
+        return (T)target;
+    }
+
+    public Method getMethod() {
+        return method;
+    }
+
+    public Object[] getArgs() {
+        return args;
+    }
+
+    public <T> T proceed() throws Exception {
+        return (T)method.invoke(target, args);
+    }
+}

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

@@ -0,0 +1,44 @@
+package org.apache.ibatis.plugin;
+
+import java.lang.reflect.Method;
+import java.util.Arrays;
+import java.util.stream.Collectors;
+
+/**
+ * @author furious 2024/4/18
+ */
+public class PointcutDefinition {
+
+    private final Class<?> type;
+    private final Method method;
+    private final Class<?>[] args;
+    private final String id;
+
+    public PointcutDefinition(Method method, PointcutRegistry registry) {
+        this.type = method.getDeclaringClass();
+        this.method = method;
+        this.args = method.getParameterTypes();
+        this.id = id(method);
+        registry.getDefinitions().put(id, this);
+    }
+
+    public static String id(Method method) {
+        return String.join("_", method.getDeclaringClass().getSimpleName(), method.getName(), Arrays.stream(method.getParameterTypes()).map(Class::getSimpleName).collect(Collectors.joining("_")));
+    }
+
+    public Class<?> getType() {
+        return type;
+    }
+
+    public Method getMethod() {
+        return method;
+    }
+
+    public Class<?>[] getArgs() {
+        return args;
+    }
+
+    public String getId() {
+        return id;
+    }
+}

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

@@ -0,0 +1,44 @@
+package org.apache.ibatis.plugin;
+
+import java.util.Arrays;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+import org.apache.ibatis.executor.Executor;
+import org.apache.ibatis.executor.parameter.ParameterHandler;
+import org.apache.ibatis.executor.resultset.ResultSetHandler;
+import org.apache.ibatis.executor.statement.StatementHandler;
+
+/**
+ * @author furious 2024/4/18
+ */
+public class PointcutRegistry {
+
+    public static final String Executor_update_MappedStatement_Object = "Executor_update_MappedStatement_Object";
+    public static final String Executor_query_MappedStatement_Object = "Executor_query_MappedStatement_Object";
+    public static final String ParameterHandler_getParameterObject_ = "ParameterHandler_getParameterObject_";
+    public static final String ParameterHandler_setParameters_PreparedStatement = "ParameterHandler_setParameters_PreparedStatement";
+    public static final String ResultSetHandler_handleResultSets_Statement = "ResultSetHandler_handleResultSets_Statement";
+    public static final String StatementHandler_parameterize_Statement = "StatementHandler_parameterize_Statement";
+    public static final String StatementHandler_update_Statement = "StatementHandler_update_Statement";
+    public static final String StatementHandler_prepare_Connection_Integer = "StatementHandler_prepare_Connection_Integer";
+    public static final String StatementHandler_query_Statement = "StatementHandler_query_Statement";
+    public static final String StatementHandler_getBoundSql_ = "StatementHandler_getBoundSql_";
+
+    private final Map<String, PointcutDefinition> definitions = new LinkedHashMap<>();
+
+    public PointcutRegistry() {
+        Arrays.stream(Executor.class.getMethods()).forEach(m -> new PointcutDefinition(m, this));
+        Arrays.stream(ParameterHandler.class.getMethods()).forEach(m -> new PointcutDefinition(m, this));
+        Arrays.stream(ResultSetHandler.class.getMethods()).forEach(m -> new PointcutDefinition(m, this));
+        Arrays.stream(StatementHandler.class.getMethods()).forEach(m -> new PointcutDefinition(m, this));
+    }
+
+    public Map<String, PointcutDefinition> getDefinitions() {
+        return definitions;
+    }
+
+    public PointcutDefinition getDefinition(String key) {
+        return definitions.get(key);
+    }
+}