com.github.fanzezhen.common.mp.generator.GeneratorTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-mp Show documentation
There is a newer version: 3.0.1-2023.10.13
package com.github.fanzezhen.common.mp.generator;

import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.*;

import java.util.Collections;

/**
 * @author zezhen.fan
 */
public class GeneratorTool {
    /**
     * 
     * 读取控制台内容
     * 
     */
    public static void generator(GeneratorBean generatorBean) {
        String dbUrl = StrUtil.isBlank(generatorBean.getDbUrl()) ? generatorBean.scanner("数据库链接（如：jdbc:mysql://localhost:3306/demo）") : generatorBean.getDbUrl();
        String dbUsername = StrUtil.isBlank(generatorBean.getDbUsername()) ? generatorBean.scanner("数据库用户名称") : generatorBean.getDbUsername();
        String dbPassword = StrUtil.isBlank(generatorBean.getDbPassword()) ? generatorBean.scanner("数据库用户密码") : generatorBean.getDbPassword();
        FastAutoGenerator generator = FastAutoGenerator.create(dbUrl, dbUsername, dbPassword);
        generator.globalConfig(builder -> builder
                // 设置作者
                .author(System.getProperty("user.name"))
                // 开启 swagger 模式
                .enableSwagger()
                // 生成结束时不打开文件管理器
                .disableOpenDir()
                // 指定输出目录
                .outputDir(System.getProperty("user.dir") + StrUtil.SLASH + generatorBean.getJavaDir()))
                .packageConfig(builder -> builder
                        // 设置父包名
                        .parent(generatorBean.getParentPackageName())
                        // 设置mapperXml生成路径
                        .pathInfo(Collections.singletonMap(OutputFile.xml, System.getProperty("user.dir") + StrUtil.SLASH + generatorBean.getMapperDir())))
                .strategyConfig(builder -> {
                    // 设置需要生成的表名
                    builder.addInclude(generatorBean.getTableNameList());
                    if (ArrayUtil.isNotEmpty(generatorBean.getIgnoreTablePrefix())) {
                        // 设置过滤表前缀
                        builder.addTablePrefix(generatorBean.getIgnoreTablePrefix());
                    }
                    builder.entityBuilder()
                            .superClass(generatorBean.getSuperEntityClass())
                            .formatFileName(generatorBean.getFormatFileName())
                            .logicDeleteColumnName(generatorBean.getLogicDeleteColumnName())
                            .enableLombok().enableChainModel().enableTableFieldAnnotation();
                })
                // 使用Freemarker引擎模板，默认的是Velocity引擎模板
//                .templateEngine(new FreemarkerTemplateEngine())
                .execute();
    }

}