Generate POJOs.groovy - 自定义idea数据库反向生成实体类

原创

于 2020-12-14 19:18:59 发布 · 241 阅读

CC 4.0 BY-SA版权

该博客介绍了如何使用GeneratePOJOs.groovy脚本来自定义IntelliJ IDEA数据库反向工程，生成包含Lombok注解的Java实体类。脚本根据数据类型映射为不同的Java类型，并自动导入所需的时间类库。此外，还提供了计算字段的方法，用于构建实体类的属性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Generate POJOs.groovy - 自定义idea数据库反向生成实体类

import com.intellij.database.model.DasTable
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.sample;"
typeMapping = [
        (~/(?i)tinyint/)                  : "Integer",
        (~/(?i)bigint/)                   : "Long",
        (~/(?i)int/)                      : "Integer",
        (~/(?i)float|double|decimal|real/): "String",
        (~/(?i)datetime|timestamp/)       : "LocalDateTime",
        (~/(?i)date/)                     : "LocalDate",
        (~/(?i)time/)                     : "LocalTime",
        (~/(?i)/)                         : "String"
]

FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files"