mybatis generator 搭配lombok ,逆向生成字段类注释

逆向生成字段类注释

自定义插 的实现方式有两种:

1,实现接口 **Plugin**
2, 继承抽象类:**PluginAdapter**  (已经实现了接口Plugin ,功能更丰富)-推荐使用

自定义插件说明

我使用的是mybatis-generator-core 是 1.3.7 版本。

					<dependencies>
                        <dependency>
                            <groupId>org.mybatis.generator</groupId>
                            <artifactId>mybatis-generator-core</artifactId>
                            <version>1.3.7</version>
                        </dependency>
                    </dependencies>

打开mybatis-generator-core-1.3.7.jar 我们看到org.mybatis.generator.plugins 下已经有很多插件了
在这里插入图片描述

我们自定义插件也要放在这个包下。所以,我们也要建一个包名为:org.mybatis.generator.plugins。
如果我们随意定的报名,在执行mybatis-generator 时,会找不到你自定义的包名。

下面开始上代码:

逆生成表字段说明

根据数据库表字段,字段说明。逆向生成java 类的时候,自动生成字段注释

MyCommentPlugin.java

package org.mybatis.generator.plugins;

import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.java.Field;
import org.mybatis.generator.api.dom.java.JavaElement;
import org.mybatis.generator.api.dom.java.JavaVisibility;
import org.mybatis.generator.api.dom.java.TopLevelClass;
import org.mybatis.generator.internal.util.StringUtility;
import util.GsonUtil;

import java.util.LinkedHashMap;
import java.util.List;

/**
 * 逆生成字段注释插件
 * @author author
 * @date 2020/7/30 16:54
 */
public class MyCommentPlugin extends PluginAdapter {


    private static final String SEPARATION = ",";
    private static final String JSON_PREFIX = "{";
    private static final String JSON_SUFFIX = "}";

    private boolean makeConstant = true;

    /**
     * 这里已经要返回true
     * true : 插件生效
     * false : 插件失效
     * @param warnings
     * @return
     */
    @Override
    public boolean validate(List<String> warnings) {
        return true;
    }

    @Override
    public boolean modelFieldGenerated(Field field, TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable, ModelClassType modelClassType) {
        this.comment(field, topLevelClass, introspectedColumn, introspectedTable);
        return true;
    }

    private void comment(JavaElement element, TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable) {
        element.getJavaDocLines().clear();
        element.addJavaDocLine("/**");
        //获取列字段注解
        String remark = introspectedColumn.getRemarks();
        if (remark != null && remark.length() > 1) {
            element.addJavaDocLine(" * " + remark);
            element.addJavaDocLine(" *");
        }
        // 根据项目要求,可以加上自定义的注解
    //    element.addJavaDocLine(" * Column:    " + introspectedColumn.getActualColumnName());
    //    element.addJavaDocLine(" * Length:    " + introspectedColumn.getLength());
    //    element.addJavaDocLine(" * DefaultValue:  " + (StringUtility.stringHasValue(introspectedColumn.getDefaultValue()) ? introspectedColumn.getDefaultValue() : "无默认值"));
    //    element.addJavaDocLine(" * Nullable:  " + introspectedColumn.isNullable());

        // mysql 中自增主键,添加 AutoIncrement:  true 说明
        boolean autoIncrement = introspectedColumn.isAutoIncrement();
        if (autoIncrement) {
            element.addJavaDocLine(" * AutoIncrement:  true");
        }
        element.addJavaDocLine(" */");
        if (this.makeConstant) {
            System.out.println("start constant " + introspectedColumn.getActualColumnName());
            if (StringUtility.stringHasValue(remark) && remark.contains(JSON_PREFIX) && remark.contains(JSON_SUFFIX)) {
                //截取常量字符串
                String commentJson = remark.substring(remark.indexOf(JSON_PREFIX), remark.lastIndexOf(JSON_SUFFIX) + 1);
                try {
                    LinkedHashMap<String, String> commentMap = GsonUtil.fromJson(commentJson,LinkedHashMap.class);

                    commentMap.forEach((key, value) -> {
                        Field field = new Field();
                        //常量字段属性名 以列字段名+_+json key 值为字段名
                        field.setName(introspectedColumn.getActualColumnName().toUpperCase() + "_" + key.toUpperCase());
                        field.setStatic(true);
                        field.setFinal(true);
                        //设置常量字段类型与列字段一致
                        field.setType(introspectedColumn.getFullyQualifiedJavaType());
                        //常量字段属性描述
                        String desc = "";
                        //常量字段属性值
                        String constant = "";
                        if (value.contains(SEPARATION)) {
                            String[] split = value.split(SEPARATION);
                            constant = split[0];
                            desc = split[1];
                        } else {
                            constant = value;
                        }
                        field.setInitializationString(constant);
                        field.setVisibility(JavaVisibility.PUBLIC);
                        field.addJavaDocLine("/**");
                        field.addJavaDocLine("* " + introspectedColumn.getActualColumnName() + ":" + desc);
                        field.addJavaDocLine("*/");
                        if (introspectedTable.getTargetRuntime() == IntrospectedTable.TargetRuntime.MYBATIS3_DSQL) {
                            context.getCommentGenerator().addFieldAnnotation(field, introspectedTable,
                                    topLevelClass.getImportedTypes());
                        } else {
                            context.getCommentGenerator().addFieldComment(field, introspectedTable);
                        }
                        //将常量加入BD实体类
                        topLevelClass.addField(field);
                    });
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
    }
}

2,使用lombok 注解方式

MyLombokPlugin.java

package org.mybatis.generator.plugins;

import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
import org.mybatis.generator.api.dom.java.Interface;
import org.mybatis.generator.api.dom.java.Method;
import org.mybatis.generator.api.dom.java.TopLevelClass;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

/**
 * @author author
 * @date 2020/7/30 16:52
 */
public class MyLombokPlugin extends PluginAdapter {

    @Override
    public boolean validate(List<String> list) {
        return true;
    }

    @Override
    public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
        // 到入需要注解的包
        topLevelClass.addImportedType("lombok.Data");
        // topLevelClass.addImportedType("lombok.Builder");
        //  topLevelClass.addImportedType("lombok.NoArgsConstructor");
        // topLevelClass.addImportedType("lombok.AllArgsConstructor");

        // 添加domain的注解
        topLevelClass.addAnnotation("@Data");
        // topLevelClass.addAnnotation("@Builder");
        // topLevelClass.addAnnotation("@NoArgsConstructor");
        //   topLevelClass.addAnnotation("@AllArgsConstructor");

        // 添加domain的注释
        topLevelClass.addJavaDocLine("/**");
        //   topLevelClass.addJavaDocLine("* @ClassName: " + topLevelClass.getType().getShortName());
        //  topLevelClass.addJavaDocLine("* @Description: ");
        topLevelClass.addJavaDocLine("* @author: Mybatis Generator");
        topLevelClass.addJavaDocLine("* @date " + date2Str(new Date()));
        topLevelClass.addJavaDocLine("*/");
        return true;
    }

    @Override
    public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass,
                                   IntrospectedTable introspectedTable) {
        // Mapper文件的注释
        interfaze.addJavaDocLine("/**");
        //   interfaze.addJavaDocLine("* @ClassName: " + interfaze.getType().getShortName());
        //   interfaze.addJavaDocLine("* @Description: ");
        interfaze.addJavaDocLine("* @author: Mybatis Generator");
        interfaze.addJavaDocLine("* @date " + date2Str(new Date()));
        interfaze.addJavaDocLine("*/");
        return true;
    }

    @Override
    public boolean modelSetterMethodGenerated(Method method, TopLevelClass topLevelClass,
                                              IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable, ModelClassType modelClassType) {
        // 不生成getter
        return false;
    }

    @Override
    public boolean modelGetterMethodGenerated(Method method, TopLevelClass topLevelClass,
                                              IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable, ModelClassType modelClassType) {
        // 不生成setter
        return false;
    }

    /**
     * 格式化时间字符串
     *
     * @param date
     * @return
     * @author: renkai721
     * @date: 2018年7月13日 上午11:17:52
     * @Title: date2Str
     * @Description:
     */
    private String date2Str(Date date) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        return sdf.format(date);
    }

}

generatorConfig.xml

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE generatorConfiguration
        PUBLIC "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"
        "http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd">
<generatorConfiguration>
       <!--引入mysql 连接 数据库的jar-->
    <classPathEntry location="D:\m2e\repository\mysql\mysql-connector-java\8.0.15\mysql-connector-java-8.0.15.jar"/>
    <context id="default" targetRuntime="MyBatis3">

        <plugin type="org.mybatis.generator.plugins.MyLombokPlugin">
            <property name="hasLombok" value="true"/>
        </plugin>
        <!--自定义注释插件-->
        <plugin type="org.mybatis.generator.plugins.MyCommentPlugin">
            <!--开启根据注释生成常量,需把常量字段定义为json-->
            <!--模板规则-->
            <!--注释{'常量后缀1':'值#描述','常量后缀2':'值#描述’}-->
            <property name="makeConstant" value="true"/>
            <!--作者-->
            <property name="author" value="author"/>
        </plugin>

        <commentGenerator>
            <property name="suppressDate" value="true"/>
            <property name="suppressAllComments" value="true"/>
        </commentGenerator>

        <jdbcConnection driverClass="com.mysql.jdbc.Driver"
                        connectionURL="jdbc:mysql://ip:3306/库名"
                        userId="username"
                        password="密码">
        </jdbcConnection>

        <javaModelGenerator targetPackage="com.xxx.entity"
                            targetProject="./src/main/java">
            <property name="trimStrings" value="true"/>
            <property name="immutable" value="false"/>
        </javaModelGenerator>
        <sqlMapGenerator targetPackage="dao"
                         targetProject="./src/main/resources">
        </sqlMapGenerator>
        <javaClientGenerator type="XMLMAPPER"
                             targetPackage="com.xxx.dao"
                             targetProject="./src/main/java">
        </javaClientGenerator>
        <table tableName="表名" domainObjectName="XxxEntity"
               enableSelectByPrimaryKey="true" enableUpdateByPrimaryKey="true"
               enableDeleteByPrimaryKey="true">
        </table>
    </context>
</generatorConfiguration>

pom.xml

<?xml version="1.0" encoding="UTF-8"?>

<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.wl.mybatis</groupId>
    <artifactId>mybatis-generator</artifactId>
    <version>1.0-SNAPSHOT</version>
    <packaging>war</packaging>

    <name>mybatis-generator Maven Webapp</name>
    <!-- FIXME change it to the project's website -->
    <url>http://www.example.com</url>

    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <maven.compiler.source>1.8</maven.compiler.source>
        <maven.compiler.target>1.8</maven.compiler.target>
    </properties>

    <dependencies>

        <dependency>
            <groupId>org.mybatis.generator</groupId>
            <artifactId>mybatis-generator-core</artifactId>
            <version>1.3.7</version>
        </dependency>

        <dependency>
            <groupId>com.google.code.gson</groupId>
            <artifactId>gson</artifactId>
            <version>2.8.6</version>
        </dependency>

        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>5.3.5</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.12</version>
        </dependency>
        <dependency>
            <groupId>org.mybatis</groupId>
            <artifactId>mybatis</artifactId>
            <version>3.5.5</version>
        </dependency>

    </dependencies>

    <build>
        <finalName>mybatis-generator</finalName>
        <pluginManagement><!-- lock down plugins versions to avoid using Maven defaults (may be moved to parent pom) -->
            <plugins>
                <!-- mybatis-generator -->
                <plugin>
                    <groupId>org.mybatis.generator</groupId>
                    <artifactId>mybatis-generator-maven-plugin</artifactId>
                    <version>1.3.7</version>
                    <configuration>
                        <!-- mybatis-generator的配置文件,根据情况调整位置 -->
                        <configurationFile>src/main/resources/generatorConfig.xml</configurationFile>
                        <verbose>true</verbose>
                        <overwrite>true</overwrite>
                    </configuration>
                    <dependencies>
                        <dependency>
                            <groupId>org.mybatis.generator</groupId>
                            <artifactId>mybatis-generator-core</artifactId>
                            <version>1.3.7</version>
                        </dependency>
                    </dependencies>
                </plugin>
            </plugins>

        </pluginManagement>
    </build>

</project>

GsonUtil.java

package util;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

/**
 * @author author
 * @date 2020/7/30 17:25
 */
public class GsonUtil {

    private static final GsonBuilder gsonBuilder = new GsonBuilder();
    private static final Gson gson = new Gson();

    public static String toJson(Object obj) {
        return gsonBuilder.serializeNulls().create().toJson(obj);
    }

    public static <T> T fromJson(String str,Class<T> clazz){
        return gson.fromJson(str ,clazz);
    }
}

最后编译MyCommentPlugin.java, MyLombokPlugin.java 建生成 class 文件
在这里插入图片描述
将这两个文件 copy 到
mybatis-generator-core-1.3.7.jar 的 org.mybatis.generator.plugins 目录下

在这里插入图片描述

最后执行:

在这里插入图片描述

最后在工程中就会生成对应的实体类:
在这里插入图片描述

最后说明:
以上两个插件功能也是网上参考大牛写的 再整理优化的。大牛的文档找不到了所以没法贴出对应大牛的文档链接,请见谅!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值