MyBatis-Plus 代码生成类

ゞ浴缸里的玫瑰 2023-10-04 20:57 2阅读 0赞

生成po  
  
PO即持久对象(Persistent Object)，它们是由一组属性和属性的get和set方法组成，PO对应于数据库的表。  
在开发持久层代码时需要根据数据表编写PO类，在实际开发中通常使用代码生成器（工具）生成PO类的代码。

1、我使用的mysql，下面是我的数据库表。

![a1d2dfffa44e4b029ca2e750341671de.png][]

2、以下是我的代码实现：

注意下面有几处地方需要你自己修改：我会标红

package com.xuecheng.generator;

import com.baomidou.mybatisplus.annotation.DbType;  
import com.baomidou.mybatisplus.annotation.FieldFill;  
import com.baomidou.mybatisplus.generator.AutoGenerator;  
import com.baomidou.mybatisplus.generator.config.\*;  
import com.baomidou.mybatisplus.generator.config.po.TableFill;  
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;  
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Arrays;

/\*\*  
\* MyBatis-Plus 代码生成类  
\*/  
public class ContentCodeGenerator \{

// TODO 修改服务名以及数据表名 我自己数据的名字是 xc\_content  
private static final String SERVICE\_NAME = "content";

//数据库账号  
private static final String DATA\_SOURCE\_USER\_NAME = "root";  
//数据库密码  
private static final String DATA\_SOURCE\_PASSWORD = "123456";  
//生成的表  
private static final String\[\] TABLE\_NAMES = new String\[\]\{

//这里是你自己数据的表  
//  "mq\_message",  
// "mq\_message\_history"  
"course\_base",  
"course\_market",  
"teachplan",  
"teachplan\_media",  
"course\_teacher",  
// "course\_publish",  
// "course\_publish\_pre"  
// "course\_category"  
\};

// TODO 默认生成entity，需要生成DTO修改此变量  
// 一般情况下要先生成 DTO类 然后修改此参数再生成 PO 类。  
private static final Boolean IS\_DTO = false;

public static void main(String\[\] args) \{  
// 代码生成器  
AutoGenerator mpg = new AutoGenerator();  
// 选择 freemarker 引擎，默认 Velocity  
mpg.setTemplateEngine(new FreemarkerTemplateEngine());  
// 全局配置  
GlobalConfig gc = new GlobalConfig();  
gc.setFileOverride(true);  
//生成路径 自己项目的路径  
gc.setOutputDir(System.getProperty("user.dir") + "/xuecheng-plus-generator/src/main/java");  
gc.setAuthor("itcast");  
gc.setOpen(false);  
gc.setSwagger2(false);  
gc.setServiceName("%sService");  
gc.setBaseResultMap(true);  
gc.setBaseColumnList(true);

if (IS\_DTO) \{  
gc.setSwagger2(true);  
gc.setEntityName("%sDTO");  
\}  
mpg.setGlobalConfig(gc);

// 数据库配置  
DataSourceConfig dsc = new DataSourceConfig();  
dsc.setDbType(DbType.MYSQL);  
 //自己数据库名 我的数据库名 xc\_content  
dsc.setUrl("jdbc:mysql://localhost/xc\_" + SERVICE\_NAME  
\+ "?serverTimezone=UTC&useUnicode=true&useSSL=false&characterEncoding=utf8");  
// dsc.setDriverName("com.mysql.jdbc.Driver");  
dsc.setDriverName("com.mysql.cj.jdbc.Driver");  
dsc.setUsername(DATA\_SOURCE\_USER\_NAME);  
dsc.setPassword(DATA\_SOURCE\_PASSWORD);  
mpg.setDataSource(dsc);

// 包配置  
PackageConfig pc = new PackageConfig();  
pc.setModuleName(SERVICE\_NAME);  
pc.setParent("com.xuecheng");

pc.setServiceImpl("service.impl");  
pc.setXml("mapper");  
pc.setEntity("model.po");  
mpg.setPackageInfo(pc);

// 设置模板  
TemplateConfig tc = new TemplateConfig();  
mpg.setTemplate(tc);

// 策略配置  
StrategyConfig strategy = new StrategyConfig();  
strategy.setNaming(NamingStrategy.underline\_to\_camel);  
strategy.setColumnNaming(NamingStrategy.underline\_to\_camel);  
strategy.setEntityLombokModel(true);  
strategy.setRestControllerStyle(true);  
strategy.setInclude(TABLE\_NAMES);  
strategy.setControllerMappingHyphenStyle(true);  
strategy.setTablePrefix(pc.getModuleName() + "\_");  
// Boolean类型字段是否移除is前缀处理  
strategy.setEntityBooleanColumnRemoveIsPrefix(true);  
strategy.setRestControllerStyle(true);

// 自动填充字段配置  
strategy.setTableFillList(Arrays.asList(  
new TableFill("create\_date", FieldFill.INSERT),  
new TableFill("change\_date", FieldFill.INSERT\_UPDATE),  
new TableFill("modify\_date", FieldFill.UPDATE)  
));  
mpg.setStrategy(strategy);

mpg.execute();  
\}

[a1d2dfffa44e4b029ca2e750341671de.png]: https://img-blog.csdnimg.cn/a1d2dfffa44e4b029ca2e750341671de.png