SpringBoot集成OpenAPI(Swagger3)和Mybatis-plus代码生成器

package com.cloud.codetool.config.codetool.controller; import java.io.File; import java.util.Arrays; import java.util.Collections; import java.util.List; import org.springframework.beans.factory.annotation.Value; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RestController; import com.baomidou.mybatisplus.annotation.FieldFill; import com.baomidou.mybatisplus.generator.FastAutoGenerator; import com.baomidou.mybatisplus.generator.config.InjectionConfig; import com.baomidou.mybatisplus.generator.config.TemplateConfig; import com.baomidou.mybatisplus.generator.config.rules.DateType; import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine; import com.baomidou.mybatisplus.generator.fill.Column; import com.cloud.codetool.config.codetool.vo.RestData; import com.cloud.codetool.config.codetool.vo.TemplateParam; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import io.swagger.v3.oas.annotations.parameters.RequestBody; @RestController @Api(tags = "代码生成工具") public class CodeTempController { @Value(value = "${spring.datasource.druid.url}") private String url; @Value(value = "${spring.datasource.druid.username}") private String userName; @Value(value = "${spring.datasource.druid.password}") private String passWord; @Value(value = "${code.tool.package}") private String packagePath; @ApiOperation(value = "按照系统数据库表生成Controler、Service、Mapper、Entity") @GetMapping("/code/generate") public RestData<?> codeTemplate(@RequestBody TemplateParam templateParam ){ String projectPath = new File("").getAbsolutePath()+"//src//main//java//"; new TemplateConfig.Builder()/*.serviceImpl("/templates/serviceImpl.java")*/; FastAutoGenerator.create(url,userName,passWord) //全局配置 .globalConfig((scanner, builder) -> builder .author(templateParam.getAuthor()) .enableSwagger() //开启覆盖已生成的文件。注释掉则关闭覆盖。 //.fileOverride() //禁止打开生成的文件目录。 .disableOpenDir() //设置时间类型 Date（默认使用Java8的时间类型，部分数据库可能无法识别） .dateType(DateType.ONLY_DATE) //生成文件的输出路径 .outputDir(projectPath)) //包设置 .packageConfig((scanner, builder) -> builder //设置代码包路径 .parent(packagePath+"."+templateParam.getParentPackage())) //策略配置 .strategyConfig((scanner, builder) -> builder //需要使用表生成的代码（传入all时，所有表生成） .addInclude(getTables(templateParam.getTableName())) .entityBuilder() //逻辑删除配置，当字段名称为deleted，会设置逻辑删除。会在实体类的该字段属性前加注解[@TableLogic]（这里已经在/resources/templates/entity.java.ftl中配置了删除） //.logicDeleteColumnName("DELETE_FLAG") //添加字段描述 .enableLombok() .addTableFills( //设定创建时间 ，更新时间自动填充 new Column("create_time", FieldFill.INSERT), new Column("update_time", FieldFill.INSERT_UPDATE) ) .enableTableFieldAnnotation() //Mapper.xml策略配置 .mapperBuilder() //启用 BaseResultMap 生成。会在mapper.xml文件生成[通用查询映射结果]配置。 .enableBaseResultMap() //启用 BaseColumnList。会在mapper.xml文件生成[通用查询结果列 ]配置 .enableBaseColumnList() .build()) //模板引擎配置，Freemarker .templateEngine(new FreemarkerTemplateEngine()) .execute(); new InjectionConfig .Builder() .beforeOutputFile((tableInfo, objectMap) -> {}) .build(); return RestData.success("代码已生成，您可以编译源代码，重启项目即可看到生成的默认接口！"); } /** * @description:处理all的时候 * @author:hutao * @mail:hutao_2017@aliyun.com * @date:2022年2月18日 上午11:32:54 */ protected static List<String> getTables(String tables) { return "all".equals(tables) ? Collections.emptyList() : Arrays.asList(tables.split(",")); } }