datagrip反向生成实体类格式

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)}