基于mybatis-plus的代码自动生成工具（自定义模板）

1. MyBatis-Plus是一个MyBatis框架的增强工具，在MyBatis的基础上只做增强不做改变，为简化开发、提高效率而生。对于mybatis-plus不了解的同学，可以去MyBatis-Plus官网看看。真的能让开发效率显著提高，本文就来和大家一起分享下它的代码自动生成工具。

2. 其实在MyBatis-Plus的官网就为我们提供了代码生成器的使用教程，它可以为完成最基本的代码生成，但是想要实现自己的效果，还需要我们自己去设置生成模板，话不多说，直接开干。

3. 效果展示
   

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableField;
import java.io.Serializable;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.*;

/**
 * <p>
 * 用户信息表
 * </p>
 *
 * @author 
 * @since 2019-11-27
 */
@Data
@EqualsAndHashCode(callSuper = false)
@Builder
@AllArgsConstructor
@NoArgsConstructor
@ApiModel(value="User对象", description="用户信息表")
public class User implements Serializable {
   

    private static final long serialVersionUID = 1L;

    public static final String ID = "id";

    public static final String USERNAME = "username";

    public static final String PASSWORD = "password";


    /**
     * 主键
     */
    @ApiModelProperty(value = "主键")
    @TableId(value = "id", type = IdType.AUTO)
    private Integer id;

    /**
     * 用户名
     */
    @ApiModelProperty(value = "用户名")
    @TableField("username")
    private String username;

    /**
     * 密码
     */
    @ApiModelProperty(value = "密码")
    @TableField("password")
    private String password;

}

5. 添加项目依赖（添加swagger和lomback是因为在生成实体类中需要用到）

   <!--mybatis plus 代码自动生成工具-->
   <dependency>
       <groupId>com.baomidou</groupId>
       <artifactId>mybatis-plus-generator</artifactId>
       <version>3.2.0</version>
   </dependency>
   <!--freemarker模板引擎-->
   <dependency>
       <groupId>org.freemarker</groupId>
       <artifactId>freemarker</artifactId>
       <version>2.3.28</version>
   </dependency>
   <!--lomback-->
   <dependency>
       <groupId>org.projectlombok</groupId>
       <artifactId>lombok</artifactId>
       <optional>true</optional>
   </dependency>
    <!--mysql驱动-->
    <dependency>
       <groupId>mysql</groupId>
       <artifactId>mysql-connector-java</artifactId>
       <scope>runtime</scope>
   </dependency>
   <!-- Swagger2 Begin -->
   <dependency>
       <groupId>io.springfox</groupId>
       <artifactId>springfox-swagger2</artifactId>
       <version>2.9.2</version>
   </dependency>
   <dependency>
       <groupId>io.springfox</groupId>
       <artifactId>springfox-swagger-ui</artifactId>
       <version>2.9.2</version>
   </dependency>
   <!-- Swagger2 End -->
   

6. 编写代码生成器代码

package cn.mybatis.plus.test;

import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
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 java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * @Auther: PWJ
 * @Date: 2019/7/2 14:17
 * @Description:
 */
// 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
public class CodeGenerator {
   

    /**
     * <p>
     * 读取控制台内容
     * </p>
     */
    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("请输入正确的" &