实战八：SpringBoot代码生成器

目录

1、依赖

2、代码生成类

3、controller.java.vm模版

---

1、依赖

        <!-- mybatis-plus自动生成文件依赖此包 -->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.5.2</version>
        </dependency>

模版引擎依赖

        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity-engine-core</artifactId>
            <version>2.3</version>
        </dependency>

2、代码生成类

package com.example.demo.utils;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;

import java.util.Collections;

public class CodeGenerator {
    public static void main(String[] args) {
        generator();

    }

    private static void generator() {
        FastAutoGenerator.create("jdbc:mysql://127.0.0.1:3306/test?serverTimezone=UTC&characterEncoding=utf-8&useSSL=false&allowPublicKeyRetrieval=true", "root", "admin123")
                .globalConfig(builder -> {
                    builder.author("cizhu") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("/Users/gejia/Desktop/demoPro/demo1/src/main/java"); // 指定输出目录
                })
//                .dataSourceConfig(builder -> builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
//                    int typeCode = metaInfo.getJdbcType().TYPE_CODE;
//                    if (typeCode == Types.SMALLINT) {
//                        // 自定义类型转换
//                        return DbColumnType.INTEGER;
//                    }
//                    return typeRegistry.getColumnType(metaInfo);
//
//                }))
                .packageConfig(builder -> {
                    builder.parent("com.example.demo") // 设置父包名
                            .moduleName(null) // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.xml, "/Users/gejia/Desktop/demoPro/demo1/src/main/resources/mapper")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
//                    builder.addInclude("t_simple") // 设置需要生成的表名
//                            .addTablePrefix("t_", "c_"); // 设置过滤表前
                    builder.entityBuilder().enableLombok(); // 启用lombok
                    builder.controllerBuilder().enableRestStyle(); // 启动RestFul

                    builder.addInclude("comment"); // 设置需要生成的表名

                })
//                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }
}

3、controller.java.vm模版

 

package ${package.Controller};

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.List;

import com.example.demo.common.Result;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.springframework.web.bind.annotation.PathVariable;
import ${package.Service}.${table.serviceName};
import ${package.Entity}.${entity};

#if(${restControllerStyle})
import org.springframework.web.bind.annotation.RestController;
#else
import org.springframework.stereotype.Controller;
#end
#if(${superControllerClassPackage})
import ${superControllerClassPackage};
#end

/**
 * <p>
 * $!{table.comment}
 * </p>
 *
 * @author ${author}
 * @since ${date}
 */
#if(${restControllerStyle})
@RestController
#else
@Controller
#end
@RequestMapping("#if(${package.ModuleName})/${package.ModuleName}#end/#if(${controllerMappingHyphenStyle})${controllerMappingHyphen}#else${table.entityPath}#end")
#if(${kotlin})
class ${table.controllerName}#if(${superControllerClass}) : ${superControllerClass}()#end

#else
    #if(${superControllerClass})
    public class ${table.controllerName} extends ${superControllerClass} {
    #else
    public class ${table.controllerName} {
    #end

@Resource
private ${table.serviceName} ${table.entityPath}Service;

// 新增或者更新
/***********************************
 * 用途说明:
 * @param ${table.entityPath}
 * 返回值说明:
 * @return Result
 ***********************************/
@PostMapping
public Result save(@RequestBody ${entity} ${table.entityPath}){
        return Result.success(${table.entityPath}Service.saveOrUpdate(${table.entityPath}));
        }

/***********************************
 * 用途说明:
 * @param id
 * 返回值说明:
 * @return Result
 ***********************************/
@DeleteMapping("/{id}")
public Result delete(@PathVariable Integer id){
        return Result.success(${table.entityPath}Service.removeById(id));
        }

/***********************************
 * 用途说明:
 * @param ids
 * 返回值说明:
 * @return Result
 ***********************************/
@PostMapping("/del/batch")
public Result deleteBatch(@RequestBody List<Integer> ids){
        return Result.success(${table.entityPath}Service.removeByIds(ids));
        }

/***********************************
 * 用途说明:
 * @param:
 * 返回值说明:
 * @return Result
 ***********************************/
@GetMapping
public Result findAll(){
        return Result.success(${table.entityPath}Service.list());
        }

/***********************************
 * 用途说明:
 * @param id
 * 返回值说明:
 * @return Result
 ***********************************/
@GetMapping("/{id}")
public Result findOne(@PathVariable Integer id){
        return Result.success(${table.entityPath}Service.getById(id));
        }

/***********************************
 * 用途说明:
 * @param pageNum pageSize ${table.entityPath}
 * 返回值说明:
 * @return Result
 ***********************************/
@GetMapping("/page")
public Result findPage(@RequestParam Integer pageNum,
@RequestParam Integer pageSize){
        QueryWrapper<${entity}> queryWrapper=new QueryWrapper<>();
        queryWrapper.orderByDesc("id");
        return Result.success(${table.entityPath}Service.page(new Page<>(pageNum,pageSize),queryWrapper));
        }
        }

#end