IDEA如何从数据库表生成带字段注释的Model，解决乱码问题，还有自动生成controller问题

最新推荐文章于 2024-03-07 09:31:49 发布

Tonels

最新推荐文章于 2024-03-07 09:31:49 发布

阅读量1.3k

点赞数

分类专栏： Idea groovy 文章标签： Idea逆向工工程 groovy乱码

本文链接：https://blog.csdn.net/qq_42105629/article/details/99663702

版权

本文介绍了如何在IDEA中从数据库表生成带有字段注释的Model，并解决了Groovy文件乱码问题。同时，文章详细讲解了如何自动生成Controller，帮助开发者提高工作效率。

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

写在前面

刚开始也从网上找了很多，但都不好用，注释，注解什么的都不太好，后来参照几个对应修改的

关于IDEA连接数据库的问题，参考一下，这个,这里主要是对之前的Groovy文件优化，解决乱码，调整出自己需要的生成脚本

一、生成Model

//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 com.google.common.base.CaseFormat
import com.intellij.database.model.DasObject
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.sample;"
typeMapping = [
        (~/(?i)bigint/)                      : "Long",
        (~/(?i)int/)                      : "Integer",
        (~/(?i)float|double|decimal|real/): "Double",
        (~/(?i)datetime|timestamp/)       : "java.time.LocalDateTime",
        (~/(?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.replace('_', '') + ".java").withPrintWriter('UTF-8') { out -> generate(out, className, fields) }
}

def generate(out, className, fields) {
  out.println "package $packageName"
  out.println ""
  out.println ""
  out.println "import lombok.Getter;"
  out.println "import lombok.Setter;"
  out.println "import javax.persistence.