17370845950

MyBatis Generator自定义插件：简化Entity类代码生成

使用MyBatis Generator生成实体类时，默认只包含getter和setter方法，需要手动添加构造方法和toString方法，增加额外工作量。本文介绍如何通过自定义插件，让MyBatis Generator自动生成包含构造方法和toString方法的Entity类。

需求：

自动生成无参构造方法和全参构造方法，并重写toString方法，简化开发流程。

例如，名为User的实体类，期望生成如下代码：

package com.example.baseproject.entity;

public class User {

    private Integer id;
    private String name;
    private String email;
    private String cellphone;

    // ... getter and setter methods ...

    public User(Integer id, String name, String email, String cellphone) {
        this.id = id;
        this.name = name;
        this.email = email;
        this.cellphone = cellphone;
    }

    public User() {}

    @Override
    public String toString() {
        return "User{" + "id=" + id + ", name='" + name + '\'' + ", email='" + email + '\'' + ", cellphone='" + cellphone + '\'' + '}';
    }
}

解决方案：

虽然MyBatis Generator自带ToStringPlugin，但其生成的代码可能与实际需求不符。更灵活的方案是自定义插件。 我们可以参考ToStringPlugin源码，并进行修改以满足需求。这只需要少量代码修改，即可实现自动生成无参构造函数、全参构造函数以及重写toString方法的功能。

通过创建自定义插件并配置到MyBatis Generator配置文件中，即可在生成Entity类时自动包含这些方法，显著提高开发效率，避免重复劳动，使代码更简洁易维护。