IDEA导出数据库对应的实体配置

  笔者时常因为各种原因重装IDEA软件，导致有些配置找不到，或者忘记如何进行配置的。本次是自我配置（结合AI提示）编写了一个比较完整的 .groovy 文件。也可能因为使用的规则不同，请结合自己的需求进行修改.。

import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
import com.intellij.database.model.ObjectKind
import java.time.LocalDate
import java.time.format.DateTimeFormatter/** Available context bindings:*   SELECTION   Iterable<DasObject>*   PROJECT     project*   FILES       files helper*/packageName = "com.wjin.domain;"
typeMapping = [// 数值类型(~/(?i)tinyint/)            : "Byte",(~/(?i)smallint/)           : "Short",(~/(?i)int|integer/)        : "Integer",(~/(?i)bigint/)             : "Long",(~/(?i)decimal|numeric/)    : "BigDecimal",(~/(?i)real|float/)         : "Float",(~/(?i)double/)             : "Double",// 字符串类型(~/(?i)char|varchar|text|longtext|mediumtext|tinytext/) : "String",// 日期时间类型(~/(?i)date/)               : "Date",(~/(?i)time/)               : "Date",(~/(?i)datetime|timestamp/) : "Date",// 其他特殊类型(~/(?i)bit/)                : "Boolean",(~/(?i)bool|boolean/)       : "Boolean",(~/(?i)blob|binary|varbinary|longblob|mediumblob|tinyblob/) : "byte[]",(~/(?i)year/)               : "Integer",// 默认类型(~/(?i).*/)                 : "String"
]FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->def tables = SELECTION.filter { it instanceof DasTable && it.getKind() == ObjectKind.TABLE }def totalTables = tables.size()def processedCount = 0tables.each { table ->try {def className = javaName(table.getName(), true)def fields = calcFields(table)def file = new File(dir, className + ".java")// 检查文件是否已存在if (file.exists()) {// 这里可以根据需要添加用户确认逻辑println "Warning: File ${file.name} already exists and will be overwritten"}// 检查目录写入权限if (!dir.canWrite()) {throw new IOException("Cannot write to directory: ${dir.absolutePath}")}// 使用UTF-8编码写入文件file.withPrintWriter("UTF-8") { out ->generate(out, className, fields, table)}processedCount++println "Generated: ${file.name} (${processedCount}/${totalTables})"} catch (Exception e) {println "Error processing table ${table.getName()}: ${e.message}"// 可以选择继续处理其他表或停止执行}}
}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 ""out.println "/**"def tableComment = table.getComment() ?: className// 确保注释正确编码out.println " * @Desc: ${tableComment.toString()}"out.println " * @Author: w_jin"// 使用更可靠的日期格式化方法def currentDate = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd"))out.println " * @Date ${currentDate}"out.println " */"out.println "public class $className {"out.println ""fields.each() {if (it.annos != "") out.println "  ${it.annos}"out.println "    /**"def fieldComment = it.comment ?: it.nameout.println "     * ${fieldComment.toString()}"out.println "     */"out.println "  private ${it.type} ${it.name};"}out.println ""fields.each() {out.println ""out.println "   /**"def fieldComment = it.comment ?: it.nameout.println "    * 获取${fieldComment.toString()}"// 对Date类型使用特殊处理if (it.type == "Date") {out.println "    * @return ${it.name} ${fieldComment.toString()}"out.println "    */"out.println "  public ${it.type} get${it.name.capitalize()}() {"out.println "    return ${it.name} != null ? (Date) ${it.name}.clone() : null;"out.println "  }"} else {out.println "    * @return ${it.name} ${fieldComment.toString()}"out.println "    */"out.println "  public ${it.type} get${it.name.capitalize()}() {"out.println "    return ${it.name};"out.println "  }"}out.println ""out.println "   /**"out.println "    * 设置${fieldComment.toString()}"// 对Date类型使用特殊处理if (it.type == "Date") {out.println "    * @param ${it.name} ${fieldComment.toString()}"out.println "    */"out.println "  public void set${it.name.capitalize()}(${it.type} ${it.name}) {"out.println "    this.${it.name} = ${it.name} != null ? (Date) ${it.name}.clone() : null;"out.println "  }"} else {out.println "    * @param ${it.name} ${fieldComment.toString()}"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.getDasType().getSpecification())def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.valuefields += [[name : javaName(col.getName(), false),type : typeStr,comment: col.getComment(),annos: ""]]}
}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]
}