idea @value提示_idea通过数据库生成java实体类（lombok版）

weixin_39870664

墨衍会员 · AI 创作全网分发码龄9年

关注

最新推荐文章于 2026-05-05 06:13:20 发布 614 阅读 AI 写同款· GEO 优化 ›

点击上方蓝色字体，选择“星标公众号”

优质文章，第一时间送达

1.连接数据库, 使用idea自带的插件Database

2.点击表名,右键。Scripted Extensions => Generate POJOs.groovy

3.选择生成的实体类的位置

4.自定义的实体类已生成，其中@Data为我们生成了基本的get set toString方法，@Id注解在tk.mybatis插件中需要使用到，@ApiModelProperty注解生成swagger文档

Generate POJOs.groovy 脚本解释

import com.intellij.database.model.DasTableimport com.intellij.database.util.Caseimport com.intellij.database.util.DasUtil /* * Available context bindings: *   SELECTION   Iterable *   PROJECT     project *   FILES       files helper */ packageName = "com.topex.admin.entity;"    // 此处指定包路径，也就对应实体类中的package com.topex.admin.entity;typeMapping = [    // 此处指定对应的类型映射，如下：数据库中bigint对应生成java的Long，int|tinyint生成Integer...        (~/(?i)bigint/)                      : "Long",        (~/(?i)int|tinyint/)                 : "Integer",        (~/(?i)float|double|decimal|real/): "java.math.BigDecimal",        (~/(?i)date|datetime|timestamp/)       : "java.util.Date",        (~/(?i)time/)                     : "java.sql.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 PrintWriter(new OutputStreamWriter(new FileOutputStream( new File(dir, className + ".java")), "utf-8")).withPrintWriter { out -> generate(out, className, fields) }} def generate(out, className, fields) {   // 从这里开始，拼实体类的具体逻辑代码  out.println "package $packageName"  out.println ""  out.println ""  out.println "import lombok.Data;"     // 因为我使用了lombok插件，使用到了Data注解，所以在引包时加了这一行  out.println "import io.swagger.annotations.ApiModelProperty;"   // 同上，使用了swagger文档，所以引入到需要的注解  out.println "import javax.persistence.Id;"  // tk.mybatis插件需用时需要@id注解，所以引入，不需要就去掉  out.println ""  out.println "@Data"  out.println "public class $className {"  out.println ""  int i = 0  fields.each() {   // 遍历字段，按下面的规则生成    // 输出注释，这里唯一的是id特殊判断了一下，如果判断it.name == id, 则多添加一行@Id    if (it.name == "id") {      if (!isNotEmpty(it.commoent)) {        out.println "\t/**"        out.println "\t * 主键id"        out.println "\t */"        out.println "\t@ApiModelProperty(value = \"主键id\", position = ${i})"      }      out.println "\t@Id"    }    if (isNotEmpty(it.commoent)) {      out.println "\t/**"      out.println "\t * ${it.commoent}"      out.println "\t */"      out.println "\t@ApiModelProperty(value = \"${it.commoent}\", position = ${i})"    }    if (it.annos != "") out.println "  ${it.annos}"    out.println "\tprivate ${it.type} ${it.name};"    out.println ""    i++  }  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    fields += [[                       name : javaName(col.getName(), false),                       type : typeStr,                       commoent: col.getComment(),                       annos: ""]]  }} def isNotEmpty(content) {  return content != null && content.toString().trim().length() > 0} 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]}

温暖提示

为了方便大家更好的学习，本公众号经常分享项目干货源码案例给大家去练习，如果本公众号没有你要学习的功能案例，你可以联系小编(微信：wcy18898375730)提供你帮助哦！