005 MP代码生成器，并自定义controller模板

一、 添加依赖

        <!--代码生成器-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.5.1</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.apache.velocity/velocity -->
        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity</artifactId>
            <version>1.7</version>
        </dependency>

二、 创建CodeCenerator.java

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

import java.util.Collections;

/**
 * @CreateDate: 2023/12/15 0015 17:25
 * @Description: mp代码生成器
 */
public class CodeGenerator {

    public static void main(String[] args) {
        generate();
    }

    private static void generate() {
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/test?serverTimezone=GMT%2b8", "root", "123456")
                .globalConfig(builder -> {
                    builder.author("zzq")//设置作者
                            .enableSwagger()//开启swagger模式
                            .fileOverride()//覆盖已经生成的问阿金
                            .outputDir("D:\\java\\work\\mag_springboot01\\src\\main\\java");//指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.example.mag_springboot01")//设置父包名
                            .moduleName("")//设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "D:\\java\\work\\mag_springboot01\\src\\main\\resources\\mapper"));//设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.addInclude("sys_user")//设置需要生成的模块名
                            .addTablePrefix("sys_", "c_");//设置过滤表前缀
                })
//                .templateEngine(new FreemarkerTemplateEngine())// 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }
}

三、自定义controller模板

1. 找到controller模板的位置，并放到resources/templates中

2. 根据自己的要求，自定义controller模板内容

package ${package.Controller};

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import java.util.List;

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
            @Autowired
            private ${table.serviceName} ${table.entityPath}Service;

            //新增或者更新
            @PostMapping
            public boolean save(@RequestBody ${entity} ${table.entityPath}){
                return ${table.entityPath}Service.saveOrUpdate(${table.entityPath});
            }

            #*
            *  其中
            *   1. user == ${table.entityPath}
            *   2. userMapper == ${table.entityPath}Mapper
            *   3. userService == ${table.entityPath}Service
            *   4. User == ${entity}
            * *#
            //删除
            @DeleteMapping("/{id}")
            public boolean delete(@PathVariable Integer id){
                return ${table.entityPath}Service.removeById(id);
            }

            //查询全部
            @GetMapping
            public List<${entity}> findAll(){
                return ${table.entityPath}Service.list();
            }

            //根据id查询
            @GetMapping("/{id}")
            public ${entity} findOne(@PathVariable Integer id){
                return ${table.entityPath}Service.getById(id);
            }

            //分页查询
            @GetMapping("/page")
            public Page<${entity}> findPage(@RequestParam Integer pageNumber, @RequestParam Integer pageSize){
                return ${table.entityPath}Service.page(new Page<>(pageNumber, pageSize));
            }
}


#end

注意：

        1. 其中的标点符号一个都不能少，否则直接报错

 四、运行CodeGenerator

五、结果

 

package com.example.mag_springboot01.controller;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.example.mag_springboot01.entity.User;
import com.example.mag_springboot01.service.IUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;

import java.util.List;

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author zzq
 * @since 2023-12-15
 */
@Controller
@RequestMapping("/user")
public class UserController {
            @Autowired
            private IUserService userService;
            //新增或者更新
            @PostMapping
            public boolean save(@RequestBody User user){
                return userService.saveOrUpdate(user);
            }
            //删除
            @DeleteMapping("/{id}")
            public boolean delete(@PathVariable Integer id){
                return userService.removeById(id);
            }
            //查询全部
            @GetMapping
            public List<User> findAll(){
                return userService.list();
            }
            //根据id查询
            @GetMapping("/{id}")
            public User findOne(@PathVariable Integer id){
                return userService.getById(id);
            }
            //分页查询
            @GetMapping("/page")
            public Page<User> findPage(@RequestParam Integer pageNumber, @RequestParam Integer pageSize){
                return userService.page(new Page<>(pageNumber, pageSize));
        }
    }

其它的如entity、mapper、service等也可以自己定义