FreeMarker代码生成器和Mybatis-plus代码生成器之二

2. Mybatis-plus代码生成器

---

注意！
使用版本：mybatis-plus-generator3.5.1及其以上版本

1. 先在pom文件中加入依赖：

<!--    代码生成器-->  
        <dependency>  
            <groupId>com.baomidou</groupId>  
            <artifactId>mybatis-plus-generator</artifactId>  
            <version>3.5.1</version>  
        </dependency>
<!--        模板引擎-->  
        <dependency>  
            <groupId>org.apache.velocity</groupId>  
            <artifactId>velocity</artifactId>  
            <version>1.7</version>  
        </dependency>

2. 然后编写代码生成器的类，一般在utils文件夹下，名为CodeGenerator
   在这一行代码中builder.addInclude(“user”)，是选定某个表进行生成代码，若改为builder.addExclude(“user”)，则是对除了user的所有表进行生成代码。

package com.example.demo1.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) {  
        generate();  
    }  
  
    private static void generate() {  
        FastAutoGenerator.create("jdbc:mysql:XXX", "UserName", "PassWord")  
                .globalConfig(builder -> {  
                    builder.author("小黄同学") // 设置作者  
                            .enableSwagger() // 开启 swagger 模式  
                            .fileOverride() // 覆盖已生成文件  
                            .outputDir(""); // 指定输出目录  
                })  
                .packageConfig(builder -> {  
                    builder.parent("com.example.demo1") // 设置父包名  
                            .moduleName(null) // 设置父包模块名  
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "XXX")); // 设置mapperXml生成路径  
                })  
                .strategyConfig(builder -> {  
                    builder.entityBuilder().enableLombok();  
//                    builder.mapperBuilder().enableMapperAnnotation().build();  
                    builder.controllerBuilder().enableHyphenStyle()  // 开启驼峰转连字符  
                            .enableRestStyle();  // 开启生成@RestController 控制器  
                    builder.addInclude("user") // 设置需要生成的表名  
                            .addTablePrefix("",""); // 设置过滤表前缀  
                })  
//                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板  
                .execute();  
  
    }  
}

3. 代码生成需要模板，有官方的模板，添加 模板引擎 依赖，MyBatis-Plus 支持 Velocity（默认）、Freemarker、Beetl，用户可以选择自己熟悉的模板引擎，如果都不满足要求，可以采用自定义模板引擎。
   自定义模板在resource文件夹下的templates文件夹下，一些模板例子如下所示：

    service的模板展示如下：
   

package ${package.Service};  
  
import ${package.Entity}.${entity};  
import ${superServiceClassPackage};  
  
/**  
 * <p>  
 * $!{table.comment} 服务类  
 * </p>  
 *  
 * @author ${author}  
 * @since ${date}  
 */  
#if(${kotlin})  
interface ${table.serviceName} : ${superServiceClass}<${entity}>  
#else  
public interface ${table.serviceName} extends ${superServiceClass}<${entity}> {  
  
}  
#end

	serviceImpl的模板如下：

package ${package.ServiceImpl};  
  
import ${package.Entity}.${entity};  
import ${package.Mapper}.${table.mapperName};  
import ${package.Service}.${table.serviceName};  
import ${superServiceImplClassPackage};  
import org.springframework.stereotype.Service;  
  
/**  
 * <p>  
 * $!{table.comment} 服务实现类  
 * </p>  
 *  
 * @author ${author}  
 * @since ${date}  
 */  
@Service  
#if(${kotlin})  
open class ${table.serviceImplName} : ${superServiceImplClass}<${table.mapperName}, ${entity}>(), ${table.serviceName} {  
  
}  
#else  
public class ${table.serviceImplName} extends ${superServiceImplClass}<${table.mapperName}, ${entity}> implements ${table.serviceName} {  
  
}  
#end

详情请见Mybatis-Plus代码生成器