idea database 快速生成實體類 註釋 序列化
阿新 • • 發佈:2018-12-15
按著下面步驟:
這個 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 }
當然這些程式碼一般都會大概看的懂,可以改成只能想要的那種格式