通过数据库表来映射创建entity方法

最新推荐文章于 2024-06-11 13:59:56 发布

辣椒炒苦瓜

最新推荐文章于 2024-06-11 13:59:56 发布

阅读量447

点赞数

文章标签： java

本文链接：https://blog.csdn.net/weixin_44376553/article/details/109066316

版权

本文介绍如何根据数据库表来映射并生成Java实体（Entity）类，从而简化开发过程。

摘要由CSDN通过智能技术生成

通过数据库表来映射创建entity方法：

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 className = javaName(table.getName(), true)
    def fields = calcFields(table)
    packageName = getPackageName(dir)
    PrintWriter printWriter = new PrintWriter(new OutputStreamWriter<