Mybatis Plus 代码生成器

最近在学习Mybatis Plus，刚刚写好一个代码生成器的工具类，拿出来给大家分享一下，相比Mybatis要方便很多，不需要配置文件

package com.utils;

import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Scanner;

// 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
// MyBatis Plus(MP)
public class MPUtils {

 /**
 * <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();
 //MP提供的工具类，做非空判断
 if (StringUtils.isNotEmpty(ipt)) {
 return ipt;
 }
 }
 throw new MybatisPlusException("请输入正确的" + tip + "！");
 }

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

 // 全局配置
 GlobalConfig gc = new GlobalConfig();
 // 获取项目的绝对路径
 String projectPath = System.getProperty("user.dir");
 // 指定文件生成路径
 gc.setOutputDir(projectPath + "/src/main/java");
 // 作者（没什么用。。。）
 gc.setAuthor("jobob");
 gc.setOpen(false);
 mpg.setGlobalConfig(gc);

 // 数据源配置
 DataSourceConfig dsc = new DataSourceConfig();
 // 基本连接字符串
 dsc.setUrl("jdbc:mysql:///testdb");
 dsc.setDriverName("com.mysql.jdbc.Driver");
 dsc.setUsername("root");
 dsc.setPassword("ok");
 mpg.setDataSource(dsc);

 // 包配置
 PackageConfig pc = new PackageConfig();
 // 要生成的包名（隶属哪个模块）
 pc.setModuleName(scanner("模块名"));
 // 父级包名
 pc.setParent("com");
 mpg.setPackageInfo(pc);

 // 设置生成模板（默认就是这个，可以不写）
 mpg.setTemplate(new TemplateConfig());

 // 策略配置
 StrategyConfig strategy = new StrategyConfig();
 // 表名生成策略
 strategy.setNaming(NamingStrategy.nochange);
 // 字段名生成策略
 strategy.setColumnNaming(NamingStrategy.nochange);
 // 是否使用Lombok优化代码
 strategy.setEntityLombokModel(true);
 // controller类是否直接返回json
 strategy.setRestControllerStyle(true);
 // 要生成的表名（控制台输入）
 strategy.setInclude(scanner("表名"));
 strategy.setSuperEntityColumns("id");
 strategy.setControllerMappingHyphenStyle(true);
 mpg.setStrategy(strategy);
 // 切换引擎
 mpg.setTemplateEngine(new FreemarkerTemplateEngine());
 // 执行代码生成器
 mpg.execute();
 }

}