IDEA + Groovy 腳本一鍵生成實體類，用法舒服，高效！
來自：CSDN，作者：悲涼的秋風
鏈接：https://blog.csdn.net/qq_34371461/article/details/80571281
一、連接數(shù)據(jù)庫
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
?*???PROJECT?????project
?*???FILES???????files?helper
?*/
packageName?=?""
typeMapping?=?[
????????(~/(?i)tinyint|smallint|mediumint/)??????:?"Integer",
????????(~/(?i)int/)?????????????????????????????:?"Long",
????????(~/(?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?=?javaClassName(table.getName(),?true)
????def?fields?=?calcFields(table)
????packageName?=?getPackageName(dir)
????PrintWriter?printWriter?=?new?PrintWriter(new?OutputStreamWriter(new?FileOutputStream(new?File(dir,?className?+?".java")),?"UTF-8"))
????printWriter.withPrintWriter?{out?->?generate(out,?className,?fields,table)}
?
//????new?File(dir,?className?+?".java").withPrintWriter?{?out?->?generate(out,?className,?fields,table)?}
}
?
//?獲取包所在文件夾路徑
def?getPackageName(dir)?{
????return?dir.toString().replaceAll("\\\\",?".").replaceAll("/",?".").replaceAll("^.*src(\\.main\\.java\\.)?",?"")?+?";"
}
?
def?generate(out,?className,?fields,table)?{
????out.println?"package?$packageName"
????out.println?""
????out.println?"import?javax.persistence.Column;"
????out.println?"import?javax.persistence.Entity;"
????out.println?"import?javax.persistence.Table;"
????out.println?"import?java.io.Serializable;"
????out.println?"import?lombok.Getter;"
????out.println?"import?lombok.Setter;"
????out.println?"import?lombok.ToString;"
????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?"@ToString"
????out.println?"@Entity"
????out.println?"@Table?(?name?=\""+table.getName()?+"\"?)"
????out.println?"public?class?$className??implements?Serializable?{"
????out.println?""
????out.println?genSerialID()
????fields.each()?{
????????out.println?""
????????//?輸出注釋
????????if?(isNotEmpty(it.commoent))?{
????????????out.println?"\t/**"
????????????out.println?"\t?*?${it.commoent.toString()}"
????????????out.println?"\t?*/"
????????}
?
????????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;"
}
????javax.persistence
????persistence-api
????1.0.2
逆鋒起筆是一個專注于程序員圈子的技術平臺，你可以收獲最新技術動態(tài)、最新內(nèi)測資格、BAT等大廠的經(jīng)驗、精品學習資料、職業(yè)路線、副業(yè)思維，微信搜索逆鋒起筆關注！
評論
圖片
表情