Idea连接上数据库生成带注解的实体类

我的使用场景：idea 2018 数据库：oracle

 

1. idea右侧点击database拉出面板 右侧找不到去调出来
2. 选择你的数据库类型 本教程适用于大部分数据库
3. 填写数据库连接信息 添加driver 点击test测试一下是否可以连上
4. 上一步成功之后 在面板找到你想要操作的表右键 选择Genetate POJOS.groovy 也可以选择 Genetate POJOS.cli 但是我的cli不知道为什么不行就不用了。最后用的是Genetate POJOS.groovy。要注意的是，idea它自带的Genetate POJOS.groovy生成的实体类不带@Table @Entity @Column等注解的。所以我在网上找到一个可以生成注解的 Genetate POJOS.groovy文件 代码在最下方
5. 选择你要生成实体类的包所在文件夹
6. Genetate 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.io.*
   import java.text.SimpleDateFormat
   
   /*
    * Available context bindings:
    *   SELECTION   Iterable<DasObject>
    *   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;"
   }