一、介绍
本教程将介绍如何使用 mybatis-plus 工具自动给我们生成 Controller、Service、Entity、Mapper、Mapper.xml 层代码
要求:
① 生成的Mapper接口，需要继承 BaseMapper
② 生成的Entity 类，需要继承 Model
③ 生成的 Service，默认名称下是以 I 开头的接口, 在生成Service层代码中需要把这个 I 去掉
二、实现步骤
① 在数据库中创建好 数据库 与 要生成代码对应的表
这里拿 sys_dict 表举例
/*==============================================================*/
/* DBMS name: MySQL 5.0 */
/* Created on: 2020/9/25 9:34:58 */
/*==============================================================*/
drop table if exists sys_dict;
/*==============================================================*/
/* Table: sys_dict */
/*==============================================================*/
create table sys_dict
(
id int not null auto_increment comment '主键',
name varchar(50) comment '名称',
parent_id int comment '父级ID',
sort int comment '排序',
type int comment '类型',
create_by int comment '创建人',
create_date datetime comment '创建时间',
primary key (id)
);
alter table sys_dict comment '数据字典';
② 创建 一个SpringBoot 项目，其中 pom.xml 内容如下:
4.0.0
org.springframework.boot
spring-boot-starter-parent
2.3.3.RELEASE
com.hytc.mall
hytcmall
0.0.1-SNAPSHOT
hytcmall
弘毅天承商城
1.8
org.springframework.boot
spring-boot-starter-web
com.baomidou
dynamic-datasource-spring-boot-starter
2.5.4
com.baomidou
mybatis-plus-boot-starter
3.1.2
com.baomidou
mybatis-plus
3.3.2
com.baomidou
mybatis-plus-generator
3.3.2
org.freemarker
freemarker
mysql
mysql-connector-java
8.0.19
org.springframework.boot
spring-boot-starter-data-redis
redis.clients
jedis
3.2.0
com.alibaba
druid-spring-boot-starter
1.1.22
com.alibaba
fastjson
1.2.58
com.auth0
java-jwt
3.10.2
org.apache.commons
commons-lang3
3.10
cn.hutool
hutool-all
5.3.0
com.xiaoleilu
hutool-all
3.3.0
org.projectlombok
lombok
1.18.8
provided
log4j
log4j
1.2.17
io.springfox
springfox-swagger2
2.9.2
io.swagger
swagger-annotations
1.5.21
io.swagger
swagger-models
1.5.21
com.github.xiaoymin
swagger-bootstrap-ui
1.9.6
org.apache.poi
poi
3.9
org.apache.poi
poi-ooxml
3.7
org.apache.poi
poi-ooxml-schemas
3.9
org.springframework.boot
spring-boot-starter-test
test
org.junit.vintage
junit-vintage-engine
org.springframework.boot
spring-boot-maven-plugin
③ 创建代码生成类
package com.hytc.mall.hytcmall.generator;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @ClassName: MysqlGenerator
* @Description:https://blog.csdn.net/u010559460/article/details/103945985/
* https://www.cnblogs.com/wyrlzy/p/12033912.html
* @Author: BYP <[email protected]>
* @Date: 2020/9/24 14:36
* @Copyright: 2019 www.tydic.com Inc. All rights reserved.
* 注意：本内容仅限于弘毅天承信息技术股份有限公司内部传阅，禁止外泄以及用于其他的商业目
*/
public class MysqlGenerator {
final static String dirPath = "E://HYTC//project//mall//font//src//main//java";
/**
*
* MySQL 生成演示
*
*/
public static void main(String[] args) {
AutoGenerator mpg = new AutoGenerator();
// 选择 freemarker 引擎，默认 Veloctiy
mpg.setTemplateEngine(new FreemarkerTemplateEngine());
// 全局配置
GlobalConfig gc = new GlobalConfig();
gc.setOutputDir(dirPath);
gc.setAuthor("baiYP");
gc.setFileOverride(true); //是否覆盖
gc.setActiveRecord(true);// 不需要ActiveRecord特性的请改为false
gc.setEnableCache(false);// XML 二级缓存
gc.setBaseResultMap(true);// XML ResultMap
gc.setBaseColumnList(true);// XML columList
// 自定义文件命名，注意 %s 会自动填充表实体属性！
// gc.setMapperName("%sDao");
// gc.setXmlName("%sMapper");
// gc.setServiceName("MP%sService");
// gc.setServiceImplName("%sServiceDiy");
// gc.setControllerName("%sAction");
mpg.setGlobalConfig(gc);
// 数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setDbType(DbType.MYSQL);
dsc.setTypeConvert(new MySqlTypeConvert(){
// 自定义数据库表字段类型转换【可选】
// public DbColumnType processTypeConvert(String fieldType) {
// System.out.println("转换类型：" + fieldType);
// // 注意！！processTypeConvert 存在默认类型转换，如果不是你要的效果请自定义返回、非如下直接返回。
// return super.processTypeConvert(fieldType);
// }
});
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
dsc.setUsername("root");
dsc.setPassword("root");
dsc.setUrl("jdbc:mysql://localhost:3306/hytcmall_20200923?serverTimezone=UTC&characterEncoding=utf8&useUnicode=true&useSSL=false");
mpg.setDataSource(dsc);
// 策略配置
StrategyConfig strategy = new StrategyConfig();
// strategy.setCapitalMode(true);// 全局大写命名 ORACLE 注意
//strategy.setTablePrefix(new String[] { "tb_", "sys_" });// 此处可以修改为您的表前缀
strategy.setNaming(NamingStrategy.underline_to_camel);// 表名生成策略
// strategy.setInclude(new String[] { "user" }); // 需要生成的表
strategy.setExclude(new String[]{"sys_dict","tb_goods"}); // 排除生成的表
// 自定义实体父类
// strategy.setSuperEntityClass("com.baomidou.demo.TestEntity");
// 自定义实体，公共字段
// strategy.setSuperEntityColumns(new String[] { "test_id", "age" });
// 自定义 mapper 父类
// strategy.setSuperMapperClass("com.baomidou.demo.TestMapper");
// 自定义 service 父类
// strategy.setSuperServiceClass("com.baomidou.demo.TestService");
// 自定义 service 实现类父类
// strategy.setSuperServiceImplClass("com.baomidou.demo.TestServiceImpl");
// 自定义 controller 父类
// strategy.setSuperControllerClass("com.baomidou.demo.TestController");
// 【实体】是否生成字段常量（默认 false）
// public static final String ID = "test_id";
// strategy.setEntityColumnConstant(true);
// 【实体】是否为构建者模型（默认 false）
// public User setName(String name) {this.name = name; return this;}
strategy.setEntityBuilderModel(true);
mpg.setStrategy(strategy);
// 包配置
PackageConfig pc = new PackageConfig();
pc.setParent("com.hytc.mall.hytcmall");
//pc.setModuleName("good");
pc.setController("controller");
pc.setEntity("entity");
pc.setMapper("mapper");
pc.setService("service");
pc.setServiceImpl("service.mpl");
pc.setXml("xml");
mpg.setPackageInfo(pc);
// 注入自定义配置，可以在 VM 中使用 cfg.abc 【可无】
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
Map map = new HashMap();
map.put("abc", this.getConfig().getGlobalConfig().getAuthor() + "-mp");
this.setMap(map);
}
};
// 自定义 xxList.jsp 生成
List focList = new ArrayList();
/* focList.add(new FileOutConfig("/template/list.jsp.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定义输入文件名称
return "D://my_" + tableInfo.getEntityName() + ".jsp";
}
});
cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg);*/
// 调整 xml 生成目录演示
/* focList.add(new FileOutConfig("/templates/mapper.xml.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
return dirPath + tableInfo.getEntityName() + "Mapper.xml";
}
});
cfg.setFileOutConfigList(focList);
*/
mpg.setCfg(cfg);
// 关闭默认 xml 生成，调整生成 至 根目录
/* TemplateConfig tc = new TemplateConfig();
tc.setXml(null);
mpg.setTemplate(tc);*/
// 自定义模板配置，可以 copy 源码 mybatis-plus/src/main/resources/templates 下面内容修改，
// 放置自己项目的 src/main/resources/templates 目录下, 默认名称一下可以不配置，也可以自定义模板名称
// TemplateConfig tc = new TemplateConfig();
// tc.setController("...");
// tc.setEntity("...");
// tc.setMapper("...");
// tc.setXml("...");
// tc.setService("...");
// tc.setServiceImpl("...");
// 如上任何一个模块如果设置 空 OR Null 将不生成该模块。
// mpg.setTemplate(tc);
// 执行生成
mpg.execute();
// 打印注入设置【可无】
System.err.println(mpg.getCfg().getMap().get("abc"));
}
}
三、项目生成后的存放地址以及代码如下
① 项目的包结构如下：
项目层级如图所示：
②controller代码如下：
package com.hytc.mall.hytcmall.controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.stereotype.Controller;
/**
*
* 数据字典 前端控制器
*
*
* @author baiYP
* @since 2020-09-24
*/
@Controller
@RequestMapping("sysDict")
public class SysDictController {
}
控制器如图所示
③service代码如下：
package com.hytc.mall.hytcmall.service;
import com.hytc.mall.hytcmall.entity.SysDict;
import com.baomidou.mybatisplus.extension.service.IService;
/**
*
* 数据字典 服务类
*
*
* @author baiYP
* @since 2020-09-24
*/
public interface ISysDictService extends IService {
}
业务类如图所示：
④serviceImpl实现类如下：
package com.hytc.mall.hytcmall.service.mpl;
import com.hytc.mall.hytcmall.entity.SysDict;
import com.hytc.mall.hytcmall.mapper.SysDictMapper;
import com.hytc.mall.hytcmall.service.ISysDictService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.stereotype.Service;
/**
*
* 数据字典 服务实现类
*
*
* @author baiYP
* @since 2020-09-24
*/
@Service
public class SysDictServiceImpl extends ServiceImpl implements ISysDictService {
}
业务类实现如图所示：
⑤entity实体如下：
package com.hytc.mall.hytcmall.entity;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.extension.activerecord.Model;
import com.baomidou.mybatisplus.annotation.TableId;
import java.time.LocalDateTime;
import java.io.Serializable;
/**
*
* 数据字典
*
*
* @author baiYP
* @since 2020-09-24
*/
public class SysDict extends Model {
private static final long serialVersionUID = 1L;
/**
* 主键
*/
@TableId(value = "id", type = IdType.AUTO)
private Integer id;
/**
* 名称
*/
private String name;
/**
* 父级ID
*/
private Integer parentId;
/**
* 排序
*/
private Integer sort;
/**
* 类型
*/
private Integer type;
/**
* 创建人
*/
private Integer createBy;
/**
* 创建时间
*/
private LocalDateTime createDate;
public Integer getId() {
return id;
}
public SysDict setId(Integer id) {
this.id = id;
return this;
}
public String getName() {
return name;
}
public SysDict setName(String name) {
this.name = name;
return this;
}
public Integer getParentId() {
return parentId;
}
public SysDict setParentId(Integer parentId) {
this.parentId = parentId;
return this;
}
public Integer getSort() {
return sort;
}
public SysDict setSort(Integer sort) {
this.sort = sort;
return this;
}
public Integer getType() {
return type;
}
public SysDict setType(Integer type) {
this.type = type;
return this;
}
public Integer getCreateBy() {
return createBy;
}
public SysDict setCreateBy(Integer createBy) {
this.createBy = createBy;
return this;
}
public LocalDateTime getCreateDate() {
return createDate;
}
public SysDict setCreateDate(LocalDateTime createDate) {
this.createDate = createDate;
return this;
}
@Override
protected Serializable pkVal() {
return this.id;
}
@Override
public String toString() {
return "SysDict{" +
"id=" + id +
", name=" + name +
", parentId=" + parentId +
", sort=" + sort +
", type=" + type +
", createBy=" + createBy +
", createDate=" + createDate +
"}";
}
}
实体如图所示：
⑥mapper接口如下：
package com.hytc.mall.hytcmall.mapper;
import com.hytc.mall.hytcmall.entity.SysDict;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
/**
*
* 数据字典 Mapper 接口
*
*
* @author baiYP
* @since 2020-09-24
*/
public interface SysDictMapper extends BaseMapper {
}
mapper接口如图所示：
⑦mapper.xml如下
id, name, parent_id, sort, type, create_by, create_date
mapper.xml如图所示：