MyBatis-Plus可视化代码生成器来了（真香）

在相关的页面中，可以进行相应的调整，在对应的 btl 模板中下载对应文件的具体模板，使用文本工具打开，直接修改源代码，文中取一种方式示例，其它方式一样。以上就是今天要讲的内容，本文简要介绍一款基于 Mybatis-Plus 的代码自助生成器，地址:文章通过实例集成的方式来详细讲解 mybatis-plus-generator-ui，从相关概念到实际集成案例，以及具体的扩展开发介绍。如果在工作中有这

你不懂、、、

1599人浏览 · 2024-03-16 07:56:01

你不懂、、、 · 2024-03-16 07:56:01 发布

来源|blog.csdn.net/yelangkingwuzuhu/article/details/128077533

前言

在基于 Mybatis 的开发模式中，很多开发者还会选择 Mybatis-Plus 来辅助功能开发，以此提高开发的效率。

虽然 Mybatis 也有代码生成的工具，但 Mybatis-Plus 由于在 Mybatis 基础上做了一些调整，因此，常规的生成工具生成的代码还有一些不太符合预期。而且对于多数据库的支持不是很好。

因此，我们需要一款支持高度定制化，带图形 UI 页面，能适配多数数据库的基础程序生成框架。本文就介绍这款基于 Mybatis-Plus 的代码自助生成器：mybatis-plus-generator-ui。

文章通过实例集成的方式来详细讲解 mybatis-plus-generator-ui，感兴趣的朋友可以自己 clone 下来，也可以自己进行扩展自定义。

一、mybatis-plus-generator-ui 是什么？

它是对 mybatis-plus-generator 进行封装，通过 Web UI 快速生成兼容 Spring boot，mybatis-plus 框架的各类业务代码。提供交互式的 Web UI 用于生成兼容 mybatis-plus 框架的相关功能代码，包括 Entity、Mapper、Mapper.xml、Service、Controller 等，可以自定义模板以及各类输出参数，也可通过 SQL 查询语句直接生成代码。

功能列表：

Table 查询：查询配置的关系型数据库表的列表查询。
输出配置：对需要生成的相关代码，比如 Entity、Mapper、Servcie、Controller 等代码模板信息进行配置，用于在转换时调用。
项目导入：可以导入其它项目配置好的信息给本项目使用。
下载模板：支持本项目配置的模板信息下载后共享。
策略配置：直接定义各种文件的生成策略。
模板上传：支持从别的项目中下载模板，同上传供本项目使用。
SQL 输入上传：支持将查询语句直接上传或者复制到输入框中。
SQL 代码生成：基于 SQL 脚本生成相应的代码。

二、 mybatis-plus-generator-ui 怎么用？

mybatis-plus-generator-ui 提供以 jar 包的形式为外部项目提供服务，通过配置的数据库配置去读取数据库的配置信息，并通过 Web UI 的方式提供给开发者使用。mybatis-plus-generator-ui 支持 POSTGRE_SQL、ORACLE、DB2、MySQL、SQLSERVER 等常见的关系型数据库。

1、maven pom 引入

<project xmlns="http://maven.apache.org/POM/4.0.0"
 xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
 <modelVersion>4.0.0</modelVersion>
 <groupId>com.yelang</groupId>
 <artifactId>mybatis-plus-generator-ui-case</artifactId>
 <version>0.0.1-SNAPSHOT</version>

 <dependencies>
  <dependency>
   <groupId>com.github.davidfantasy</groupId>
   <artifactId>mybatis-plus-generator-ui</artifactId>
   <version>1.4.5</version>
  </dependency>

  <dependency>
   <groupId>org.postgresql</groupId>
   <artifactId>postgresql</artifactId>
   <version>42.2.25</version>
  </dependency>
 </dependencies>
</project>

2、新建程序入口，以 main 函数的方式运行

mybatis-plus-generator-ui 在 1.4.0 版本之后，可支持将 GeberatorUIServer 独立部署为一个单独的 spring boot 项目，通过页面指定目标项目根目录的方式为多个项目提供源码生成服务。这种方式适用于有多个项目库需要独立进行开发的模式。实例关键代码如下：

package com.yelang;

import com.github.davidfantasy.mybatisplus.generatorui.GeneratorConfig;
import com.github.davidfantasy.mybatisplus.generatorui.MybatisPlusToolsApplication;
import com.github.davidfantasy.mybatisplus.generatorui.mbp.NameConverter;

public class GeneratorMain {

 public static void main(String[] args) {
  GeneratorConfig config = GeneratorConfig.builder().jdbcUrl("jdbc:postgresql://127.0.0.1:5432/ghyapp")
    .userName("ghy01").password("ghy01").driverClassName("org.postgresql.Driver")
    // 数据库schema，POSTGRE_SQL,ORACLE,DB2类型的数据库需要指定
    // .schemaName("myBusiness")
    // 如果需要修改各类生成文件的默认命名规则，可自定义一个NameConverter实例，覆盖相应的名称转换方法：
    .nameConverter(new NameConverter() {
     /**
      * 自定义Service类文件的名称规则
      */
     public String serviceNameConvert(String tableName) {
      return this.entityNameConvert(tableName) + "Service";
     }

     /**
      * 自定义Controller类文件的名称规则
      */
     public String controllerNameConvert(String tableName) {
      return this.entityNameConvert(tableName) + "Action";
     }
    }).basePackage("com.github.davidfantasy.mybatisplustools.example").port(8068).build();

  MybatisPlusToolsApplication.run(config);

 }

}

在上面的配置中，我们连接的示例数据库是 PostgerSQL，需要在 Maven 中定义相应的驱动程序，并且在上述代码中正确配置相应的类。最后指定了程序的运行端口为 8086，这种运行方式跟 SpringBoot 非常相似。

3、实例运行

运行以上的 main 方法，在控制台可以看到以下输出即为成功部署。

在输出的日志中，可以看到程序的运行端口，以及默认的模板目录地址。在浏览器中输入访问地址http://localhost:8068/，即可进行配置生成。

三、mybatis-plus-generator-ui 代码生成

1、Table 的查询和浏览

可以直接浏览和查询配置的数据源的数据表信息，可选择一个或多个生成模板代码。

2、输出配置

内置 Entity、Mapper、Service、Controller 等 6 种类型代码的模板配置，可以上传模板进行替换，并修改各类参数，配置参数已经按照影响的文件类型重新进行了分类，并加入了部分文本说明；也可以自行添加其它类型的自定义输出文件。所有的配置项都会按照项目包名进行保存，只需一次性设置就可以了。

3、策略配置

将每次生成代码时可能变动的内容加入到代码生成选项中，方便调整每次的生成策略，比如：是否覆盖原文件，生成文件的种类等等：

4、SQL 配置生成

通过输入查询 SQL，可自动在 Mapper（Xml 及 Java）中生成对应的查询方法，DTO 对象和 ResultMap（结果集映射配置）

5、代码生成

四、自定义扩展

1、相关模板调整

在相关的页面中，可以进行相应的调整，在对应的 btl 模板中下载对应文件的具体模板，使用文本工具打开，直接修改源代码，文中取一种方式示例，其它方式一样。

2、代码层级的配置

在一些团队中，肯定对 Mapper 的定义为 Dao，Controller 层需要定义为 Action，通过修改代码模板 btl 的方式是可以的，还有一种方式是通过调整内部映射的方式来进行修改。主要使用的类是 NameConverter。

/**
* 自定义Service类文件的名称规则
*/
public String serviceNameConvert(String tableName) {
 return this.entityNameConvert(tableName) + "Service";
}

/**
* 自定义Controller类文件的名称规则
*/
public String controllerNameConvert(String tableName) {
  return this.entityNameConvert(tableName) + "Action";
}

除了 Service、Controller、Entity、FieldName 都可以实现自定义的扩展。下面是 NameConverter 类的核心代码，这里有详细的定义。

package com.github.davidfantasy.mybatisplus.generatorui.mbp;

import cn.hutool.core.util.StrUtil;
import com.github.davidfantasy.mybatisplus.generatorui.dto.Constant;
import com.google.common.base.Strings;

import static com.github.davidfantasy.mybatisplus.generatorui.dto.Constant.DOT_JAVA;
import static com.github.davidfantasy.mybatisplus.generatorui.dto.Constant.DOT_XML;

/**
 * 自定义各类名称转换的规则
 */
public interface NameConverter {

    /**
     * 自定义Entity.java的类名称
     *
     * @param tableName 表名称
     * @return
     */
    default String entityNameConvert(String tableName) {
        if (Strings.isNullOrEmpty(tableName)) {
            return "";
        }
        tableName = tableName.substring(tableName.indexOf(StrUtil.UNDERLINE) + 1, tableName.length());
        return StrUtil.upperFirst(StrUtil.toCamelCase(tableName.toLowerCase()));
    }

    /**
     * 自定义表字段名到实体类属性名的转换规则
     *
     * @param fieldName 表字段名称
     * @return
     */
    default String propertyNameConvert(String fieldName) {
        if (Strings.isNullOrEmpty(fieldName)) {
            return "";
        }
        if (fieldName.contains("_")) {
            return StrUtil.toCamelCase(fieldName.toLowerCase());
        }
        return fieldName;
    }

    /**
     * 自定义Mapper.java的类名称
     */
    default String mapperNameConvert(String tableName) {
        return entityNameConvert(tableName) + "Mapper";
    }

    /**
     * 自定义Mapper.xml的文件名称
     */
    default String mapperXmlNameConvert(String tableName) {
        return entityNameConvert(tableName) + "Mapper";
    }

    /**
     * 自定义Service.java的类名称
     */
    default String serviceNameConvert(String tableName) {
        return "I" + entityNameConvert(tableName) + "Service";
    }

    /**
     * 自定义ServiceImpl.java的类名称
     */
    default String serviceImplNameConvert(String tableName) {
        return entityNameConvert(tableName) + "ServiceImpl";
    }

    /**
     * 自定义Controller.java的类名称
     */
    default String controllerNameConvert(String tableName) {
        return entityNameConvert(tableName) + "Controller";
    }

    /**
     * 自定义其它生成文件的文件名（不包括entity,mapper.java,mapper.xml,service,serviceImpl,controller这6种）
     *
     * @param fileType  在页面上输入的输出文件标识
     * @param tableName 关联的数据表名称名称
     * @return 生成文件的名称，带后缀
     */
    default String outputFileNameConvert(String fileType, String tableName) {
        if (fileType.equals(Constant.FILE_TYPE_ENTITY)) {
            return this.entityNameConvert(tableName) + DOT_JAVA;
        } else if (fileType.equals(Constant.FILE_TYPE_MAPPER)) {
            return this.mapperNameConvert(tableName) + DOT_JAVA;
        } else if (fileType.equals(Constant.FILE_TYPE_MAPPER_XML)) {
            return this.mapperXmlNameConvert(tableName) + DOT_XML;
        } else if (fileType.equals(Constant.FILE_TYPE_SERVICE)) {
            return this.serviceNameConvert(tableName) + DOT_JAVA;
        } else if (fileType.equals(Constant.FILE_TYPE_SERVICEIMPL)) {
            return this.serviceImplNameConvert(tableName) + DOT_JAVA;
        } else if (fileType.equals(Constant.FILE_TYPE_CONTROLLER)) {
            return this.controllerNameConvert(tableName) + DOT_JAVA;
        }
        return this.entityNameConvert(tableName) + fileType;
    }

}