（SpringBoot-Jpa）使用Idea数据库自动脚本Generate POJOS生成 Entity对象，

拼搏现实的明天。 2022-05-28 12:48 141阅读 0赞

*  因：使用SpringBoot -jpa，需要手动配置Entity

![70][]

*  但是如果你的表中有很多属性，或者有很多表怎么办？？ 每个手动写？ 还是用mybatis。写mapper？？？
 *  解决：使用idea自动工具  
    ![70 1][]
 *  使用工具可以生成pojo对象，但是缺少了注解：
 *  手动修改李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
    
    /*
     * Available context bindings:
     *   SELECTION   Iterable<DasObject>
     *   PROJECT     project
     *   FILES       files helper
     */
    
    packageName = "com.inyu.entity;"
    typeMapping = [
            (~/(?i)int/)                      : "long",
            (~/(?i)float|double|decimal|real/): "double",
            (~/(?i)datetime|timestamp/)       : "java.sql.Timestamp",
            (~/(?i)date/)                     : "java.sql.Date",
            (~/(?i)time/)                     : "java.sql.Time",
            (~/(?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 = javaName(table.getName(), true)
      def fields = calcFields(table)
      new File(dir, className + ".java").withPrintWriter { out -> generate(out,table, className, fields) }
    }
    
    def generate(out,table, className, fields) {
      def tableName = table.getName();
      out.println "package $packageName"
      out.println ""
      out.println ""
      out.println "public class $className {"
      out.println ""
      fields.each() {
        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),
                           colum: col.getName(),
                           type : typeStr,
                           annos: ""]]
      }
    }
    
    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]
    }

[70]: /images/20220528/00606a7dbc9d46ed9345066a0795fc5c.png
[70 1]: /images/20220528/c55a778f4f794393b8eafe6ba363eaef.png