tnc-be/tnc-generate/src/main/java/org/jeecg/codegen/window/JeecgOneUtil.java

package org.jeecg.codegen.window;

import lombok.SneakyThrows;
import org.jeecg.codegen.generate.impl.CodeGenerateOne;
import org.jeecg.codegen.generate.pojo.TableVo;
import org.jeecg.common.constant.enums.CgformEnum;

import java.util.List;

/**
 * 单表代码生成
 */
public class JeecgOneUtil {
    public static void main(String[] args) {

        // 默认配置
//      defaultGen();

        // 枚举配置
//        enumGen();

        // 自定义配置
        genTwo();
    }


    /**
     * 使用默认配置生成代码
     */
    @SneakyThrows
    private static void defaultGen(){
        TableVo tableVo = new TableVo();
        // 表名
        tableVo.setTableName("sys_user");
        // 实体名
        tableVo.setEntityName("SysUser");
        // 包名
        tableVo.setEntityPackage("user");
        // 描述
        tableVo.setFtlDescription("系统用户");
        // 单表代码生成器
        CodeGenerateOne generate = new CodeGenerateOne(tableVo);
        // 默认配置 生成代码
        List<String> codeFileList = generate.generateCodeFile();

        for (String fileName : codeFileList) {
            System.out.println(fileName);
        }
    }

    /**
     * 枚举配置生成代码
     */
    @SneakyThrows
    private static void enumGen(){
        TableVo tableVo = new TableVo();
        // 表名
        tableVo.setTableName("test_note");
        // 实体名
        tableVo.setEntityName("TestNote");
        // 包名
        tableVo.setEntityPackage("note");
        // 描述
        tableVo.setFtlDescription("note");
        // 单表代码生成器
        CodeGenerateOne generate = new CodeGenerateOne(tableVo);
        // 自定义配置 代码生成路径
        String generateRootPath = "D:\\codeGen";
        // 代码风格 code，单表只有这一种选择
        String code = "one";
        CgformEnum cgformEnum = CgformEnum.getCgformEnumByConfig(code);
        // 模板目录
        String templatePath = cgformEnum.getTemplatePath();
        // 风格目录
        String stylePath = cgformEnum.getStylePath();

        List<String> codeFileList = generate.generateCodeFile(generateRootPath,templatePath,stylePath);

        for (String fileName : codeFileList) {
            System.out.println(fileName);
        }
    }

    /**
     * 自定义配置生成代码
     */
    @SneakyThrows
    private static void genTwo(){
        TableVo tableVo = new TableVo();
        // 表名
        tableVo.setTableName("sys_user");
        // 实体名
        tableVo.setEntityName("SysUser");
        // 包名
        tableVo.setEntityPackage("user");
        // 描述
        tableVo.setFtlDescription("系统用户");
        // 单表代码生成器
        CodeGenerateOne generate = new CodeGenerateOne(tableVo);
        // 自定义配置 代码生成路径
        String generateRootPath = "D:\\codeGen";
        // 模板目录
        String templatePath = "/jeecg/code-template-online";
        // 风格目录
        String stylePath = "default.two";

        List<String> codeFileList = generate.generateCodeFile(generateRootPath,templatePath,stylePath);

        for (String fileName : codeFileList) {
            System.out.println(fileName);
        }
    }
}