IDEA 自带生成代码 Generate POJOs.groovy。自定义设置

IDEA自带的代码生成脚本 Generate POJOs.groovy

1.连接对应数据库(此处用MYSQL)

版本为8.0之后需要设置时区。进入命令行（Win + R）连接数据库。

mysql -hlocalhost -uroot -p

输入show variables like’%time_zone’; 显示 SYSTEM 就是没有设置时区。（这里已经设置了）

show variables like'%time_zone';

输入set global time_zone = ‘+8:00’;设置

set global time_zone = '+8:00';

如果Idea还是出现Server returns invalid timezone. Go to Advanced tab and set serverTimezone property manually.
去IDEA处设置

2 自定义POJOs.groovy

找到位置,可以自己复制创建一个属于自己的，原来的不动。

package com.alibaba.mybatis.entity

import com.intellij.database.model.DasTable
import com.intellij.database.model.ObjectKind
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil

import java.text.SimpleDateFormat
import java.time.LocalDateTime
import java.time.format.DateTimeFormatter

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 *   update by yitianRen 20200324
 */

packageName = ""
typeMapping = [
        (~/(?i)int8/)                     : "Long",
        (~/(?i)int/)                      : "Integer",
        (~/(?i)bool/)                     : "Boolean",
        (~/(?i)float|double|decimal|real/): "Double",
        (~/(?i)jsonb/)                    : "JSONObject",
        (~/(?i)datetime|timestamp/)       : "LocalDateTime",
        (~/(?i)date/)                     : "java.util.Date",
        (~/(?i)time/)                     : "java.util.Time",
        (~/(?i)/)                         : "String"
]

FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
    SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    packageName = getPackageName(dir)   //自动生成包的位置
    PrintWriter printWriter = new PrintWriter(new OutputStreamWriter(new FileOutputStream(new File(dir, className + ".java")), "UTF-8"))
    printWriter.withPrintWriter { out -> generate(out, className, fields, table) }
//输出实体类
}

// 获取包所在文件夹路径
def getPackageName(dir) {
    return dir.toString().replaceAll("\\\\", ".").replaceAll("/", ".").replaceAll("^.*src(\\.main\\.java\\.)?", "") + ";"
}

def generate(out, className, fields, table) {
    out.println "package $packageName"
    out.println ""
    out.println "import lombok.Data;"
    out.println "import lombok.EqualsAndHashCode;"
    out.println "import lombok.experimental.Accessors;"
    out.println "import com.baomidou.mybatisplus.annotation.TableId;"
    out.println "import com.baomidou.mybatisplus.annotation.TableField;"
    out.println "import com.baomidou.mybatisplus.annotation.TableName;"
    out.println "import com.fasterxml.jackson.annotation.JsonFormat;"
    out.println "import com.fasterxml.jackson.databind.annotation.JsonSerialize;"
    out.println "import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer;"
    out.println "import io.swagger.annotations.ApiModel;"
    out.println "import io.swagger.annotations.ApiModelProperty;"
    out.println "import java.time.LocalDateTime;"
    out.println ""
    // 生成开头文档
    out.println ""
    out.println "/** \n" +
            " * @author xiaoyi\n" + //1. 修改idea为自己名字
            " * @create " + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()) + " \n" +
            " * @description  \n" +
            " */"
    out.println ""
    out.println "@Data"
    out.println "@TableName(value =\"" + table.getName() + "\")"
    out.println "@Accessors(chain = true)"
    out.println "@EqualsAndHashCode(callSuper = false)"
    out.println "@ApiModel(value = \"" + table.getComment() + "对象\" , description=\"" + table.getName() + "\")"
    out.println "public class $className  implements Serializable {"
    out.println "    private static final long serialVersionUID = 1L;"
    fields.each() {
        out.println ""
        if (it.name == "id") out.println "    @TableId(value=\"${it.field}\")"
        else if (it.type == "JSONObject") out.println "    //@TableName(value =\"" + table.getName() + "\", autoResultMap = true)\n    @TableField(value=\"${it.field}\", typeHandler = JacksonTypeHandler.class)"
        else out.println "    @TableField(value=\"${it.field}\")"
        out.println "    @ApiModelProperty(value=\"${it.commoent}\")"
        if (it.type == "LocalDateTime") out.println "    @JsonSerialize(using = LocalDateTimeSerializer.class)\n    @JsonFormat(shape=JsonFormat.Shape.STRING, pattern=\"yyyy-MM-dd HH:mm:ss\")"
        out.println "    private ${it.type} ${it.name};"
    }
    out.println ""

    out.println "}"
}

def calcFields(table) {
    DasUtil.getColumns(table).reduce([]) { fields, col ->
        def spec = Case.LOWER.apply(col.getDataType().getSpecification())
        def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
        def comm = [
                field   : col.getName(),
                name    : javaName(col.getName(), false),
                type    : typeStr,
                commoent: col.getComment()
        ]
        //对于表中主键自定义注解
        if ("pk".equals(Case.LOWER.apply(col.getName()))) {
            comm.annos = "\t@Id\n"
            //自增主键需要
            comm.annos += "@Column(name = \"" + col.getName() + "\")"
        }
        fields += [comm]//字段对照
    }
}

def javaName(str, capitalize) {
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}


def isNotEmpty(content) {
    return content != null && content.toString().trim().length() > 0
}



def isNotEmpty(content) {
    return content != null && content.toString().trim().length() > 0
}

里面介绍了部分的功能可以自行添加一下自己需要的
成功后结果是这样的

package com.test.entity;

import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.fasterxml.jackson.annotation.JsonFormat;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;

import java.io.Serializable;
import java.time.LocalDateTime;


/**
 * @author xiaoyi
 * @create 2023-07-13 14:33:23
 * @description
 */

@Data
@TableName(value ="test")
@Accessors(chain = true)
@EqualsAndHashCode(callSuper = false)
@ApiModel(value = "test" , description="test")
public class AppWasteDisposalInfo  implements Serializable {
    private static final long serialVersionUID = 1L;

    @TableId(value="id")
    @ApiModelProperty(value="主键id")
    private String id;

    @TableField(value="test1")
    @ApiModelProperty(value="test1")
    private String test1;

    @TableField(value="test2")
    @ApiModelProperty(value="test2")
    private String test2;

    @TableField(value="test3")
    @ApiModelProperty(value="test3")
    private String test3;

    @TableField(value="create_by")
    @ApiModelProperty(value="创建人")
    private String createBy;

    @TableField(value="create_time")
    @ApiModelProperty(value="创建时间")
    @JsonSerialize(using = LocalDateTimeSerializer.class)
    @JsonFormat(shape=JsonFormat.Shape.STRING, pattern="yyyy-MM-dd HH:mm:ss")
    private LocalDateTime createTime;

    @TableField(value="update_by")
    @ApiModelProperty(value="修改人")
    private String updateBy;

    @TableField(value="update_time")
    @ApiModelProperty(value="修改时间")
    @JsonSerialize(using = LocalDateTimeSerializer.class)
    @JsonFormat(shape=JsonFormat.Shape.STRING, pattern="yyyy-MM-dd HH:mm:ss")
    private LocalDateTime updateTime;

}

一个苦逼的程序员自述,今天心血来潮,刚好新项目需要自己生成实体类，然后就记录下来了。最开始自己也遇到很多问题，太久没弄这个了。什么时区问题，自定义里面缺少内容问题。搞得很烦。自己记录下来。以后也能更方便查阅。如果有兄弟们刚好需要。更是美事一件。(好想发个表情包，可是好想不行)