预期把mybatis打包进来

Author: miemieYaho

mybatis-plus-core/src/test/java/com/baomidou/mybatisplus/code/Overwrite.java

@@ -0,0 +1,39 @@
+package com.baomidou.mybatisplus.code;
+
+import lombok.Builder;
+import lombok.Data;
+import lombok.Singular;
+
+import java.util.List;
+
+/**
+ * @author miemie
+ * @since 2025/9/1
+ */
+@Data
+@Builder
+public class Overwrite {
+    private String front;
+    private String behind;
+    private int interval;
+    /*  */
+    @Singular
+    private List<Content> contents;
+    @Singular("addImport")
+    private List<String> imports;
+
+    @Data
+    @Builder
+    public static class Content {
+        @Builder.Default
+        private Operate operate = Operate.INSERT;
+        private int frontDown;
+        private int behindUp;
+        private String code;
+    }
+
+    enum Operate {
+        INSERT,
+        DELETE
+    }
+}

mybatis-plus-core/src/test/java/com/baomidou/mybatisplus/code/OverwriteFile.java

@@ -0,0 +1,21 @@
+package com.baomidou.mybatisplus.code;
+
+import lombok.Data;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.Function;
+
+/**
+ * @author miemie
+ * @since 2025/9/1
+ */
+@Data
+public abstract class OverwriteFile {
+
+    private List<Overwrite> steps = new ArrayList<>();
+
+    public void addStep(Function<Overwrite.OverwriteBuilder, Overwrite.OverwriteBuilder> function) {
+        this.steps.add(function.apply(Overwrite.builder()).build());
+    }
+}

mybatis-plus-core/src/test/java/com/baomidou/mybatisplus/code/OverwriteRunner.java

@@ -0,0 +1,64 @@
+package com.baomidou.mybatisplus.code;
+
+import com.baomidou.mybatisplus.code.sub.MapperMethod;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.jar.JarEntry;
+import java.util.jar.JarFile;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+import java.util.stream.Collectors;
+
+/**
+ * @author miemie
+ * @since 2025/9/1
+ */
+public class OverwriteRunner {
+
+    public static void main(String[] args) throws Exception {
+        List<OverwriteFile> files = List.of(new MapperMethod());
+        Map<String, OverwriteFile> map = finJar(files);
+        map.forEach((k, v) -> {
+            System.out.println(k);
+        });
+    }
+
+    private static Map<String, OverwriteFile> finJar(List<OverwriteFile> files) throws IOException {
+        Map<String, OverwriteFile> map = files.stream().collect(Collectors.toMap(i -> i.getClass().getSimpleName(), i -> i));
+        Map<String, OverwriteFile> result = new HashMap<>();
+        String ver = findVer();
+        Path jarPath = Paths.get(System.getProperty("user.home"), ".m2", "repository", "org", "mybatis", "mybatis",
+            ver, "mybatis-" + ver + "-sources.jar");
+        try (JarFile jarFile = new JarFile(jarPath.toFile())) {
+            for (JarEntry entry : jarFile.stream().toList()) {
+                String name = entry.getName();
+                if (!name.endsWith(".java")) {
+                    continue;
+                }
+                String className = name.substring(name.lastIndexOf("/") + 1, name.length() - 5);
+                if (map.containsKey(className)) {
+                    result.put(name, map.get(className));
+                }
+            }
+        }
+        return result;
+    }
+
+    private static String findVer() throws IOException {
+        String userDir = System.getProperty("user.dir");
+        String content = new String(Files.readAllBytes(Paths.get(userDir + "/build.gradle")));
+        // 定义正则表达式，匹配 mybatisVersion = '3.5.19'
+        Pattern pattern = Pattern.compile("mybatisVersion\\s*=\\s*['\"]([\\d.]+)['\"]");
+        Matcher matcher = pattern.matcher(content);
+        if (matcher.find()) {
+            return matcher.group(1);
+        }
+        throw new RuntimeException();
+    }
+}

mybatis-plus-core/src/test/java/com/baomidou/mybatisplus/code/sub/MapperMethod.java

@@ -0,0 +1,59 @@
+package com.baomidou.mybatisplus.code.sub;
+
+import com.baomidou.mybatisplus.code.Overwrite;
+import com.baomidou.mybatisplus.code.OverwriteFile;
+
+/**
+ * {@link org.apache.ibatis.binding.MapperMethod}
+ *
+ * @author miemie
+ * @since 2025/9/1
+ */
+public class MapperMethod extends OverwriteFile {
+
+    public MapperMethod() {
+        // import
+        addStep(i -> i
+            .addImport("com.baomidou.mybatisplus.core.metadata.IPage")
+            .addImport("com.baomidou.mybatisplus.core.toolkit.Assert"));
+        // execute
+        addStep(i -> i
+            .front("result = executeForCursor(sqlSession, args);")
+            .interval(8)
+            .behind("case FLUSH:")
+            .content(Overwrite.Content.builder()
+                .code("""
+                    if (IPage.class.isAssignableFrom(method.getReturnType())) {
+                        result = executeForIPage(sqlSession, args);
+                    } else {
+                    """)
+                .frontDown(1).build())
+            .content(Overwrite.Content.builder()
+                .code("}")
+                .behindUp(3)
+                .build())
+        );
+        addStep(i -> i
+            .behind("private Object rowCountResult(int rowCount) {")
+            .content(Overwrite.Content.builder()
+                .code("""
+                    @SuppressWarnings("all")
+                    private <E> Object executeForIPage(SqlSession sqlSession, Object[] args) {
+                        IPage<E> result = null;
+                        for (Object arg : args) {
+                            if (arg instanceof IPage) {
+                                result = (IPage<E>) arg;
+                                break;
+                            }
+                        }
+                        Assert.notNull(result, "can't found IPage for args!");
+                        Object param = method.convertArgsToSqlCommandParam(args);
+                        List<E> list = sqlSession.selectList(command.getName(), param);
+                        result.setRecords(list);
+                        return result;
+                    }
+                    """)
+                .behindUp(2).build())
+        );
+    }
+}