1 基于mybatis Generator
1.1 依赖
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.5.1</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.5.1</version>
</dependency>
<!--模板引擎-->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.1</version>
</dependency>
1.2 启动类添加扫描
添加MapperScan 注解
@SpringBootApplication
@MapperScan("com.jxn.mapper")
public class JxnTymhZsbgwApplication {
public static void main(String[] args) {
SpringApplication.run(JxnTymhZsbgwApplication.class, args);
}
}
1.3 编写代码生成配置
public static void main(String[] args) {
// 全局配置
DataSourceConfig dataSourceConfig = new DataSourceConfig.
Builder("jdbc:mysql://localhost:3306/mybatis?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=UTF-8&useSSL=false&allowMultiQueries=true",
"root",
"123456")
.build();
AutoGenerator mpg = new AutoGenerator(dataSourceConfig);
String projectPath = System.getProperty("user.dir");
GlobalConfig globalConfig = new GlobalConfig.Builder()
// .fileOverride() //第二次生成覆盖第一次
.outputDir(projectPath + "/src/main/java")
.author("YouYT") // 作者
// .enableKotlin()
.enableSwagger() // 开启 swagger 模式
.dateType(DateType.TIME_PACK) // 时间策略
.commentDate("yyyy-MM-dd")
.build();
mpg.global(globalConfig);
PackageConfig packageConfig = new PackageConfig.Builder()
.parent("com.jxn")
.entity("pojo")
.service("service")
.serviceImpl("service.impl")
.mapper("mapper")
.xml("mapper.xml")
.controller("controller")
.pathInfo(Collections.singletonMap(OutputFile.mapperXml, "D://"))
.build();
mpg.packageInfo(packageConfig);
// InjectionConfig injectionConfig = new InjectionConfig.Builder()
// .beforeOutputFile((tableInfo, objectMap) -> {
// System.out.println("tableInfo: " + tableInfo.getEntityName() + " objectMap: " + objectMap.size());
// })
// .customMap(Collections.singletonMap(projectPath + "/jxn-tymh-zsbgw/src/main/resources/mapper/", "Mapper"))
// .build();
// mpg.injection(injectionConfig);
StrategyConfig strategyConfig = new StrategyConfig.Builder()
.enableCapitalMode()
.enableSkipView()
.disableSqlFilter()
.likeTable(new LikeTable("USER"))
.addInclude("t_admin_user")
.addTablePrefix("t_", "c_")
.build();
strategyConfig
.entityBuilder()
.disableSerialVersionUID()
.enableLombok()
.enableTableFieldAnnotation()
.enableActiveRecord()
.versionColumnName("version")
.naming(NamingStrategy.underline_to_camel)
.columnNaming(NamingStrategy.underline_to_camel)
.addTableFills(new Column("create_date", FieldFill.INSERT))
.addTableFills(new Property("update_date", FieldFill.INSERT_UPDATE))
.idType(IdType.AUTO)
.formatFileName("%s")
.build();
strategyConfig
.controllerBuilder()
// .enableHyphenStyle()
.enableRestStyle()
.build();
strategyConfig
.serviceBuilder()
.superServiceClass(IService.class)
.superServiceImplClass(ServiceImpl.class)
.formatServiceFileName("%sService")
.formatServiceImplFileName("%sServiceImp")
.build();
strategyConfig
.mapperBuilder()
.superClass(BaseMapper.class)
.enableMapperAnnotation()
.enableBaseResultMap()
.enableBaseColumnList()
.formatMapperFileName("%sMapper")
.formatXmlFileName("%sMapper")
.build();
mpg.strategy(strategyConfig);
mpg.execute();
}
2 插件
2.1 安装 Easy Code插件
2.2 setting修改生成策略
2.2.1 controller
##导入宏定义
$!{define.vm}
##设置表后缀(宏定义)
#setTableSuffix("Controller")
##保存文件(宏定义)
#save("/controller", "Controller.java")
##包路径(宏定义)
#setPackageSuffix("controller")
##定义服务名
#set($serviceName = $!tool.append($!tool.firstLowerCase($!tableInfo.name), "Service"))
##定义实体对象名
#set($entityName = $!tool.firstLowerCase($!tableInfo.name))
import $!{tableInfo.savePackageName}.pojo.$!tableInfo.name;
import $!{tableInfo.savePackageName}.service.$!{tableInfo.name}Service;
import org.springframework.web.bind.annotation.*;
import org.springframework.util.StringUtils;
import javax.annotation.Resource;
import java.io.Serializable;
##表注释(宏定义)
#tableComment("表控制层")
@RestController
@RequestMapping("$!tool.firstLowerCase($!tableInfo.name)")
@Api(tags = "$!{tableInfo.comment}")
public class $!{tableName} {
/**
* 服务对象
*/
@Resource
private $!{tableInfo.name}Service $!{serviceName};
/**
* 新增数据
*
* @param $!entityName 实体对象
* @return 新增结果
*/
@PostMapping
public Result $!{tableInfo.name}Insert(@RequestBody $!tableInfo.name $!entityName) {
try {
$!{serviceName}.save($!entityName);
return Result.ok();
} catch (Exception e) {
e.printStackTrace();
return Result.fail();
}
}
/**
* 修改数据
*
* @param $!entityName 实体对象
* @return 修改结果
*/
@PutMapping
public Result $!{tableInfo.name}Update(@RequestBody $!tableInfo.name $!entityName) {
try {
$!{serviceName}.updateById($!entityName);
return Result.ok();
} catch (Exception e) {
e.printStackTrace();
return Result.fail();
}
}
/**
* 删除数据
*
* @param id 主键
* @return 删除结果
*/
@DeleteMapping("{id}")
public Result $!{tableInfo.name}Delete(@PathVariable Serializable id) {
try {
if (StringUtils.isEmpty(id)) {
return Result.fail("参数异常");
}
$!{serviceName}.removeById(id);
return Result.ok();
} catch (Exception e) {
e.printStackTrace();
return Result.fail();
}
}
}
2.2.2 service
##导入宏定义
$!{define.vm}
##设置表后缀(宏定义)
#setTableSuffix("Service")
##保存文件(宏定义)
#save("/service", "Service.java")
##包路径(宏定义)
#setPackageSuffix("service")
import com.baomidou.mybatisplus.extension.service.IService;
import $!{tableInfo.savePackageName}.pojo.$!tableInfo.name;
##表注释(宏定义)
#tableComment("表服务接口")
public interface $!{tableName} extends IService<$!tableInfo.name> {
}
2.2.3 serviceImpl
##导入宏定义
$!{define.vm}
##设置表后缀(宏定义)
#setTableSuffix("ServiceImpl")
##保存文件(宏定义)
#save("/service/impl", "ServiceImpl.java")
##包路径(宏定义)
#setPackageSuffix("service.impl")
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import $!{tableInfo.savePackageName}.mapper.$!{tableInfo.name}Mapper;
import $!{tableInfo.savePackageName}.pojo.$!{tableInfo.name};
import $!{tableInfo.savePackageName}.service.$!{tableInfo.name}Service;
import org.springframework.stereotype.Service;
##表注释(宏定义)
#tableComment("表服务实现类")
@Service("$!tool.firstLowerCase($tableInfo.name)Service")
public class $!{tableName} extends ServiceImpl<$!{tableInfo.name}Mapper, $!{tableInfo.name}> implements $!{tableInfo.name}Service {
}
2.2.4 实体类
##导入宏定义
$!{define.vm}
##保存文件(宏定义)
#save("/pojo", ".java")
##包路径(宏定义)
#setPackageSuffix("pojo")
##自动导入包(全局变量)
$!{autoImport.vm}
import com.baomidou.mybatisplus.extension.activerecord.Model;
import java.io.Serializable;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
@ApiModel(value = "$!{tableInfo.comment}")
@TableName("$!{tableInfo.name}")
@Data
public class $!{tableInfo.name} extends Model<$!{tableInfo.name}> {
#foreach($column in $tableInfo.fullColumn)
#if(${column.comment})#end
@ApiModelProperty(value="$!{column.comment}")
#if(${column.name}=='id')
@TableId(value = "id", type = IdType.AUTO)
#end
#if(${column.name}=='createDate')
@TableField(fill = FieldFill.INSERT)
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", locale = "zh", timezone = "GMT+8")
#end
#if(${column.name}=='updateDate')
@TableField(fill = FieldFill.INSERT_UPDATE)
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", locale = "zh", timezone = "GMT+8")
#end
private $!{tool.getClsNameByFullName($column.type)} $!{column.name};
#end
}
2.2.5 mapper
##导入宏定义
$!{define.vm}
##设置表后缀(宏定义)
#setTableSuffix("Mapper")
##保存文件(宏定义)
#save("/mapper", "Mapper.java")
##包路径(宏定义)
#setPackageSuffix("mapper")
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import $!{tableInfo.savePackageName}.pojo.$!tableInfo.name;
import org.apache.ibatis.annotations.Mapper;
##表注释(宏定义)
#tableComment("表数据库访问层")
@Mapper
public interface $!{tableName} extends BaseMapper<$!tableInfo.name> {
}
2.2.6 xml
##引入mybatis支持
$!{mybatisSupport.vm}
##设置保存名称与保存位置
$!callback.setFileName($tool.append($!{tableInfo.name}, "Mapper.xml"))
$!callback.setSavePath($tool.append($modulePath, "/src/main/resources/mapper"))
##拿到主键
#if(!$tableInfo.pkColumn.isEmpty())
#set($pk = $tableInfo.pkColumn.get(0))
#end
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="$!{tableInfo.savePackageName}.mapper.$!{tableInfo.name}Mapper">
</mapper>