IDEA自定义POJO.groovy

最新推荐文章于 2023-07-13 15:20:08 发布

李博客

最新推荐文章于 2023-07-13 15:20:08 发布

阅读量374

点赞数

文章标签： java

本文链接：https://blog.csdn.net/weixin_40863853/article/details/124347534

版权

参考：IDEA自定义Generate POJOs.groovy_祥哥哥咩~的博客-CSDN博客


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.time.LocalDateTime
import java.time.format.DateTimeFormatter

packageName = "com.daoshu.alarm.entity;"
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)//更改实体生成规则
  new File(dir, className + ".java").withPrintWriter("utf-8"){ out -> generate(out, className, fields, table) }//输出实体类
}

def generate(out, className, fields, table) {
  out.println "package $packageName"
  out.println ""
  out.println "import java.io.Serializable;"
  out.println "import com.fasterxml.jackson.databind.annotation.JsonSerialize;"
  out.println "import com.fasterxml.jackson.annotation.JsonFormat;"
  out.println "import io.swagger.annotations.ApiModel;"
  out.println "import io.swagger.annotations.ApiModelProperty;"
  out.println "import java.util.Date;"
  out.println ""
  out.println "@ApiModel(value = \"" + table.getComment() + "\")"
  out.println "public class $className  implements Serializable {"
  fields.each() {
    out.println ""
    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 ""
  fields.each() {
    out.println ""
    out.println "  public ${it.type} get${it.name.capitalize()}() {"
    out.println "    return ${it.name};"
    out.println "  }"
    out.println ""
    out.println "  public void set${it.name.capitalize()}(${it.type} ${it.name}) {"
    out.println "    this.${it.name} = ${it.name};"
    out.println "  }"
    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
}

时间类型还有瑕疵需要手动修改