简单的代码生成器-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_40820739/article/details/112988610

这个代码生成器主要用于根据配置文件自动生成MyBatis的Mapper、Service和ServiceImpl类。它读取逆向工程配置文件，使用MyBatisGenerator库进行生成，并提供替换模板字符串的功能，便于快速创建基于特定实体类的新文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简单的代码生成器

public class Generator {
    public void generator() throws Exception {
        List<String> warnings = new ArrayList<String>();
        boolean overwrite = true;
        /** 指向逆向工程配置文件 */
        File configFile = new File("meng-admin/src/main/resources/generatorConfig.xml");
        ConfigurationParser parser = new ConfigurationParser(warnings);
        Configuration config = parser.parseConfiguration(configFile);
        DefaultShellCallback callback = new DefaultShellCallback(overwrite);
        MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
        myBatisGenerator.generate(null);
    }

    public static void generMapper(String className) throws Exception {
        BufferedReader br = new BufferedReader(new FileReader("meng-admin/src/main/java/me/yexinkai/dao/DemoDao.java"));
        BufferedWriter wr =
            new BufferedWriter(new FileWriter("meng-admin/src/main/java/me/yexinkai/dao/" + className + "Dao.java"));
        String line = null;
        while ((line = br.readLine()) != null) {
            line = line.replace("DemoQueryCriteria", className);
            wr.write(line);
            wr.newLine();
        }
        wr.close();
        br.close();
    }

    public static void generService(String className) throws Exception {
        BufferedReader br =
            new BufferedReader(new FileReader("meng-admin/src/main/java/me/yexinkai/service/DemoService.java"));
        BufferedWriter wr = new BufferedWriter(
            new FileWriter("meng-admin/src/main/java/me/yexinkai/service/" + className + "Service.java"));
        String line = null;
        while ((line = br.readLine()) != null) {
            line = line.replace("DemoQueryCriteria", className);
            wr.write(line);
            wr.newLine();
        }
        wr.close();
        br.close();
    }

    public static void generateServiceImpl(String className) throws Exception {
        // 把传递过来的实体类的名字政成小写一份
        String lowerObjName = className.substring(0, 1).toLowerCase() + className.substring(1, className.length());
        // 创建文件输入流
        BufferedReader br = new BufferedReader(
            new FileReader("meng-admin/src/main/java/me/yexinkai/service/impl/DemoServiceImpl.java"));
        BufferedWriter bw = new BufferedWriter(
            new FileWriter("meng-admin/src/main/java/me/yexinkai/service/impl/" + className + "ServiceImpl.java"));
        String line = null;
        while ((line = br.readLine()) != null) {
            line = line.replace("Demo", className);
            line = line.replace("demo", lowerObjName);
            bw.write(line);
            bw.newLine();
        }
        bw.close();
        br.close();
    }

    public static void generSrc(String className) throws Exception {
        generMapper(className);
        generService(className);
        generateServiceImpl(className);
    }

    public static void main(String[] args) throws Exception {
        Generator generator = new Generator();
        generator.generator();
        generSrc("Agent");
    }
}