idea自动生成实体类的groovy文件

SomeOtherTime SomeOtherTime 博客等级 墨衍会员 · AI 创作全网分发 墨衍智能分发 本文由作者通过墨衍一键同步至各平台 1分发平台 1.6k累计阅读 1.7k平均单平台 已同步平台 优快云 微信公众号 微博 知乎 掘金 百家号 博客园 抖音 小红书 你的文章也可以这样分发 墨衍会员 ¥399起/年，写一次发全网 我也要 码龄16年
关注
原创 2024-12-17 15:50:37 1.6k 阅读 AI 写同款· GEO 优化 ›
Groovy环境安装及创建第一个Groovy项目 阅读详情

 注意：第一个属性是主键。参考Intellij IDEA 通过数据库表生成带注解的实体类详细步骤_Hunter的博客-优快云博客_idea根据表生成实体类的步骤。

PO:

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.io.*
import java.text.SimpleDateFormat

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */
packageName = "com.zhengyuan.emergence.model.po;"
typeMapping = [
        (~/(?i)tinyint|smallint|mediumint/)      : "Integer",
        (~/(?i)int/)                             : "Integer",
        (~/(?i)bool|bit/)                        : "Boolean",
        (~/(?i)float|double|decimal|real/)       : "Double",
        (~/(?i)datetime|timestamp|date|time/)    : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                                : "String"
]


FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
  SELECTION.filter { it instanceof DasTable && it.getKind() == ObjectKind.TABLE }.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 javax.persistence.*;"
  out.println "import java.io.Serializable;"
  out.println "import lombok.Data;"
  out.println "import lombok.ToString;"
  out.println "import org.hibernate.annotations.DynamicInsert;"
  out.println "import org.hibernate.annotations.DynamicUpdate;"
  out.println "import io.swagger.annotations.ApiModelProperty;"
  Set types = new HashSet()

  fields.each() {
    types.add(it.type)
  }

  if (types.contains("Date")) {
    out.println "import java.util.Date;"
  }

  if (types.contains("InputStream")) {
    out.println "import java.io.InputStream;"
  }
  out.println ""
//    out.println "/**\n" +
//            " * @Description  \n" +
//            " * @Author  Hunter\n" +
//            " * @Date "+ new SimpleDateFormat("yyyy-MM-dd").format(new Date()) + " \n" +
//            " */"
  out.println ""
//    out.println "@Setter"
//    out.println "@Getter"
  out.println "@Data"
  out.println "@ToString"
  out.println "@Entity"
  out.println "@DynamicInsert"
  out.println "@DynamicUpdate"
  out.println "@Table (name =\"" + table.getName() + "\" )"
  out.println "public class $className  implements Serializable {"
  out.println ""
  out.println genSerialID()
  def flag = "1"
  fields.each() {
    out.println ""
    // 输出注释
    if (isNotEmpty(it.commoent)) {
      out.println "\t/**"
      out.println "\t * ${it.commoent.toString()}"
      out.println "\t */"
      out.println "\t@ApiModelProperty(value = \"" + it.commoent + "\")"
    }
    if (isNotEmpty(flag)) {
      out.println "\t@Id"
      flag = ""
    }
    if (it.annos != "") out.println "   ${it.annos.replace("[@Id]", "")}"

    // 输出成员变量
    out.println "\tprivate ${it.type} ${it.name};"
  }

  // 输出get/set方法
//    fields.each() {
//        out.println ""
//        out.println "\tpublic ${it.type} get${it.name.capitalize()}() {"
//        out.println "\t\treturn this.${it.name};"
//        out.println "\t}"
//        out.println ""
//
//        out.println "\tpublic void set${it.name.capitalize()}(${it.type} ${it.name}) {"
//        out.println "\t\tthis.${it.name} = ${it.name};"
//        out.println "\t}"
//    }
  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 = [
            colName : col.getName(),
            name    : javaName(col.getName(), false),
            type    : typeStr,
            commoent: col.getComment(),
            annos   : "\t@Column(name = \"" + col.getName() + "\" )"]
    if ("id".equals(Case.LOWER.apply(col.getName())))
      comm.annos += ["@Id"]
    fields += [comm]
  }
}

// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
//def javaClassName(str, capitalize) {
//    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
//            .collect { Case.LOWER.apply(it).capitalize() }
//            .join("")
//            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
//    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
//    s = s[1..s.size() - 1]
//    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
//}

def javaName(str, capitalize) {
//    def s = str.split(/(?<=[^\p{IsLetter}])/).collect { Case.LOWER.apply(it).capitalize() }
//            .join("").replaceAll(/[^\p{javaJavaIdentifierPart}]/, "_")
//    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
  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
}

static String changeStyle(String str, boolean toCamel) {
  if (!str || str.size() <= 1)
    return str

  if (toCamel) {
    String r = str.toLowerCase().split('_').collect { cc -> Case.LOWER.apply(cc).capitalize() }.join('')
    return r[0].toLowerCase() + r[1..-1]
  } else {
    str = str[0].toLowerCase() + str[1..-1]
    return str.collect { cc -> ((char) cc).isUpperCase() ? '_' + cc.toLowerCase() : cc }.join('')
  }
}

static String genSerialID() {
  return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

Repository:

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.io.*
import java.text.SimpleDateFormat

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */
packageName = "com.example.demo76"
typeMapping = [
        (~/(?i)tinyint|smallint|mediumint/)      : "Integer",
        (~/(?i)int/)                             : "Integer",
        (~/(?i)bool|bit/)                        : "Boolean",
        (~/(?i)float|double|decimal|real/)       : "Double",
        (~/(?i)datetime|timestamp|date|time/)    : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                                : "String"
]


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

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    new File(dir, className + "Repository" + ".java").withPrintWriter("UTF-8") { out -> generate(out, className, fields, table) }
}


def generate(out, className, fields, table) {
    out.println "package ${packageName}.repository;"
    out.println ""
    out.println "import ${packageName}.po.${className};"
    out.println "import org.springframework.data.jpa.repository.JpaRepository;"
    out.println "import org.springframework.data.jpa.repository.JpaSpecificationExecutor;"


    out.println ""
//    out.println "/**\n" +
//            " * @Description  \n" +
//            " * @Author  Hunter\n" +
//            " * @Date "+ new SimpleDateFormat("yyyy-MM-dd").format(new Date()) + " \n" +
//            " */"
    out.println ""
//    out.println "@Setter"
//    out.println "@Getter"
    keyType = ""
    fields.each() {
        if (keyType.equals("")) keyType = it.type
    }
    out.println "public interface ${className}Repository extends JpaRepository<${className}, ${keyType}>, JpaSpecificationExecutor<${className}> {"
    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 = [
                colName : col.getName(),
                name    : javaName(col.getName(), false),
                type    : typeStr,
                commoent: col.getComment(),
                annos   : "\t@Column(name = \"" + col.getName() + "\" )"]
        if ("id".equals(Case.LOWER.apply(col.getName())))
            comm.annos += ["@Id"]
        fields += [comm]
    }
}

// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
//def javaClassName(str, capitalize) {
//    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
//            .collect { Case.LOWER.apply(it).capitalize() }
//            .join("")
//            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
//    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
//    s = s[1..s.size() - 1]
//    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
//}

def javaName(str, capitalize) {
//    def s = str.split(/(?<=[^\p{IsLetter}])/).collect { Case.LOWER.apply(it).capitalize() }
//            .join("").replaceAll(/[^\p{javaJavaIdentifierPart}]/, "_")
//    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
    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
}

static String changeStyle(String str, boolean toCamel) {
    if (!str || str.size() <= 1)
        return str

    if (toCamel) {
        String r = str.toLowerCase().split('_').collect { cc -> Case.LOWER.apply(cc).capitalize() }.join('')
        return r[0].toLowerCase() + r[1..-1]
    } else {
        str = str[0].toLowerCase() + str[1..-1]
        return str.collect { cc -> ((char) cc).isUpperCase() ? '_' + cc.toLowerCase() : cc }.join('')
    }
}

static String genSerialID() {
    return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

mybaitis-plus :

po:

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.io.*
import java.text.SimpleDateFormat

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */
packageName = "com.wisdomcity.laian.river.model.po;"
typeMapping = [
        (~/(?i)tinyint|smallint|mediumint/)      : "Integer",
        (~/(?i)int/)                             : "Integer",
        (~/(?i)bool|bit/)                        : "Boolean",
        (~/(?i)float|double|decimal|real/)       : "Double",
        (~/(?i)datetime|timestamp|date|time/)    : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                                : "String"
]


FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
  SELECTION.filter { it instanceof DasTable && it.getKind() == ObjectKind.TABLE }.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 lombok.Data;"
  out.println "import lombok.AllArgsConstructor;"
  out.println "import lombok.NoArgsConstructor;"
  out.println "import lombok.ToString;"
  out.println "import io.swagger.annotations.ApiModelProperty;"
  out.println "import io.swagger.annotations.ApiModel;"
  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.baomidou.mybatisplus.annotation.IdType;"

  Set types = new HashSet()

  fields.each() {
    types.add(it.type)
  }

  if (types.contains("Date")) {
    out.println "import java.util.Date;"
  }

  if (types.contains("InputStream")) {
    out.println "import java.io.InputStream;"
  }
  out.println ""
//    out.println "/**\n" +
//            " * @Description  \n" +
//            " * @Author  Hunter\n" +
//            " * @Date "+ new SimpleDateFormat("yyyy-MM-dd").format(new Date()) + " \n" +
//            " */"
  out.println ""
//    out.println "@Setter"
//    out.println "@Getter"
  out.println "@Data"
  out.println "@ToString"
  out.println "@NoArgsConstructor"
  out.println "@AllArgsConstructor"
  out.println "@TableName(value = \"" + table.getName() + "\")"
  out.println "@ApiModel(\"" + table.getComment() + "\")"
  out.println "public class $className implements Serializable {"
  out.println ""
  out.println genSerialID()
  def flag = "1"
  fields.each() {
    out.println ""
    // 输出注释
    if (isNotEmpty(it.commoent)) {
//            out.println "\t/**"
//            out.println "\t * ${it.commoent.toString()}"
//            out.println "\t */"
      out.println "\t@ApiModelProperty(value = \"" + it.commoent + "\")"
    }
    if (isNotEmpty(flag)) {
      out.println "${it.annos2}"
      flag = ""
    }
    else if (it.annos != "") out.println "   ${it.annos.replace("[@Id]", "")}"

    // 输出成员变量
    out.println "\tprivate ${it.type} ${it.name};"
  }

  // 输出get/set方法
//    fields.each() {
//        out.println ""
//        out.println "\tpublic ${it.type} get${it.name.capitalize()}() {"
//        out.println "\t\treturn this.${it.name};"
//        out.println "\t}"
//        out.println ""
//
//        out.println "\tpublic void set${it.name.capitalize()}(${it.type} ${it.name}) {"
//        out.println "\t\tthis.${it.name} = ${it.name};"
//        out.println "\t}"
//    }
  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 = [
            colName : col.getName(),
            name    : javaName(col.getName(), false),
            type    : typeStr,
            commoent: col.getComment(),
            annos   : "\t@TableField(value = \"" + col.getName() + "\")",
            annos2   : "\t@TableId(value = \"" + col.getName() + "\",type = IdType.AUTO)"]
    if ("id".equals(Case.LOWER.apply(col.getName())))
      comm.annos += ["@Id"]
    fields += [comm]
  }
}

// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
//def javaClassName(str, capitalize) {
//    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
//            .collect { Case.LOWER.apply(it).capitalize() }
//            .join("")
//            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
//    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
//    s = s[1..s.size() - 1]
//    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
//}

def javaName(str, capitalize) {
//    def s = str.split(/(?<=[^\p{IsLetter}])/).collect { Case.LOWER.apply(it).capitalize() }
//            .join("").replaceAll(/[^\p{javaJavaIdentifierPart}]/, "_")
//    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
  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
}

static String changeStyle(String str, boolean toCamel) {
  if (!str || str.size() <= 1)
    return str

  if (toCamel) {
    String r = str.toLowerCase().split('_').collect { cc -> Case.LOWER.apply(cc).capitalize() }.join('')
    return r[0].toLowerCase() + r[1..-1]
  } else {
    str = str[0].toLowerCase() + str[1..-1]
    return str.collect { cc -> ((char) cc).isUpperCase() ? '_' + cc.toLowerCase() : cc }.join('')
  }
}

static String genSerialID() {
  return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

mapper: 

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.io.*
import java.text.SimpleDateFormat

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */
packageName = "com.zjzy.mapper"
typeMapping = [
        (~/(?i)tinyint|smallint|mediumint/)      : "Integer",
        (~/(?i)int/)                             : "Integer",
        (~/(?i)bool|bit/)                        : "Boolean",
        (~/(?i)float|double|decimal|real/)       : "Double",
        (~/(?i)datetime|timestamp|date|time/)    : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                                : "String"
]


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

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    new File(dir, className + "Mapper" + ".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 com.baomidou.mybatisplus.core.mapper.BaseMapper;"
    out.println "import org.springframework.stereotype.Repository;"
    out.println "import com.zjzy.model.po.${className};"


    out.println ""
//    out.println "/**\n" +
//            " * @Description  \n" +
//            " * @Author  Hunter\n" +
//            " * @Date "+ new SimpleDateFormat("yyyy-MM-dd").format(new Date()) + " \n" +
//            " */"
    out.println ""
//    out.println "@Setter"
//    out.println "@Getter"
    keyType = ""
    fields.each() {
        if (keyType.equals("")) keyType = it.type
    }
    out.println "@Repository"
    out.println "public interface ${className}Mapper extends BaseMapper<${className}>{"
    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 = [
                colName : col.getName(),
                name    : javaName(col.getName(), false),
                type    : typeStr,
                commoent: col.getComment(),
                annos   : "\t@Column(name = \"" + col.getName() + "\" )"]
        if ("id".equals(Case.LOWER.apply(col.getName())))
            comm.annos += ["@Id"]
        fields += [comm]
    }
}

// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
//def javaClassName(str, capitalize) {
//    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
//            .collect { Case.LOWER.apply(it).capitalize() }
//            .join("")
//            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
//    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
//    s = s[1..s.size() - 1]
//    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
//}

def javaName(str, capitalize) {
//    def s = str.split(/(?<=[^\p{IsLetter}])/).collect { Case.LOWER.apply(it).capitalize() }
//            .join("").replaceAll(/[^\p{javaJavaIdentifierPart}]/, "_")
//    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
    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
}

static String changeStyle(String str, boolean toCamel) {
    if (!str || str.size() <= 1)
        return str

    if (toCamel) {
        String r = str.toLowerCase().split('_').collect { cc -> Case.LOWER.apply(cc).capitalize() }.join('')
        return r[0].toLowerCase() + r[1..-1]
    } else {
        str = str[0].toLowerCase() + str[1..-1]
        return str.collect { cc -> ((char) cc).isUpperCase() ? '_' + cc.toLowerCase() : cc }.join('')
    }
}

static String genSerialID() {
    return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

service: 

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.io.*
import java.text.SimpleDateFormat

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */
packageName = "com.zjzy.service.base"
typeMapping = [
        (~/(?i)tinyint|smallint|mediumint/)      : "Integer",
        (~/(?i)int/)                             : "Integer",
        (~/(?i)bool|bit/)                        : "Boolean",
        (~/(?i)float|double|decimal|real/)       : "Double",
        (~/(?i)datetime|timestamp|date|time/)    : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                                : "String"
]


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

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    new File(dir, className + "Service" + ".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 com.zjzy.mapper.${className}Mapper;"
    out.println "import com.zjzy.model.po.${className};"
    out.println "import org.springframework.stereotype.Component;"


    out.println ""
//    out.println "/**\n" +
//            " * @Description  \n" +
//            " * @Author  Hunter\n" +
//            " * @Date "+ new SimpleDateFormat("yyyy-MM-dd").format(new Date()) + " \n" +
//            " */"
    out.println ""
//    out.println "@Setter"
//    out.println "@Getter"
    keyType = ""
    fields.each() {
        if (keyType.equals("")) keyType = it.type
    }
    out.println "@Component"
    out.println "public class ${className}Service extends BaseService<${className}Mapper,${className}>{"
    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 = [
                colName : col.getName(),
                name    : javaName(col.getName(), false),
                type    : typeStr,
                commoent: col.getComment(),
                annos   : "\t@Column(name = \"" + col.getName() + "\" )"]
        if ("id".equals(Case.LOWER.apply(col.getName())))
            comm.annos += ["@Id"]
        fields += [comm]
    }
}

// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
//def javaClassName(str, capitalize) {
//    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
//            .collect { Case.LOWER.apply(it).capitalize() }
//            .join("")
//            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
//    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
//    s = s[1..s.size() - 1]
//    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
//}

def javaName(str, capitalize) {
//    def s = str.split(/(?<=[^\p{IsLetter}])/).collect { Case.LOWER.apply(it).capitalize() }
//            .join("").replaceAll(/[^\p{javaJavaIdentifierPart}]/, "_")
//    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
    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
}

static String changeStyle(String str, boolean toCamel) {
    if (!str || str.size() <= 1)
        return str

    if (toCamel) {
        String r = str.toLowerCase().split('_').collect { cc -> Case.LOWER.apply(cc).capitalize() }.join('')
        return r[0].toLowerCase() + r[1..-1]
    } else {
        str = str[0].toLowerCase() + str[1..-1]
        return str.collect { cc -> ((char) cc).isUpperCase() ? '_' + cc.toLowerCase() : cc }.join('')
    }
}

static String genSerialID() {
    return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

xml 

package com.zjzy.dao.servicebase

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.io.*
import java.text.SimpleDateFormat

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */
packageName = "com.zjzy.mapper"
typeMapping = [
        (~/(?i)tinyint|smallint|mediumint/)      : "Integer",
        (~/(?i)int/)                             : "Integer",
        (~/(?i)bool|bit/)                        : "Boolean",
        (~/(?i)float|double|decimal|real/)       : "Double",
        (~/(?i)datetime|timestamp|date|time/)    : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                                : "String"
]


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

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    new File(dir, className + "Mapper" + ".xml").withPrintWriter("UTF-8") { out -> generate(out, className, fields, table) }
}


def generate(out, className, fields, table) {
    out.println "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" +
            "<!DOCTYPE mapper\n" +
            "        PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\"\n" +
            "        \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">\n" +
            "<mapper namespace=\"${packageName}.${className}Mapper\">"
//    out.println "/**\n" +
//            " * @Description  \n" +
//            " * @Author  Hunter\n" +
//            " * @Date "+ new SimpleDateFormat("yyyy-MM-dd").format(new Date()) + " \n" +
//            " */"
    out.println ""
    out.println "</mapper>"
}
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 = [
                colName : col.getName(),
                name    : javaName(col.getName(), false),
                type    : typeStr,
                commoent: col.getComment(),
                annos   : "\t@Column(name = \"" + col.getName() + "\" )"]
        if ("id".equals(Case.LOWER.apply(col.getName())))
            comm.annos += ["@Id"]
        fields += [comm]
    }
}

// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
//def javaClassName(str, capitalize) {
//    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
//            .collect { Case.LOWER.apply(it).capitalize() }
//            .join("")
//            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
//    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
//    s = s[1..s.size() - 1]
//    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
//}

def javaName(str, capitalize) {
//    def s = str.split(/(?<=[^\p{IsLetter}])/).collect { Case.LOWER.apply(it).capitalize() }
//            .join("").replaceAll(/[^\p{javaJavaIdentifierPart}]/, "_")
//    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
    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
}

static String changeStyle(String str, boolean toCamel) {
    if (!str || str.size() <= 1)
        return str

    if (toCamel) {
        String r = str.toLowerCase().split('_').collect { cc -> Case.LOWER.apply(cc).capitalize() }.join('')
        return r[0].toLowerCase() + r[1..-1]
    } else {
        str = str[0].toLowerCase() + str[1..-1]
        return str.collect { cc -> ((char) cc).isUpperCase() ? '_' + cc.toLowerCase() : cc }.join('')
    }
}

static String genSerialID() {
    return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

收录于
典型demo

当前文章被收录于：

典型demo
34 篇文章 2 人学习
订阅专栏 查看详情
后端

当前文章被收录于：

后端
20 篇文章 0 人学习
订阅专栏 查看详情
标签
#idea #hibernate #java
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
IDEA如何从数据库表生成带字段注释的Model，解决乱码问题，还有自动生成controller问题 都是基于groovy 一、生成Model //import com.intellij.database.model.DasTable //import com.intellij.database.model.ObjectKind //import com.intellij.database.util.Case //import com.intellij.database.util.DasUtil ... 阅读详情

相关推荐

Intellij Idea自动生成实体类

目录前言Intellij Idea自动生成jpa实体类1.idea连接数据库2.添加自定义groovy脚本2.1.domain脚本2.1.1 脚本2.1.2.效果2.2.dao脚本2.2.1.脚本2.2.1.效果2.3.service脚本2.3.1.脚本2.3.1.效果2.4.ext脚本2.4.1.脚本2.4.2.效果3.执行groovy脚本 前言 为了提高开发效率，对于简单但纯粹费时的dao，domain，service我们都希望能根据数据库字段自动生成，于是idea提供了能执行自定义groovy脚本的

qq_38531706的博客 6523

Idea 配置Groovy环境

近期在使用Streamsets中使用到了Groovy Evaluator组件，但是sdc中调试Groovy不方便，索性就在Idea中调试，在此分享一下在Idea中如何配置Groovy环境。

菜鸟叔叔的博客 9264

Groovy编程在IntelliJ IDEA中的实践

在本篇博客中，我们将深入探讨如何在IntelliJ IDEA中创建和运行Groovy项目，并提供详细的步骤和代码示例。本教程适用于Groovy 2.5.3版本，IntelliJ IDEA 2018.2.5社区版以及Java 11环境。

m0_62153576的博客 1733

【groovy #2】使用IDEA创建第一个groovy程序

前言 在groovy第一解决的讲解中已经讲到，使用idea可以进行groovy的程序运行。以下是创建一个groovy工程的方式： [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SH80xgaE-1616136817301)(https://raw.githubusercontent.com/zhubaiyi/MyTypora/main/img/image-20210319143335579.png)] 点击确定后可以产生新的窗口： 注意这里完成了一半，在新的窗口中点击文件—

weixin_43757056的博客 806

ICM42670-P嵌入式开发实战：从寄存器配置到姿态解算全解析

本文全面解析ICM42670-P高性能6轴MEMS运动传感器的嵌入式开发实战，涵盖寄存器配置、数据采集滤波、低功耗优化及姿态解算算法。通过详细代码示例和应用案例，帮助开发者快速掌握从硬件对接到姿态跟踪系统的完整开发流程，适用于可穿戴设备和智能家居等低功耗应用场景。

gold的博客 1064

前端监控体系：从Sentry到LogRocket错误追踪

Sentry是一款开源的实时错误追踪系统，支持多种语言和框架，包括JavaScript、React、Vue等。它提供了丰富的错误信息、堆栈跟踪和环境数据，帮助开发者快速定位和解决问题。LogRocket是一款前端监控和用户行为分析工具，它提供了视频回放、性能监控、错误追踪等功能。与Sentry不同，LogRocket更注重用户行为的全面分析。前端监控体系是确保应用稳定性和用户体验的重要保障。通过集成Sentry和LogRocket等监控工具，开发者可以实时追踪应用中的错误和用户行为，快速定位和解决问题。

shejizuopin的博客 1312

groovy、idea自动生成java-controller、service、serviceImpl、mapper、mapperXML、entity 自动生成实体类

import com/*** @Description 控制器。

weixin_52208935的博客 1913

单细胞分析（五）——使用Harmony进行数据整合和去批次

为了保证单细胞RNA测序数据的准确性和可靠性，需要对数据进行样本去批次处理。去批次的方法包括Harmony，SeuratV3等，这些方法可以有效地消除批次效应和技术噪声，并提高单细胞RNA测序数据的质量，为后续的数据分析提供更加可靠的基础。

阅读文献，记录学习笔记 3万+

使用idea自动生成bean文件

之前使用mybatis+mysql的时候，利用generater插件生成代码。 现在公司使用JPA+sqlserver进行开发。很刺激，换了数据库之后就不知道该怎么写代码了。建表之后重新生成bean。网上搜了一下发现idea可以直接生成bean文件。 一、连接数据库 idea连接数据库，方法基本一样，如图：   二、创建bean文件夹 三、生成Extensions 打开连...

稚子候门 1万+

墨刀原型设计工具操作使用指南及实践操作

该指南通过异常场景处理、像素级参数说明、复杂交互示例，覆盖了墨刀操作的全细节，适合从新手到资深用户的进阶学习。结合 PlantUML 思维导图的层级化展示，可快速定位具体功能的操作要点与问题解决方法，真正实现 “照指南即可操作” 的实用价值。

hy行者勇哥的博客 4943

Ansys Zemax实战：5步搞定衍射光学元件(DOE)设计，附赠超透镜(metalens)避坑指南

本文详细介绍了使用Ansys Zemax进行衍射光学元件(DOE)和超透镜(metalens)设计的5步实战流程。从基础光学系统搭建到相位面设计验证，再到制造准备与实测对比，全面解析了OpticStudio中的关键技巧和常见避坑指南，帮助工程师高效完成复杂光学元件设计。

orange的博客 617

idea自动生成实体类、mapper代码

idea自动生成代码，可以生成实体类，以及mapper文件中的实体类代码、insert代码。

weixin_43993069的博客 887

Labwindows CVI写上位机与STM32下位机通信(二)

接着上篇，这篇主要谈谈流量计数据读取。我们使用的MEMS流量计采用Modbus协议Modbus 使用RS-485 作为硬件载体。 一. Modbus协议   该协议定义了 Modbus 总线Master（主设备）与Slave（从设备）之间的通讯报文格式，对于主设备来说，Modbus 协议是联系上位机（如PC、PLC、HMI 等）的接口，而且所有的通讯都是“透明的”。控制器通信使用主-从技术，即

yang2011079080010的专栏 1万+

【IntelliJ IDEA工具】spring boot使用JPA自动生成实体类（数据库表逆向工程自动生成带注释的实体类）

1、maven依赖： <!-- JPA，用于自动生成表结构 --> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-jpa</artifactId> </dependency> &

Tastill的博客 1780

深度强化学习：Actor-Critic、A2C、A3C

Actor-Critic算法是强化学习中常用的一种方法，policy-based，通过演员和评论员或者是策略网络和价值网络的结合进行学习

weixin_51995147的博客 1427

RPG MV游戏数据修改指南：从JSON解析到脚本进阶

游戏数据修改是游戏开发与逆向工程中的常见实践，其核心原理在于解析和操作游戏资源文件。许多现代游戏引擎采用结构化数据格式（如JSON）存储游戏配置，这为外部修改提供了可能。通过理解文件结构、数据格式与游戏逻辑的关联，开发者或爱好者可以调整游戏平衡、修复问题或实现个性化定制，这不仅能深化对游戏设计机制的理解，也具备实际的技术学习价值。本文以基于Web技术栈的RPG Maker MV（RPG MV）引擎为例，其游戏数据多以明文JSON存储，结构清晰。文章将具体阐述如何安全地修改角色属性、物品效果、敌人数据及技能公

weixin_29001419的博客 403

mysql自动生成的groovy文件

import com.intellij.database.model.DasTable import com.intellij.database.util.Case import com.intellij.database.util.DasUtil /* * Available context bindings: * SELECTION Iterable<DasObject> * PROJECT project * FILES files helpe

weixin_44749897的博客 219

Devexpress16.2.6破解工具

Devexpress16.2.6破解工具Devexpress16.2.6破解工具Devexpress16.2.6破解工具Devexpress16.2.6破解工具

idea中导出数据库的表的实体类并带有注释

一、连接数据库 打开项目： 1、点击右侧的Database图标 2、点击 + 号 3、选择 datasource 4、选择 mysql 5、填写数据库连接的 IP地址，比如本地数据库可以填写：localhost或者127.0.0.1 6、填写数据库开放的端口号，一般没设置的话默认都是3306 7、你需要连接的数据库名 8、写数据库的用户名 9、写数据库密码 10、个驱动需要点击下载，图中是已经下载好了 11、自己的数据库连接url，然后可以点击9所在按钮进行测试连接，本地连接失败检查是否开启了mysq.

Alain999的博客 2503

WINCC7.3破解文件

WINCC7.3硬件够破解文件，有需要的下载！WINCC7.3硬件够破解文件，有需要的下载！

VB摄像头监控

该VB模块可对摄像头捕获的画面进行录像等，可在创建的窗口进行控制

在idea中连接mysql,将数据表自动生成实体类

步骤: 选中数据表,右键 --> scripted Extensions --> Generate Pojos.groovy --> 选择存放实体类的文件夹 --> ok 注意 : 要修改package 的全类名 如果数据库中的字段后面有注释, 可以将Scratches and Consoles --> Extensions -->Database Tools and SQL --> shema 文件夹中的 Generate POJOs.groovy ...

yiXin_Chen的博客 929

复分析 阿尔福斯(Ahlfors) 第三版答案 英文

答案分为十章，有部分习题没有答案，全英文复分析阿尔福斯(Ahlfors) 第三版答案

idea 配置mysql逆向_使用Go语言+IDEA+mysql 逆向生成dao 和 domain 的方法

参考地址：使用方式第一步：安装idea:自行百度第二步：Idea配置mysql 连接View -> Tool Windows -> DataSource 出现数据连接页面点击 + 号，出现数据库连接设置，name: 连接名称comments: 备注host:本地或者内网ipuser:用户名password：密码Database：连接数据库名称url: jdbc连接注意此处如果是mysq...

weixin_32141627的博客 362

idea 通过groovy生成实体

idea 通过groovy生成实体

GebiMr_hu的博客 353

IDEA不愧为神器，结合Groovy脚本，简直无敌！

作者：人间哪有四月天segmentfault.com/a/1190000020189801 导语如今，有许许多多的插件或者编辑器都支持根据数据表自动生成数据实体类了, 比如IDEA, ...

Java笔记虾 1251

idea怎么手动导入database_idea database 导入实体带注解

截取表名 1变成0Generate MyPOJOs.groovyimport com.intellij.database.model.DasTableimport com.intellij.database.model.ObjectKindimport com.intellij.database.util.Caseimport com.intellij.database.util.DasUtili...

weixin_33228847的博客 512

Java开发指南&&装环境捷径

Linux 安装GIT [https://blog.youkuaiyun.com/qq_42690368/article/details/82319238]

qiu125897的博客 168

springboot jpa 自动生成实体类的 文件 Generate POJOs.groovy

idea 通过 jpa Generate POJOs.groovy 自动生成实体类的 文件 直接粘贴就可以用 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.SimpleDateF

你宇哥就是你宇哥的博客 913
上一篇： （转）BT下载不死！Magnet（磁力链接）开创网络BT2.0时代！！
下一篇： 根据数据库表名和主键自动生成 springbootJpa的 Dao层和Service层工具类
点赞 0
评论 0
书签 书签 2
SomeOtherTime
SomeOtherTime
博客等级 码龄16年 206粉丝 308原创
+ 关注 专栏目录 订阅专栏
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
表情包 插入表情
表情包 代码片
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值