Mybatis-Plus生成器生成Controller、Service、Entity、Mapper、xml-超简单好用
package com.tunda.openai;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
import com.baomidou.mybatisplus.generator.config.po.TableField;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.IColumnType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;
import com.tunda.openai.model.entity.BaseEntity;
import java.util.ArrayList;
/**
* @author xiaozq
* @date 2023/6/1 11:23
* mybatis-plus code generator.
**/
public class MybatisPlusGenerator {
//生成文件所在项目路径
private static String baseProjectPath = "D:\\generatorCode";
//基本包名
private static String basePackage = "";
//作者(一般公司名简称)
private static String authorName = "tunda";
//要生成的表名，为空时生成所有表
private static String[] tables = {"t_goods_info"};
//table前缀:生成代码的时候可以忽略前缀，生成实体类：
/**例如: private static String prefix = "t_";
则t_goods_info的表，生成实体类则是GoodsInfo而不是TGoodsInfo
*/
private static String prefix = "";
//数据库配置四要素
private static String driverName = "";
private static String url = "jdbc:mysql://127.158.254.38:3306/tusnsda_gpt?serverTimezone=GMT%2B8&zeroDateTimeBehavior=CONVERT_TO_NULL&useUnicode=true&characterEncoding=utf-8&useSSL=false";
private static String username = "root";
private static String password = "root";
public static void main(String[] args) {
// 代码生成器
AutoGenerator mpg = new AutoGenerator();
// 1、全局配置
GlobalConfig gc = new GlobalConfig();
//可以将生成得代码单独放在文件夹里，检查完没问题在手动将包复制到项目中
gc.setOutputDir(baseProjectPath);
gc.setAuthor(authorName);// 生成实体类的文档注解中作者名
gc.setOpen(true);// 生成代码后自动打开生成目录文件
gc.setFileOverride(true); // 是否覆盖原来生成的
gc.setIdType(IdType.AUTO);// 指定生成的主键类型：自增长
gc.setDateType(DateType.ONLY_DATE);
gc.setActiveRecord(false);
mpg.setGlobalConfig(gc);
// 2、数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl(url);
dsc.setDriverName(driverName);
dsc.setUsername(username);
dsc.setPassword(password);
dsc.setDbType(DbType.MYSQL); // mysql数据库
gc.setServiceName("%sService"); //去掉Service接口的首字母'I'
dsc.setTypeConvert(new MySqlTypeConvert() {
@Override
public IColumnType processTypeConvert(GlobalConfig globalConfig, TableField tableField) {
return super.processTypeConvert(globalConfig, tableField);
}
});
mpg.setDataSource(dsc);
// 3、包配置
PackageConfig pc = new PackageConfig();
pc.setModuleName(null)// 生成的controller路径映射为一个"/",覆盖默认的"//"
.setParent(basePackage)// 生成controller，service，xml并设置路径。
.setEntity("entity")
.setMapper("mapper")
.setService("service")
.setServiceImpl("")
.setController("controller");
mpg.setPackageInfo(pc);
// 5、策略配置
StrategyConfig strategy = new StrategyConfig();
strategy.setCapitalMode(true);// 开启全局大写命名
strategy.setNaming(NamingStrategy.underline_to_camel);// 实体类名驼峰
strategy.setColumnNaming(NamingStrategy.underline_to_camel);// 属性名驼峰
strategy.setInclude(tables);// 设置要映射的表名注解@Table
strategy.setSuperEntityClass(BaseEntity.class); // 预先定义一个BaseEntity类实现Serializable接口
strategy.setEntityLombokModel(true); // 使用 lombok
strategy.setRestControllerStyle(true);// 开启 restful风格
strategy.setEntityTableFieldAnnotationEnable(false);// 不生成实体类字段swagger类型注解,而是采用文档类型注解：/**xxx*/
if (prefix != null) strategy.setTablePrefix(prefix); // 表前缀
if (ArrayUtils.isNotEmpty(tables)) strategy.setInclude(tables); // 表名
// 6、自动填充配置：非必须
TableFill createTime = new TableFill("create_time", FieldFill.INSERT);
TableFill updateTime = new TableFill("update_time", FieldFill.INSERT_UPDATE);
ArrayList<TableFill> tableFills = new ArrayList<>();
tableFills.add(createTime);
tableFills.add(updateTime);
strategy.setTableFillList(tableFills);
strategy.setLogicDeleteFieldName("deleted");// 指定逻辑删除字段
mpg.setStrategy(strategy);
mpg.setTemplateEngine(new VelocityTemplateEngine());
// 执行生成器
mpg.execute();
}
}