java根据pojo生成mapper

使用命令生成class文件，执行就生成了

package com.main.util; import java.io.File; import java.io.FileWriter; import java.io.IOException; public class UtilMapper { /** * 批量生成POJO对应的Mapper文件 * @param pojoDir POJO文件目录路径 * @param daoDir DAO文件目录路径 * @param basePackage 基础包名 */ public static void generateMappers(String pojoDir, String daoDir, String basePackage) { File pojoDirectory = new File(pojoDir); File daoDirectory = new File(daoDir); // 确保DAO目录存在 if (!daoDirectory.exists()) { daoDirectory.mkdirs(); } // 获取所有POJO文件 File[] pojoFiles = pojoDirectory.listFiles((dir, name) -> name.endsWith(".java")); if (pojoFiles == null) { System.out.println("未找到POJO文件"); return; } int generatedCount = 0; int skippedCount = 0; for (File pojoFile : pojoFiles) { String pojoName = pojoFile.getName().replace(".java", ""); String mapperFileName = pojoName + "Mapper.java"; File mapperFile = new File(daoDirectory, mapperFileName); // 如果Mapper文件已存在，跳过 if (mapperFile.exists()) { System.out.println("跳过已存在的Mapper: " + mapperFileName); skippedCount++; continue; } // 生成Mapper文件内容 String mapperContent = generateMapperContent(pojoName, basePackage); try (FileWriter writer = new FileWriter(mapperFile)) { writer.write(mapperContent); System.out.println("生成Mapper文件: " + mapperFileName); generatedCount++; } catch (IOException e) { System.err.println("生成Mapper文件失败: " + mapperFileName + " - " + e.getMessage()); } } System.out.println("\n生成完成！"); System.out.println("成功生成: " + generatedCount + " 个"); System.out.println("跳过已存在: " + skippedCount + " 个"); System.out.println("总计处理: " + (generatedCount + skippedCount) + " 个"); } /** * 生成单个Mapper文件内容 * @param pojoName POJO类名 * @param basePackage 基础包名 * @return Mapper文件内容 */ private static String generateMapperContent(String pojoName, String basePackage) { StringBuilder content = new StringBuilder(); content.append("package ").append(basePackage).append(".dao;\n\n"); content.append("import ").append(basePackage).append(".pojo.").append(pojoName).append(";\n"); content.append("import tk.mybatis.mapper.common.Mapper;\n\n"); content.append("public interface ").append(pojoName).append("Mapper extends Mapper<").append(pojoName).append("> {\n"); content.append("}\n"); return content.toString(); } /** * 主方法，用于直接运行 */ public static void main(String[] args) { // 默认路径配置 - 相对于当前工作目录 String pojoDir = "src/main/java/com/main/pojo"; String daoDir = "src/main/java/com/main/dao"; String basePackage = "com.main"; System.out.println("开始批量生成Mapper文件..."); System.out.println("POJO目录: " + pojoDir); System.out.println("DAO目录: " + daoDir); System.out.println("基础包名: " + basePackage); System.out.println("========================================="); generateMappers(pojoDir, daoDir, basePackage); } }