1. 想詳細(xì)了解MyBatis-Plus代碼生成器可以先去官網(wǎng)了解
image.png
2. 集成
先新建spring boot工程mp-generator ，并添加依賴(如果是聚合項(xiàng)目场靴，可以新建模塊集成)
<!-- spring web依賴 -->
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-web</artifactId>
</dependency>
<!-- mysql和mybatis-plus依賴 -->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.4.0</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.4.0</version>
</dependency>
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
<version>2.3.31</version>
</dependency>
<!-- 工具依賴 -->
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<optional>true</optional>
</dependency>
<dependency>
<groupId>io.springfox</groupId>
<artifactId>springfox-swagger2</artifactId>
<version>2.9.2</version>
</dependency>
<dependency>
<groupId>org.apache.commons</groupId>
<artifactId>commons-lang3</artifactId>
</dependency>
3. 添加配置
新建entity包怀伦，添加一個(gè)實(shí)體類的父類
package com.mp.generator.entity;
import com.baomidou.mybatisplus.annotation.*;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import java.io.Serializable;
import java.util.Date;
@Data
public abstract class BaseEntity implements Serializable {
private static final long serialVersionUID = 1L;
@ApiModelProperty(value = "主鍵")
@TableId(value = "id", type = IdType.AUTO)
private Long id;
@ApiModelProperty(value = "是否刪除瓮床，0 未刪除， 1 刪除")
@TableLogic
@TableField(value = "deleted", fill = FieldFill.INSERT)
private Integer deleted;
@ApiModelProperty(value = "是否可用难礼，1 可用挣郭，0 不可用")
@TableField(value = "available", fill = FieldFill.INSERT)
private Integer available;
@ApiModelProperty(value = "創(chuàng)建時(shí)間")
@TableField(value = "create_time", fill = FieldFill.INSERT)
private Date createTime;
@ApiModelProperty(value = "修改時(shí)間")
@TableField(value = "update_time", fill = FieldFill.INSERT_UPDATE)
private Date updateTime;
}
新建config包，在包下添加Config類蟀架，配置全局信息
package com.mp.generator.config;
public class Config {
public static String URL = "jdbc:mysql://127.0.0.1:3306/dynamic1?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&failOverReadOnly=false&zeroDateTimeBehavior=convertToNull&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true";
public static String USERNAME = "root";
public static String PASSWORD = "123456";
public static String DRIVER = "com.mysql.cj.jdbc.Driver";
/**
* 包名
*/
public static final String PACKAGE_PARENT = "com.mp.generator";
public static final String PACKAGE_NAME_CONTROLLER = "controller";
public static final String PACKAGE_NAME_MODEL = "entity";
public static final String PACKAGE_NAME_DAO = "mapper";
public static final String PACKAGE_NAME_SERVICE = "service";
public static final String PACKAGE_NAME_SERVICE_IMPL = "service.impl";
/**
* 文件名后綴：Dao
*/
public static final String FILE_NAME_DAO = "%sMapper";
/**
* 文件名后綴：MapperXml
*/
public static final String FILE_NAME_XML = "%sMapper";
/**
* MP開頭，Service結(jié)尾
*/
public static final String FILE_NAME_SERVICE = "%sService";
/**
* 文件名后綴：ServiceImpl
*/
public static final String FILE_NAME_SERVICE_IMPL = "%sServiceImpl";
/**
* 文件名后綴：Controller
*/
public static final String FILE_NAME_CONTROLLER = "%sController";
/**
* 邏輯刪除字段
*/
public static final String FIELD_LOGIC_DELETE_NAME = "deleted";
/**
* 作者
*/
public static final String AUTHOR = "generator";
/**
* 生成文件的輸出目錄
*/
// public static String projectPath = System.getProperty("user.dir") + "/mp-generator";
public static String projectPath = System.getProperty("user.dir");
/**
* 輸出目錄
*/
public static final String outputDir = projectPath + "/src/main/java";
/**
* 模板引擎榆骚。velocity / freemarker / beetl
*/
public static final String TEMPLATE_ENGINE = "freemarker";
/**
* 是否支持Swagger片拍，默認(rèn)不支持
*/
public static final Boolean SWAGGER_SUPPORT = true;
}
新建utils包，添加CommonUtils類妓肢，配置生成信息
package com.mp.generator.utils;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine;
import com.baomidou.mybatisplus.generator.engine.BeetlTemplateEngine;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;
import com.mp.generator.config.Config;
import com.mp.generator.entity.BaseEntity;
import org.apache.commons.lang3.StringUtils;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class CommonUtils {
//讀取控制臺(tái)內(nèi)容
public static String scanner(String tip) {
Scanner scanner = new Scanner(System.in);
System.out.println("請(qǐng)輸入" + tip + "：");
if (scanner.hasNext()) {
String ipt = scanner.next();
if (StringUtils.isNotEmpty(ipt)) {
return ipt;
}
}
throw new MybatisPlusException("請(qǐng)輸入正確的" + tip + "捌省！");
}
//數(shù)據(jù)連接信息
private static DataSourceConfig dataSourceConfig() {
return new DataSourceConfig()
.setDbType(DbType.MYSQL)
.setUrl(Config.URL)
.setUsername(Config.USERNAME)
.setPassword(Config.PASSWORD)
.setDriverName(Config.DRIVER)
;
}
// 配置
private static GlobalConfig globalConfig() {
return new GlobalConfig()
.setAuthor(scanner("作者"))
.setOutputDir(Config.outputDir)
.setFileOverride(true) // 是否覆蓋已有文件
.setOpen(false) // 是否打開輸出目錄
.setDateType(DateType.ONLY_DATE) // 時(shí)間采用java 8，（操作工具類：JavaLib => DateTimeUtils）
.setEnableCache(false)// XML 二級(jí)緩存
.setBaseResultMap(false)// XML ResultMap
.setBaseColumnList(false)// XML columList
.setKotlin(false) //是否生成 kotlin 代碼
.setMapperName(Config.FILE_NAME_DAO) //自定義文件命名碉钠，注意 %s 會(huì)自動(dòng)填充表實(shí)體屬性纲缓！
.setXmlName(Config.FILE_NAME_XML)
.setServiceName(Config.FILE_NAME_SERVICE)
.setServiceImplName(Config.FILE_NAME_SERVICE_IMPL)
.setControllerName(Config.FILE_NAME_CONTROLLER)
.setSwagger2(Config.SWAGGER_SUPPORT) // model swagger2
;
}
//實(shí)體類配置
private static StrategyConfig strategyConfig() {
return new StrategyConfig()
.setChainModel(true) // 【實(shí)體】是否為構(gòu)建者模型（默認(rèn) false）
.setNaming(NamingStrategy.underline_to_camel)
.setColumnNaming(NamingStrategy.underline_to_camel)
.setSuperEntityClass(BaseEntity.class)
.setEntityLombokModel(true)
.setRestControllerStyle(true)
.setInclude(scanner("表名，多個(gè)英文逗號(hào)分割").split(","))
.setControllerMappingHyphenStyle(false)
.setEntityTableFieldAnnotationEnable(true) //是否生成實(shí)體時(shí)喊废，生成字段注解,包括@TableName("")
;
}
// 包信息配置
private static PackageConfig packageConfig() {
return new PackageConfig()
// .setModuleName(scanner("模塊名"))
.setParent(Config.PACKAGE_PARENT)
// 如果是單體項(xiàng)目請(qǐng)注釋掉 .setModuleName(scanner("模塊名"))即可色徘，把下面注釋取消
.setController(Config.PACKAGE_NAME_CONTROLLER)
.setEntity(Config.PACKAGE_NAME_MODEL)
.setMapper(Config.PACKAGE_NAME_DAO)
.setService(Config.PACKAGE_NAME_SERVICE)
.setServiceImpl(Config.PACKAGE_NAME_SERVICE_IMPL)
;
}
private static InjectionConfig injectionConfig() {
InjectionConfig injectionConfig = new InjectionConfig() {
@Override
public void initMap() {
// to do nothing
}
};
String templatePath = "/templates/mapper.xml.ftl";
// 如果模板引擎是 velocity
// String templatePath = "/templates/mapper.xml.vm";
// 自定義輸出配置
List<FileOutConfig> focList = new ArrayList<>();
// 自定義配置會(huì)被優(yōu)先輸出
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定義輸出文件名 ， 如果你 Entity 設(shè)置了前后綴操禀、此處注意 xml 的名稱會(huì)跟著發(fā)生變化！横腿！
return Config.projectPath + "/src/main/resources/mapper/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
injectionConfig.setFileOutConfigList(focList);
return injectionConfig;
}
private static TemplateConfig templateConfig() {
TemplateConfig templateConfig = new TemplateConfig();
//配置自定義輸出模板颓屑，指定自定義模板路徑，注意不要帶上.ftl/.vm, 會(huì)根據(jù)使用的模板引擎自動(dòng)識(shí)別
templateConfig.setController("templates/controllerGenerator.java");
templateConfig.setServiceImpl("templates/serviceImplGenerator.java");
templateConfig.setMapper("templates/mapperGenerator.java");
templateConfig.setXml(null);
return templateConfig;
}
// 獲取模板引擎
private static AbstractTemplateEngine getTemplateEngine() {
String templateEngine = Config.TEMPLATE_ENGINE;
switch (templateEngine) {
case "velocity":
return new VelocityTemplateEngine();
case "freemarker":
return new FreemarkerTemplateEngine();
case "beetl":
return new BeetlTemplateEngine();
default:
return new VelocityTemplateEngine();
}
}
// 執(zhí)行器
public static void execute() {
GlobalConfig globalConfig = globalConfig();
DataSourceConfig dataSourceConfig = dataSourceConfig();
PackageConfig packageConfig = packageConfig();
StrategyConfig strategyConfig = strategyConfig();
InjectionConfig injectionConfig = injectionConfig();
AbstractTemplateEngine templateEngine = getTemplateEngine();
new AutoGenerator()
.setGlobalConfig(globalConfig)
.setDataSource(dataSourceConfig)
.setStrategy(strategyConfig)
.setPackageInfo(packageConfig)
.setTemplateEngine(templateEngine)
.setCfg(injectionConfig)
.setTemplate(templateConfig())
.execute();
}
}
修改自己想要的模板耿焊，在resource目錄下新建templates目錄揪惦，然后存放所有的模板
mapperGenerator.java.ftl
package ${package.Mapper};
import ${package.Entity}.${entity};
import ${superMapperClassPackage};
import org.springframework.stereotype.Repository;
/**
* <p>
* ${table.comment!} Mapper 接口
* </p>
*
* @author ${author}
* @since ${date}
*/
@Repository
<#if kotlin>
interface ${table.mapperName} : ${superMapperClass}<${entity}>
<#else>
public interface ${table.mapperName} extends ${superMapperClass}<${entity}> {
}
</#if>
serviceImplGenerator.java.ftl
package ${package.ServiceImpl};
import ${package.Entity}.${entity};
import ${package.Mapper}.${table.mapperName};
import ${package.Service}.${table.serviceName};
import ${superServiceImplClassPackage};
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
/**
* <p>
* ${table.comment!} 服務(wù)實(shí)現(xiàn)類
* </p>
*
* @author ${author}
* @since ${date}
*/
@Slf4j
@Service
<#if kotlin>
open class ${table.serviceImplName} : ${superServiceImplClass}<${table.mapperName}, ${entity}>(), ${table.serviceName} {
}
<#else>
public class ${table.serviceImplName} extends ${superServiceImplClass}<${table.mapperName}, ${entity}> implements ${table.serviceName} {
}
</#if>
controllerGenerator.java.ftl
package ${package.Controller};
import io.swagger.annotations.Api;
import org.springframework.web.bind.annotation.RequestMapping;
<#if restControllerStyle>
import org.springframework.web.bind.annotation.RestController;
<#else>
import org.springframework.stereotype.Controller;
</#if>
<#if superControllerClassPackage??>
import ${superControllerClassPackage};
</#if>
/**
* <p>
* ${table.comment!} 前端控制器
* </p>
*
* @author ${author}
* @since ${date}
*/
<#if restControllerStyle>
@RestController
<#else>
@Controller
</#if>
@RequestMapping("/api/mp-generator<#if package.ModuleName?? && package.ModuleName != "">/${package.ModuleName}</#if>/<#if controllerMappingHyphenStyle??>${controllerMappingHyphen}<#else>${table.entityPath}</#if>")
@Api(value = "/api/mp-generator<#if package.ModuleName?? && package.ModuleName != "">/${package.ModuleName}</#if>/<#if controllerMappingHyphenStyle??>${controllerMappingHyphen}<#else>${table.entityPath}</#if>", tags = {"${table.comment!}接口"})
<#if kotlin>
class ${table.controllerName}<#if superControllerClass??> : ${superControllerClass}()</#if>
<#else>
<#if superControllerClass??>
public class ${table.controllerName} extends ${superControllerClass} {
<#else>
public class ${table.controllerName} {
</#if>
}
</#if>
4. 添加一個(gè)啟動(dòng)類運(yùn)行
package com.mp.generator;
import com.mp.generator.utils.CommonUtils;
public class MySQLCodeGenerator {
public static void main(String[] args) {
CommonUtils.execute();
}
}
sql腳本
/*
Navicat Premium Data Transfer
Source Server : 本地
Source Server Type : MySQL
Source Server Version : 50729
Source Host : 127.0.0.1:3306
Source Schema : dynamic1
Target Server Type : MySQL
Target Server Version : 50729
File Encoding : 65001
Date: 18/03/2021 19:02:02
*/
SET NAMES utf8mb4;
SET FOREIGN_KEY_CHECKS = 0;
-- ----------------------------
-- Table structure for user_info
-- ----------------------------
DROP TABLE IF EXISTS `user_info`;
CREATE TABLE `user_info` (
`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '主鍵',
`available` int(1) NULL DEFAULT 1 COMMENT '是否可用，1 可用罗侯，0 不可用',
`create_time` datetime(0) NULL DEFAULT NULL COMMENT '創(chuàng)建時(shí)間',
`deleted` int(1) NULL DEFAULT 0 COMMENT '是否刪除器腋，0 未刪除， 1 刪除',
`update_time` datetime(0) NULL DEFAULT NULL COMMENT '修改時(shí)間',
`avatar` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '頭像',
`email` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '郵箱',
`password` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '密碼',
`phone` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '手機(jī)號(hào)',
`salt` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '鹽',
`sex` int(1) NULL DEFAULT NULL COMMENT '性別 0未知 1女 2男',
`user_name` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NOT NULL COMMENT '姓名',
PRIMARY KEY (`id`) USING BTREE,
UNIQUE INDEX `user_info_user_name`(`user_name`) USING BTREE
) ENGINE = InnoDB AUTO_INCREMENT = 31 CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '用戶信息表' ROW_FORMAT = Dynamic;
-- ----------------------------
-- Records of user_info
-- ----------------------------
INSERT INTO `user_info` VALUES (1, 1, '2021-01-13 13:30:32', 0, '2021-01-15 15:19:15', NULL, '1111@qq.com', '55555', '1234567890', NULL, 0, 'testd57fdf65-6dea-44ae-8493-59bfb93dfec611111');
INSERT INTO `user_info` VALUES (2, 1, '2021-01-13 13:30:32', 0, '2021-01-15 15:19:15', NULL, '1111@qq.com', '55555', '1234567890', NULL, 0, 'test3ef83e0c-d20d-498d-852d-052bab06199922222');
SET FOREIGN_KEY_CHECKS = 1;
