按着下面步骤:
这个 Generate MyPOJOs.groovy 是我自定义的
在这里创建才可以用
这里创建一个文件即可。把下面代码复制进去即可
import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
import java.io.*
/*
* Available context bindings:
* SELECTION Iterable<DasObject>
* PROJECT project
* FILES files helper
*/
packageName = "com.xxx.busi.entity;;"
typeMapping = [
(~/(?i)int/) : "Long",
(~/(?i)float|double|decimal|real/): "Double",
(~/(?i)datetime|timestamp/) : "java.util.Date",
(~/(?i)date/) : "java.util.Date",
(~/(?i)time/) : "java.util.Date",
(~/(?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 tableNameDispose = table.getName().substring((table.getName().indexOf("dr_") + 2), table.getName().length())
out.println tableNameDispose
def className = javaName(tableNameDispose, true)
out.println className
def fields = calcFields(table)
PrintWriter output = new PrintWriter(new OutputStreamWriter(new FileOutputStream( new File(dir, className + ".java")), "utf-8"))
output.withPrintWriter { out -> generate(out, className, fields,table.getName()) }
}
def generate(out, className, fields,tableName) {
out.println "package $packageName"
out.println "import java.io.Serializable;"
out.println "import javax.persistence.Entity;"
out.println "import javax.persistence.Table;"
out.println ""
out.println "@Entity"
out.println "@Table(name = \"${tableName}\")"
out.println "public class $className implements Serializable{"
out.println ""
def msum= (long)(Math.random()*1000000000000000000+1000000000000000000)
out.println " private static final long serialVersionUID = ${msum}L;"
fields.each() {
if (isNotEmpty(it.commoent)) {
out.println ""
out.println "\t/**"
out.println "\t * ${it.commoent}"
out.println "\t */"
}
if (it.annos != "") out.println " ${it.annos}"
out.println " private ${it.type} ${it.name};"
}
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
fields += [[
name : javaName(col.getName(), false),
type : typeStr,
commoent: 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]
}
def isNotEmpty(content) {
return content != null && content.toString().trim().length() > 0
}
当然这些代码一般都会大概看的懂,可以改成只能想要的那种格式