一、实现效果
- 不生成service接口只生成service类
- 生成文件夹名称为controller、service、dao(mapper.java)、mapper(mapper.xml)
- 代码中的部分配置通过mybatisPlusGeneartor.properties文件
- 整合swagger
二、代码、配置文件、模板说明
- CodeGenerator.java 代码生成启动类
- classpath/resources/mybatisPlusGneratorConfig.properties 配置文件
- classpath/templates 代码生成模板
三、代码、配置文件、模板内容
package com.seentao.exam;
import cn.hutool.setting.dialect.Props;
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.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.seentao.framework.base.BaseServiceImpl;
import org.apache.commons.lang3.StringUtils;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class CodeGenerator {
public static String scanner(String tip) {
Scanner scanner = new Scanner(System.in);
StringBuilder help = new StringBuilder();
help.append("请输入" + tip + ":");
System.out.println(help.toString());
if (scanner.hasNext()) {
String ipt = scanner.next();
if (StringUtils.isNotEmpty(ipt)) {
return ipt;
}
}
throw new MybatisPlusException("请输入正确的" + tip + "!");
}
public static void main(String[] args) {
Props props= new Props("generator/mybatisPlusGneratorConfig.properties");
AutoGenerator mpg = new AutoGenerator();
GlobalConfig gc = new GlobalConfig();
String projectPath = System.getProperty("user.dir");
gc.setOutputDir(projectPath + "/src/main/java");
gc.setAuthor(props.getStr("author"));
gc.setSwagger2(true);
gc.setOpen(false);
mpg.setGlobalConfig(gc);
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl(props.getStr("jdbc.url"));
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
dsc.setUsername(props.getStr("jdbc.username"));
dsc.setPassword(props.getStr("jdbc.password"));
mpg.setDataSource(dsc);
PackageConfig pc = new PackageConfig();
pc.setModuleName(scanner("模块名"));
pc.setParent(props.getStr("parent.package"));
mpg.setPackageInfo(pc);
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
}
};
String templatePath = "/templates/mapper.xml.ftl";
List<FileOutConfig> focList = new ArrayList<>();
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
return projectPath + "/src/main/java/com/aaa/bbb/" + pc.getModuleName()
+ "/mapper/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
templatePath = "/templates/mapper.java.ftl";
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
String expand = projectPath + "/src/main/java/com/aaa/bbb/" +pc.getModuleName() + "/" + "dao";
String entityFile = String.format((expand + File.separator + "%s" + ".java"), tableInfo.getMapperName());
return entityFile;
}
});
templatePath = "/templates/controller.java.ftl";
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
String expand = projectPath + "/src/main/java/com/aaa/bbb/" +pc.getModuleName() + "/" + "controller";
String entityFile = String.format((expand + File.separator + "%s" + ".java"), tableInfo.getControllerName());
return entityFile;
}
});
templatePath = "/templates/serviceImpl.java.ftl";
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
String expand = projectPath + "/src/main/java/com/aaa/bbb/" +pc.getModuleName() + "/" + "service";
String entityFile = String.format((expand + File.separator + "%s" + ".java"), tableInfo.getServiceImplName().substring(0,tableInfo.getServiceImplName().length()-4));
return entityFile;
}
});
cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg);
TemplateConfig templateConfig = new TemplateConfig();
templateConfig.setXml(null);
templateConfig.setMapper(null);
templateConfig.setService(null);
templateConfig.setServiceImpl(null);
mpg.setTemplate(templateConfig);
StrategyConfig strategy = new StrategyConfig();
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
strategy.setEntityLombokModel(true);
strategy.setRestControllerStyle(true);
strategy.setSuperEntityClass("com.aaa.bbb.base.BaseEntity");
strategy.setSuperControllerClass("com.aaa.bbb.base.BaseController");
strategy.setSuperMapperClass("com.aaa.bbb.base.FrameworkBaseMapper");
strategy.setSuperServiceImplClass(BaseServiceImpl.class);
strategy.setInclude(scanner("表名,多个英文逗号分割").split(","));
strategy.setControllerMappingHyphenStyle(true);
strategy.setTablePrefix(pc.getModuleName() + "_");
mpg.setStrategy(strategy);
mpg.setTemplateEngine(new FreemarkerTemplateEngine());
mpg.execute();
}
}
- mybatisPlusGneratorConfig.properties
parent.package = com.aaa.bbb
jdbc.url=jdbc:mysql://xxx:3306/db?serverTimezone=UTC&useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true
jdbc.username=root
jdbc.password=111111
author = xiaojie
package ${package.Controller};
import org.springframework.web.bind.annotation.*;
import org.springframework.beans.factory.annotation.Autowired;
import ${package.Service}.${table.serviceImplName?substring(0,(table.serviceImplName)?length-4)};
import ${package.Entity}.${entity};
import com.seentao.framework.result.Result;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiImplicitParam;
import javax.validation.constraints.NotBlank;
import org.springframework.validation.annotation.Validated;
import lombok.extern.slf4j.Slf4j;
<
import org.springframework.web.bind.annotation.RestController;
<
import org.springframework.stereotype.Controller;
</
<
import ${superControllerClassPackage};
</
/**
*
*
* @author ${author}
* @since ${date}
* @version v1.0
*/
<
@Slf4j
@Api(tags = {"${table.comment!}"})
@RestController
@Validated
<
@Controller
</
@RequestMapping("<
<
class ${table.controllerName}<
<
<
public class ${table.controllerName} extends ${superControllerClass} {
<
public class ${table.controllerName} {
</
@Autowired
private ${table.serviceImplName?substring(0,(table.serviceImplName)?length-4)} ${table.serviceImplName?substring(0,(table.serviceImplName)?length-4)};
@RequestMapping(value = "/getById")
@ApiOperation(value = "注释", notes = "详细说明")
@ApiImplicitParam(name = "id", value = "主键id", dataType="String",required = true)
public Result<${entity}> getById(@NotBlank(message="id不能为空") String id){
return null;
}
}
</
package ${package.ServiceImpl?substring(0,package.ServiceImpl?length-5)};
import ${package.Entity}.${entity};
import ${package.Mapper?substring(0,package.Mapper?length-7)+".dao"}.${table.mapperName};
import ${superServiceImplClassPackage};
import org.springframework.stereotype.Service;
/**
* <p>
* ${table.comment!} 服务实现类
* </p>
*
* @author ${author}
* @since ${date}
*/
@Service
<
open class ${table.serviceImplName?substring(0,(table.serviceImplName)?length-4)} : ${superServiceImplClass}<${table.mapperName}, ${entity}>(){
}
<
public class ${table.serviceImplName?substring(0,(table.serviceImplName)?length-4)} extends ${superServiceImplClass}<${table.mapperName}, ${entity}>{
}
</
package ${package.Mapper?substring(0,package.Mapper?length-7)+".dao"};
import ${package.Entity}.${entity};
import ${superMapperClassPackage};
/**
* <p>
* ${table.comment!} Mapper 接口
* </p>
*
* @author ${author}
* @since ${date}
*/
<
interface ${table.mapperName} : ${superMapperClass}<${entity}>
<
public interface ${table.mapperName} extends ${superMapperClass}<${entity}> {
}
</
<?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="${package.Mapper?substring(0,package.Mapper?length-7)+".dao"}.${table.mapperName}">
<!-- 通用查询映射结果 -->
<resultMap id="BaseResultMap" type="${package.Entity}.${entity}">
<#list table.fields as field>
<#if field.keyFlag><#--生成主键排在第一位-->
<id column="${field.name}" property="${field.propertyName}" jdbcType="${field.type}"/>
</#if>
</#list>
<#list table.commonFields as field><#--生成公共字段 -->
<result column="${field.name}" property="${field.propertyName}" jdbcType="${field.type}"/>
</#list>
<#list table.fields as field>
<#if !field.keyFlag><#--生成普通字段 -->
<result column="${field.name}" property="${field.propertyName}" jdbcType="${field.type}"/>
</#if>
</#list>
</resultMap>
<!-- 通用查询结果列 -->
<sql id="Base_Column_List">
<#list table.commonFields as field>
${field.name},
</#list>
${table.fieldNames}
</sql>
</mapper>