子项目的逆向工程-代码生成器

2024-01-27 22:01:40

1 代码生成器

这个代码生成器，是在项目里的项目才能准确成功。刚好适合我的测试项目。

我这里继续学习，上一个进度，点击这里。

1.1 新增pom

        <!--mybatis-plus代码生成器模版（默认）-->
        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity-engine-core</artifactId>
        </dependency>
        <!--mybatis-plus代码生成器模版（freemarker）-->
        <dependency>
            <groupId>org.freemarker</groupId>
            <artifactId>freemarker</artifactId>
        </dependency>
        <!--baomidou-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.4.1</version>
        </dependency>

1.2 tools

package com.zhouqiang.demo.tools;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * @author ：zhouqiang
 * @date ：2021/8/23 15:52
 * @description：
 * @version: $
 */
public class CodeGenerator {
    /**
     * <p>
     * 读取控制台内容
     * </p>
     */
    public static String scanner(String tip) {
        Scanner scanner = new Scanner(System.in);
        StringBuilder help = new StringBuilder();
        help.append("请输入" + tip + "：");
        System.out.println(help.toString());
        if (scanner.hasNext()) {
            String ipt = scanner.next();
            if (ipt != null && ipt != "") {
                return ipt;
            }
        }
        throw new MybatisPlusException("请输入正确的" + tip + "！");
    }

    public static void main(String[] args) {
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();
        CodeGenerator codeGenerator = new CodeGenerator();


　　　　　　//这里适配的子项目的代码生成器，单体项目需要自己做出更改
　　　　　　// 这是baomidou的 官方文档 https://mp.baomidou.com/guide/crud-interface.html

        String projectName = "/demo-01";
        String packageName = "com.zhouqiang.demo";
        String projectPath = System.getProperty("user.dir") + projectName;

        mpg.setGlobalConfig(codeGenerator.getGlobalConfig(false, projectName));
        mpg.setDataSource(codeGenerator.getDataSourceConfig());

        // 包配置
        PackageConfig pc = codeGenerator.getPackageConfig(packageName);
        mpg.setPackageInfo(pc);

        // 自定义配置
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };

        // 如果模板引擎是 freemarker
        String templatePath = "/templates/mapper.xml.ftl";
        // 如果模板引擎是 velocity
        // String templatePath = "/templates/mapper.xml.vm";

        // 自定义输出配置
        List<FileOutConfig> focList = new ArrayList<>();
        // 自定义配置会被优先输出
        focList.add(new FileOutConfig(templatePath) {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
                return projectPath + "/src/main/resources/mapper/search"
                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        cfg.setFileOutConfigList(focList);
        mpg.setCfg(cfg);

        // 配置模板
        TemplateConfig templateConfig = new TemplateConfig();

        // 配置自定义输出模板
        //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
        // templateConfig.setEntity("templates/entity2.java");
        // templateConfig.setService();
        // templateConfig.setController();

        templateConfig.setXml(null);
        mpg.setTemplate(templateConfig);

        // 策略配置
        StrategyConfig strategy = codeGenerator.getStrategyConfig(pc.getModuleName());
        mpg.setStrategy(strategy);
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }

    /**
     * 设置包名
     *
     * @param packageName 父路径包名
     * @return PackageConfig 包名配置
     */
    private PackageConfig getPackageConfig(String packageName) {
        return new PackageConfig()
                .setParent(packageName)
                .setXml("mapper")
                .setMapper("mapper.search")
                .setController("controller")
                .setService("service")
                .setEntity("entity.search");
    }

    /**
     * 全局配置
     *
     * @param serviceNameStartWithI false
     * @return GlobalConfig
     */
    private GlobalConfig getGlobalConfig(boolean serviceNameStartWithI, String projectName) {
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setBaseColumnList(true)
                .setBaseResultMap(true)

                .setActiveRecord(true)
                .setAuthor("sys")
                //设置输出路径
                .setOutputDir(System.getProperty("user.dir") + "/" + projectName + "/src/main/java/")
//                .setFileOverride(true)
                .setOpen(false)
        ;
        if (!serviceNameStartWithI) {
            //设置service名
            globalConfig.setServiceName("%sService");
        }
        return globalConfig;
    }

    /**
     * 策略配置
     *
     * @param
     * @return StrategyConfig
     */
    private StrategyConfig getStrategyConfig(String moduleName) {
        return new StrategyConfig()
                // 全局大写命名
//                .setCapitalMode(true)
                .setEntityLombokModel(true)
                .setRestControllerStyle(true)
                //从数据库表到文件的命名策略
                .setNaming(NamingStrategy.underline_to_camel)
                .setColumnNaming(NamingStrategy.underline_to_camel)
                // 公共父类
                // strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
                .setControllerMappingHyphenStyle(true)
                .setTablePrefix(moduleName + "_")
                //需要生成的的表名，多个表名传数组
                .setInclude(scanner("表名，多个英文逗号分割").split(","));
    }

    /**
     * 配置数据源
     *
     * @return 数据源配置 DataSourceConfig
     */
    private DataSourceConfig getDataSourceConfig() {
        return getDataSourceConfig2();
    }


    private DataSourceConfig getDataSourceConfig2() {
        return new DataSourceConfig().setDbType(DbType.MYSQL)
                .setUrl("jdbc:mysql://***********?zeroDateTimeBehavior=convertToNull")
                .setUsername("***")
                .setPassword("*****")
                .setDriverName("com.mysql.cj.jdbc.Driver");
    }


}

红色部分，自己根据自己的需求做适当更改项目的名称与结构。