使用idea Scripted Extensions生成entity实体类groovy

最新推荐文章于 2023-11-28 10:17:47 发布
最新推荐文章于 2023-11-28 10:17:47 发布
阅读量839 收藏
点赞数
分类专栏： IntelliJ IDEA 文章标签： intellij idea
原文链接：https://blog.csdn.net/linmengmeng_1314/article/details/101599559
版权

在这里插入图片描述
jpa自定义模板

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.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/)       : "BigDecimal",
        (~/(?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 = javaName(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) }

//    new File(dir, className + ".java").withPrintWriter { out -> generate(out, className, fields,table) }
}

// 获取包所在文件夹路径
def getPackageName(dir) {
    return dir.toString().replaceAll("\\\\", ".").replaceAll("/", ".").replaceAll("^.*src(\\.main\\.java\\.)?", "") + ";"
}

def generate(out, className, fields, table) {
    out.println "package $packageName"
    out.println ""
    out.println "import javax.persistence.Column;"
    out.println "import javax.persistence.Entity;"
    out.println "import javax.persistence.Table;"
    out.println "import javax.persistence.Id;"
    out.println "import javax.persistence.GeneratedValue;"
    out.println "import java.io.Serializable;"
    Set types = new HashSet()

    fields.each() {
        types.add(it.type)
    }

    if (types.contains("Date")) {
        out.println "import java.util.Date;"
    }

    if (types.contains("BigDecimal")) {
        out.println "import java.math.BigDecimal;"
    }

    if (types.contains("InputStream")) {
        out.println "import java.io.InputStream;"
    }
    out.println ""
    out.println "/**\n" +
            " * @Description  \n" +
            " * @Author  linmengmeng\n" + //1. 修改idea为自己名字
            " * @Date " + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()) + " \n" +
            " */"
    out.println ""
    out.println "@Entity"
    out.println "@Table ( name =\"" + table.getName() + "\" , schema = \"\")" //2. schema = \"后面添加自己的表空间名称(mysql可以不添加, 不用这个schema属性也行)
    out.println "public class $className  implements Serializable {"
    out.println ""
    out.println genSerialID()
    fields.each() {
        out.println ""
        // 输出注释
        if (isNotEmpty(it.commoent)) {
            out.println "\t/**"
            out.println "\t * ${it.commoent.toString()}"
            out.println "\t */"
        }

        if ((it.annos+"").indexOf("[@Id]") >= 0) out.println "\t@Id"

        if (it.annos != "") out.println "   ${it.annos.replace("[@Id]", "")}"


        // 输出成员变量
        out.println "\tprivate ${it.type} ${it.name};"
    }

    // 输出get/set方法
    fields.each() {
        out.println ""
        out.println "\tpublic ${it.type} get${it.name.capitalize()}() {"
        out.println "\t\treturn this.${it.name};"
        out.println "\t}"
        out.println ""

        out.println "\tpublic void set${it.name.capitalize()}(${it.type} ${it.name}) {"
        out.println "\t\tthis.${it.name} = ${it.name};"
        out.println "\t}"
    }

    // 输出toString方法
    out.println ""
    out.println "\t@Override"
    out.println "\tpublic String toString() {"
    out.println "\t\treturn \"{\" +"
    fields.each() {
        out.println "\t\t\t\t\t\"${it.name}='\" + ${it.name} + '\\'' +"
    }
    out.println "\t\t\t\t'}';"
    out.println "\t}"

    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
        def comm = [
                colName : col.getName(),
                name    : javaName(col.getName(), false),
                type    : typeStr,
                commoent: col.getComment(),
                annos   : "\t@Column(name = \"" + col.getName() + "\" )"]
        if ("id".equals(Case.LOWER.apply(col.getName())))
            comm.annos += ["@Id"]
        fields += [comm]
    }
}

// 这里是处理数据库表前缀的方法，这里处理的是t_xxx命名的表
// 已经修改为使用javaName, 如果有需要可以在def className = javaName(table.getName(), true)中修改为javaClassName
// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要去掉表的前缀，那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
def javaClassName(str, capitalize) {
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
    s = s[1..s.size() - 1]
    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

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]
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

def isNotEmpty(content) {
    return content != null && content.toString().trim().length() > 0
}

static String changeStyle(String str, boolean toCamel) {
    if (!str || str.size() <= 1)
        return str

    if (toCamel) {
        String r = str.toLowerCase().split('_').collect { cc -> Case.LOWER.apply(cc).capitalize() }.join('')
        return r[0].toLowerCase() + r[1..-1]
    } else {
        str = str[0].toLowerCase() + str[1..-1]
        return str.collect { cc -> ((char) cc).isUpperCase() ? '_' + cc.toLowerCase() : cc }.join('')
    }
}

//生成序列化的serialVersionUID 
static String genSerialID() {
    return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

原文地址：https://blog.csdn.net/linmengmeng_1314/article/details/101599559

jpa自定义模板2

import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil

import java.text.SimpleDateFormat
import java.util.Map
import java.util.HashMap

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */

packageName = "cn"//默认的包路径，后续替换
author = System.getProperty("user.name")
pkIdText = "\u4e3b\u952eID";//约定含有“主键ID”的文本注释列为主键
dateTime = getDateTime();
typeMapping = [
        (~/(?i)bigint/)                   : "Long",
        (~/(?i)int/)                      : "Integer",
        (~/(?i)number/)                      : "BigDecimal",
        (~/(?i)float|double|decimal|real/): "Double",
        (~/(?i)tinyint/)                  : "Boolean",
        (~/(?i)datetime|timestamp|date/)  : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                         : "String"
]

def getPackageName(dir) {
    return dir.toString().replaceAll("\\\\", ".").replaceAll("/", ".")
            .replaceAll("^.*src(\\.main\\.java\\.)?", "") + ";"
}

def getDateTime(){
    return new SimpleDateFormat("yyyy/MM/dd HH:mm").format(new Date());
}

def getImportMapping(){
    Map<String , String> map = new HashMap();
    map.put("BigDecimal" , "import java.math.BigDecimal;")
    map.put("Date" , "import java.util.Date;")
    map.put("InputStream" , "import java.io.InputStream;")
    return map;
}


FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
    SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    new File(dir, className + ".java").withPrintWriter("UTF-8") { out -> generate(dir , table , out, className, fields) }
}

/**
 * 生成的时候，选择项目的代码路径，从包路径中找src/java/main路径后的路径开始替换
 */
def generate(dir, table, out, className, fields) {
    packageName = getPackageName(dir);
    tableName = table.getName();
    out.println "package $packageName"
    out.println ""
    //约定按照列的描述中含有主键ID的字段备注为主键，某些关系表中的第一个字段不一定就是主键
    existPk(fields , out);
    out.println "import javax.persistence.*;"
    out.println "import io.swagger.annotations.ApiModel;"
    out.println "import io.swagger.annotations.ApiModelProperty;"
    out.println "import lombok.Data;"

    Set<String> types = new HashSet<>();
    fields.each() {
        types.add(it.type)
    }

    Iterator<String> it = types.iterator();
    while(it.hasNext()){
        String next = it.next();
        if(getImportMapping().containsKey(next)){
            out.println getImportMapping().get(next)
        }
    }

    out.println ""
    out.println "/**"
    out.println " * " + table.getComment()
    out.println " * @auth $author"
    out.println " * @date $dateTime"
    out.println " */"
    out.println "@ApiModel"
    out.println "@Entity"
    out.println "@Data"
    out.println "@Table(name=\"$tableName\")"
    out.println "public class $className {"
    out.println ""
    fields.each() {
        if (it.annos != "") out.println "  ${it.annos}"
        if (it.comment != null && it.comment.indexOf(pkIdText) != -1){
            out.println "    @Id"
            out.println "    @GeneratedValue(generator = \"main-uuid\")"
            out.println "    @GenericGenerator(name = \"main-uuid\", strategy = \"uuid\")"
        }
        out.println "    @Column(name = \"${it.column}\")"
        out.println "    @ApiModelProperty(value = \"${it.comment}\")"
        out.println "    private " + it.type + " ${it.name};"
        out.println ""
    }
    //get set 使用lombok
    /*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), //表字段名称转为Java命名
                           type : typeStr, //表字段类型
                           comment: col.getComment(), //表字段注释
                           column: col.getName(),
                           annos: ""]]
    }
}

def javaName(str, capitalize) {
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

def existPk(fields , out){
    fields.each() {
        if (it.comment != null && it.comment.indexOf(pkIdText) != -1){
            out.println "import org.hibernate.annotations.GenericGenerator;"
        }
    }
}

mybatisplus-lombok-swagger模板

import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil

import java.text.SimpleDateFormat
import java.util.Map
import java.util.HashMap

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */

author = "chendd";
pkIdText = "\u4e3b\u952eID";//约定含有“主键ID”的文本注释列为主键
logicText = "\u903b\u8f91\u5220\u9664";//约定含有逻辑删除的文本列为逻辑删除字段

dateTime = getDateTime();
typeMapping = [
        (~/(?i)bigint/)                   : "Long",
        (~/(?i)int/)                      : "Integer",
        (~/(?i)float|double|decimal|real/): "Double",
        (~/(?i)tinyint/)                  : "Boolean",
        (~/(?i)datetime|timestamp|date/)  : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                         : "String"
]

def getPackageName(dir , suffix) {
    return dir.toString().replaceAll("\\\\", ".").replaceAll("/", ".")
            .replaceAll("^.*src(\\.main\\.java\\.)?", "") + suffix
}

def getDateTime(){
    return new SimpleDateFormat("yyyy/MM/dd HH:mm").format(new Date());
}

def getImportMapping(type){
    Map<String , String> map = new HashMap();
    map.put("Date" , "java.util.Date")
    map.put("InputStream" , "java.io.InputStream")
    if(map.containsKey(type)){
        return map.get(type)
    }
    return type
}

FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
    SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    //生成Model 实体类
    generateModel(table , dir , className , fields);
    //生成Mapper 接口类
    generateMapper(table , dir , className);
    //生成Service 接口类
    generateServiceInterface(table , dir , className);
    //生成Service 实现类
    generateServiceClass(table , dir , className);
    //生成Controller 控制类
    generateController(table , dir , className);
}

def generateModel(table , dir , className , fields){
    String parent = new File(dir.toString()).getPath();
    File folder = new File(parent + "/model");
    if (!folder.exists()) folder.mkdirs();
    new File(folder , className + ".java").withPrintWriter("UTF-8") { out -> generateModelImpl(dir , table , out, className, fields) }
}

/**
 * 生成的时候，选择项目的代码路径，从包路径中找src/java/main路径后的路径开始替换
 */
def generateModelImpl(dir, table, out, className, fields) {
    def packageName = getPackageName(dir , ".model;");
    tableName = table.getName();
    out.println "package $packageName"
    out.println ""
    //约定按照列的描述中含有主键ID的字段备注为主键，某些关系表中的第一个字段不一定就是主键
    modelExistPk(fields , out);
    out.println "import com.baomidou.mybatisplus.annotation.TableField;"
    out.println "import com.baomidou.mybatisplus.annotation.TableName;"
    modelExistLogic(fields , out);
    out.println "import io.swagger.annotations.ApiModel;"
    out.println "import io.swagger.annotations.ApiModelProperty;"
    out.println "import lombok.Data;"
    out.println ""
    out.println "/**"
    out.println " * " + table.getComment()
    out.println " * @author $author"
    out.println " * @date $dateTime"
    out.println " */"
    out.println "@Data"
    out.println "@ApiModel"
    out.println "@TableName(\"$tableName\")"
    out.println "public class $className {"
    out.println ""
    fields.each() {
        if (it.annos != "") out.println "  ${it.annos}"
        if (it.comment.indexOf(pkIdText) != -1){
            out.println "    @TableId(value = \"${it.name}\" , type = IdType.ASSIGN_ID)"
        }
        if (it.comment.indexOf(logicText) != -1){
            out.println "    @TableField(\"${it.name}\")"
            out.println "    @TableLogic"
            out.println "    @JsonIgnore"
            out.println "    @ApiModelProperty(value = \"\u6570\u636e\u72b6\u6001\uff0c\u53ef\u7528\u6216\u7981\u7528\" , example = \"DISABLE\" , hidden = true)"
        } else {
            out.println "    @TableField(\"${it.name}\")"
            out.println "    @ApiModelProperty(value = \"${it.comment}\")"
        }
        out.println "    private " + getImportMapping(it.type) + " ${it.name};"
        out.println ""
    }
    /*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), //表字段名称转为Java命名
                           type : typeStr, //表字段类型
                           comment: col.getComment() == null ? "" : col.getComment(), //表字段注释
                           annos: ""]]
    }
}

def javaName(str, capitalize) {
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

def modelExistPk(fields , out){
    fields.each() {
        if (it.comment.indexOf(pkIdText) != -1){
            out.println "import com.baomidou.mybatisplus.annotation.IdType;"
            out.println "import com.baomidou.mybatisplus.annotation.TableId;"
        }

    }
}

def modelExistLogic(fields , out){
    fields.each() {
        if (it.comment.indexOf(logicText) != -1){
            out.println "import com.baomidou.mybatisplus.annotation.TableLogic;"
            out.println "import com.fasterxml.jackson.annotation.JsonIgnore;"
        }

    }
}

//------------创建Mapper
def generateMapper(table , dir , className){
    String parent = new File(dir.toString()).getPath();
    File folder = new File(parent + "/mapper");
    if (!folder.exists()) folder.mkdirs();
    new File(folder , className + "Mapper.java").withPrintWriter("UTF-8") { out -> generateMapperImpl(table , dir , out, className) }
}

def generateMapperImpl(table , dir , out , className) {
    def packageName = getPackageName(dir , "");
    out.println "package $packageName" + ".mapper;"
    out.println ""
    out.println "import com.baomidou.mybatisplus.core.mapper.BaseMapper;"
    out.println "import $packageName" + ".model." + className + ";"
    out.println ""
    out.println "/**"
    out.println " * " + table.getComment() + "Mapper\u63a5\u53e3\u5b9a\u4e49"
    out.println " * @author $author"
    out.println " * @date $dateTime"
    out.println " */"
    out.println "public interface $className" + "Mapper extends BaseMapper<$className> {"
    out.println ""
    out.println "    "
    out.println "}"
}

//------------创建Service接口
def generateServiceInterface(table , dir , className){
    String parent = new File(dir.toString()).getPath();
    File folder = new File(parent + "/service");
    if (!folder.exists()) folder.mkdirs();
    new File(folder , "I" + className + "Service.java").withPrintWriter("UTF-8") { out -> generateServiceInterfaceImpl(table , dir , out, className) }
}

def generateServiceInterfaceImpl(table , dir , out , className) {
    def packageName = getPackageName(dir , "");
    out.println "package $packageName" + ".service;"
    out.println ""
    out.println "import $packageName" + ".model." + "$className;"
    out.println "import com.baomidou.mybatisplus.extension.service.IService;"
    out.println ""
    out.println "/**"
    out.println " * " + table.getComment() + "Service\u63a5\u53e3\u5b9a\u4e49"
    out.println " * @author $author"
    out.println " * @date $dateTime"
    out.println " */"
    out.println "public interface I$className" + "Service extends IService<$className> {"
    out.println ""
    out.println "    "
    out.println "}"

}

//------------创建Service实现类
def generateServiceClass(table , dir , className){
    String parent = new File(dir.toString()).getPath();
    File folder = new File(parent + "/service/impl");
    if (!folder.exists()) folder.mkdirs();
    new File(folder , className + "Service.java").withPrintWriter("UTF-8") { out -> generateServiceClassImpl(table , dir , out, className) }
}

def generateServiceClassImpl(table , dir , out, className) {
    def packageName = getPackageName(dir , "");
    out.println "package $packageName" + ".service.impl;"
    out.println ""
    out.println "import $packageName" + ".model." + "$className;"
    out.println "import $packageName" + ".mapper." + "$className" + "Mapper;"
    out.println "import $packageName" + ".service." + "I$className" + "Service;"
    out.println "import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;"
    out.println "import org.springframework.stereotype.Service;"
    out.println "import javax.annotation.Resource;"
    out.println ""
    out.println "/**"
    out.println " * " + table.getComment() + "Service\u63a5\u53e3\u5b9e\u73b0"
    out.println " * @author $author"
    out.println " * @date $dateTime"
    out.println " */"
    out.println "@Service"
    out.println "public class $className" + "Service extends ServiceImpl<$className" + "Mapper , $className> implements I$className" + "Service {"
    out.println ""
    out.println "    @Resource"
    out.println "    private $className" + "Mapper " + className.uncapitalize() + "Mapper;"
    out.println ""
    out.println "}"
}

//------------创建Controller控制类
def generateController(table , dir , className) {
    String parent = new File(dir.toString()).getPath();
    File folder = new File(parent + "/controller");
    if (!folder.exists()) folder.mkdirs();
    new File(folder , className + "Controller.java").withPrintWriter("UTF-8") { out -> generateControllerImpl(table , dir , out, className) }
}

def generateControllerImpl(table , dir , out, className) {
    def packageName = getPackageName(dir , "");
    out.println "package $packageName" + ".controller;"
    out.println ""
    out.println "import cn.chendd.blog.base.controller.BaseController;"
    out.println "import $packageName" + ".service." + "I$className" + "Service;"
    out.println "import javax.annotation.Resource;"
    out.println "import io.swagger.annotations.Api;"
    out.println "import io.swagger.annotations.ApiSort;"
    out.println "import org.springframework.stereotype.Controller;"
    out.println "import org.springframework.web.bind.annotation.RequestMapping;"
    out.println "import org.springframework.http.MediaType;"
    out.println ""
    out.println "/**"
    out.println " * " + table.getComment() + "Controller\u63a5\u53e3\u5b9a\u4e49"
    out.println " * @author $author"
    out.println " * @date $dateTime"
    out.println " */"
    out.println "@Api(value = \"" + table.getComment() + "\" , tags = \"" + table.getComment() + "\")"
    out.println "@ApiSort(10)"
    out.println "@RequestMapping(value = \"\" , produces = MediaType.APPLICATION_JSON_VALUE)"
    out.println "@Controller"
    out.println "public class $className" + "Controller extends BaseController {"
    out.println ""
    out.println "    @Resource"
    out.println "    private I$className" + "Service " + className.uncapitalize() + "Service;"
    out.println ""
    out.println "}"
}

mybatisplus-lombok-swagger模板

import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil

import java.text.SimpleDateFormat
import java.util.Map
import java.util.HashMap

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */

packageName = "cn.chendd.blog;"//默认的包路径，后续替换
author = "chendd";
pkIdText = "\u4e3b\u952eID";//约定含有“主键ID”的文本注释列为主键
logicText = "\u903b\u8f91\u5220\u9664";//约定含有逻辑删除的文本列为逻辑删除字段

dateTime = getDateTime();
typeMapping = [
        (~/(?i)bigint/)                   : "Long",
        (~/(?i)int/)                      : "Integer",
        (~/(?i)float|double|decimal|real/): "Double",
        (~/(?i)tinyint/)                  : "Boolean",
        (~/(?i)datetime|timestamp|date/)  : "Date",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                         : "String"
]

def getPackageName(dir , suffix) {
    return dir.toString().replaceAll("\\\\", ".").replaceAll("/", ".")
            .replaceAll("^.*src(\\.main\\.java\\.)?", "") + suffix + ";"
}

def getDateTime(){
    return new SimpleDateFormat("yyyy/MM/dd HH:mm").format(new Date());
}

def getImportMapping(type){
    Map<String , String> map = new HashMap();
    map.put("Date" , "java.util.Date")
    map.put("InputStream" , "java.io.InputStream")
    if(map.containsKey(type)){
        return map.get(type)
    }
    return type
}

FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
    SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}

def generate(table, dir) {
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    //生成Model 文件
    generateModel(table , dir , className , fields);
    //生成Mapper 文件
    //new File(dir , className + ".java")
}

def generateModel(table , dir , className , fields){
    String parent = new File(dir.toString()).getPath();
    File folder = new File(parent + "/model");
    if (!folder.exists()) folder.mkdirs();
    new File(folder , className + ".java").withPrintWriter("UTF-8") { out -> generateModelImpl(dir , table , out, className, fields) }
}

/**
 * 生成的时候，选择项目的代码路径，从包路径中找src/java/main路径后的路径开始替换
 */
def generateModelImpl(dir, table, out, className, fields) {
    packageName = getPackageName(dir , ".model");
    tableName = table.getName();
    out.println "package $packageName"
    out.println ""
    //约定按照列的描述中含有主键ID的字段备注为主键，某些关系表中的第一个字段不一定就是主键
    modelExistPk(fields , out);
    out.println "import com.baomidou.mybatisplus.annotation.TableField;"
    out.println "import com.baomidou.mybatisplus.annotation.TableName;"
    modelExistLogic(fields , out);
    out.println "import io.swagger.annotations.ApiModel;"
    out.println "import io.swagger.annotations.ApiModelProperty;"
    out.println "import lombok.Data;"
    out.println ""
    out.println "/**"
    out.println " * " + table.getComment()
    out.println " * @auth $author"
    out.println " * @date $dateTime"
    out.println " */"
    out.println "@Data"
    out.println "@ApiModel"
    out.println "@TableName(\"$tableName\")"
    out.println "public class $className {"
    out.println ""
    fields.each() {
        if (it.annos != "") out.println "  ${it.annos}"
        if (it.comment.indexOf(pkIdText) != -1){
            out.println "    @TableId(value = \"${it.name}\" , type = IdType.ASSIGN_ID)"
        }
        if (it.comment.indexOf(logicText) != -1){
            out.println "    @TableField(\"${it.name}\")"
            out.println "    @TableLogic"
            out.println "    @JsonIgnore"
            out.println "    @ApiModelProperty(value = \"\u6570\u636e\u72b6\u6001\uff0c\u53ef\u7528\u6216\u7981\u7528\" , example = \"DISABLE\" , hidden = true)"
        } else {
            out.println "    @TableField(\"${it.name}\")"
            out.println "    @ApiModelProperty(value = \"${it.comment}\")"
        }
        out.println "    private " + getImportMapping(it.type) + " ${it.name};"
        out.println ""
    }
    /*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), //表字段名称转为Java命名
                           type : typeStr, //表字段类型
                           comment: col.getComment() == null ? "" : col.getComment(), //表字段注释
                           annos: ""]]
    }
}

def javaName(str, capitalize) {
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

def modelExistPk(fields , out){
    fields.each() {
        if (it.comment.indexOf(pkIdText) != -1){
            out.println "import com.baomidou.mybatisplus.annotation.IdType;"
            out.println "import com.baomidou.mybatisplus.annotation.TableId;"
        }

    }
}

def modelExistLogic(fields , out){
    fields.each() {
        if (it.comment.indexOf(logicText) != -1){
            out.println "import com.baomidou.mybatisplus.annotation.TableLogic;"
            out.println "import com.fasterxml.jackson.annotation.JsonIgnore;"
        }

    }
}
91老码
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
Generate POJOs.groovy
04-23
这个脚本使用也很简单，在需要转换的表上点右键，在”Scripted Extensions”菜单中点击”Generate POJOs.groovy”子菜单，就可以运行这个脚本了，运行后会弹出文件选择框让我们来选择存放生成 POJO 文件的目录，选好后就会生成表对应的 POJO 了。
Unity Script Extensions – Attribute （特性）还算多吧
游戏路上的小学生
01-25 6427
属性 （Attribute）使用 Unity 的C#语言 ，利用属性（Attribute）来类定义和变量定义或区分其他的变量，您可以设置一种特殊行为。* 1例如，您添加[SerializeField]属性变量，私有变量标识序列化。 [SerializeField] int  计数; 序列化您'的值是存储到场景和预制体。因为值存储在meta元数据中，在Inspector检查器中设置的值。 你还可
如何使用idea连接MySQL数据库，并生成表对应的实体类
weixin_42216142的博客
06-02 2483
在开发过程中，有一些同学不喜欢开过多的软件，影响电脑运行效率，Navicat之类的，有时候也会加载较慢。在后台框架没有自动生成实体类的时候，自己手动敲表的实体类也比较耽误时间。所以，很多同学都会偏向于在idea里面连接数据库。下面就简单介绍一下，使用idea去连接mysql数据库的步骤，以及生成实体类的步骤。 STEP1：点击Database → 点击Database框左上角的"+"号 → 选择 Data Source → 选择MySQL； STEP 2：填写数据库连接配置 连接成功后，可以看到数据库及
Scripted Intro
01-20
工程文件 : Scripted Intro 运行环境 : vs2005 使用技术 : directX window编程 命令脚本编程 不是原书自带的程序,是自己写的可以实现同样功能的程序. 有任何疑问可以去http://blog.csdn.net/worm003查阅相关文章.
BasicUsageEnvironment---Copy.zip_ScriptEd
09-23
/ Basic Usage Environment: for a simple, non-scripted, console application // C++ header
idea 链接database使用Scripted Extensions自动生成代码
dongying1751的博客
11-11 3201
说明： 以下内容是根据idea自带的scripted Extentions 中的 Generate POJOs.groovy 改造而来，可以自动生成xml、dao、数据源文件共计5个文件，可以自动匹配商品中心的需求【根据门店、大区、log等自动添加多数据源注解】，如不符合要求可根据groovy语法自行改造。 遗留问题： copyright中的【所】和【、】依旧存在乱码现象； 不能自动生成到项目的所在文件地址，需要手动copy； 只有selectList、insertSelective、select
idea根据数据库自动生成实体bean
longtime的博客
09-17 2354
idea根据数据库自动生成实体beanGenerate MyPOJOs.groovy Generate MyPOJOs.groovy 其他的步骤可以自行csdn一下, 我这里只是贴 .groovy代码: import com.intellij.database.model.DasTable import com.intellij.database.model.ObjectKind import...
IDEA生成带注解、注释的实体类/自定义Genarate POJOS.groovy
weixin_44954052的博客
10-22 2294
使用idea默认配置生成Entity 1、在视图中找到数据库配置，连上数据源； 2、找到需要生成实体类的表，右键选项：Scripted Extensions,IDEA默认脚本为Generate POJOs.groovy，点此选项； 3、在弹出框选择实体类要放置的位置，确认后生成文件； 自定义配置生成Entity 从生成的实体类可以看出并不能满足我们的需求，当字段过多时，自己添加注释注解会浪费大量时间；所以我们可以自定义脚本实现需求； 1、还是在表格右键选项：Scripted Extensions
idea2022没有Scripted Extensions选项
个人网站：https://www.longzipeng.online
09-01 2011
idea2022没有Scripted Extensions选项
idea不显示Extensions/Datebase Tools ans SQL/schema/
滕青山博客
12-08 1120
如果不显示Extensions/Datebase Tools ans SQL/schema/，右键右侧数据库的一张表，scripted extentions/goto script
scripted-pipeline1
04-02
脚本管道1
scriptEd-MapGame
05-30
寻找资本 ##Scope 您刚刚受雇于一家教育公司，负责开发一款可以帮助学生学习地理的游戏。 ##Project 描述 您的经理为该项目制定了以下要求： 创建一个游戏，最多 4 名玩家可以在空白地图上猜测首都的位置。 一旦所有玩家都点击了游戏结束并根据最近的点击显示玩家排名。 点击最接近实际首都的玩家获胜。 ##学习目标 在本项目结束时，员工将能够： 向页面添加 css 和 javascript 资源 使用 jquery API 查找 html 元素并附加点击事件 使用 jquery API 查找 html 元素更改其内容 使用 Leafletjs API 执行基本命令 ##参考/工具 Leafletjs - ( ) - 地图 API jquery ( ) 引导程序（ ） ＃＃指示 设置 index.html 文档并包含所有 javascript 和 css 资源。
改善编程体验: IdeaVimExtension介绍
weixin_34405925的博客
03-21 529
改善编程体验: IdeaVimExtension介绍 动机 Intellij IDEA是个非常优秀的开发环境,提供了相当好用的快捷键,让用户有机会脱离鼠标来使用IDE. 另外官方还提供了IdeaVim插件,熟悉vim的用户可以使用几乎与vim相同的方式操作IDEA的编辑器来编写代码.但是作为非英文用户,却有一个"输入法切换"的痛点. 考虑如下两种情况: ...
idea的根据数据库表生成实体类的Generate POJOS消失了
u012413709的博客
08-11 1504
标题解决方法就是，点击GotoScriptedDirectory跳转到Extensions文件夹，右键查找初始化的选项restorydefaultExtensions,就可以了
在Visual Studio Code中安装加速TypeScript程序开发的插件
最新发布
jackson_lingua的博客
11-28 976
本文简要介绍了在marketplace能够找到的，对TypeScript开发有极大帮助的Extensions(扩展)，这些也成为插件的组件，将使TypeScript开发工作高效有趣，事半功倍。
自己写 IDEA 插件，要啥功能就有啥功能！
weixin_43114485的博客
08-05 793
作者 | 木杉的博客来源 | http://imushan.com写Java代码的时候，经常会涉及到重复性的操作，这个时候就会想要是有这样一个插件就好了，如果是大家都会遇到的场景，IDE或许已经提供了，再不然也有可能有人编写了相关的插件。要是这个操作是你们的编码环境特有的，那就只能自己写工具了。所以这里来学学如何编写IDEA插件，让自己的编程环境更加强大，更好的进行装逼。开发环境开发IDEA插件有以下这些依赖：IntelliJ IDEA Community EditionIntelliJ IDEA Co..
IntelliJ IDEA插件结构
hawkdowen的专栏
11-17 1万+
插件是扩展IDEA功能的唯一途径。一款插件使用IDEA或其他插件暴露的API实现它的功能。这篇文章关注插件系统的结构和插件的生命周期。文章中不会指出任何可能被插件使用的其他API接口。 文章中包含了以下主题： 插件内容 插件类加载器 插件组件 插件扩展和扩展点 插件交互（Action) 插件服务 插件配置文件
idea如何生成实体类
热门推荐
yanchenbin88的专栏
08-15 1万+
前几天一直在网上找idea如何生成实体类的文章，搜索了一些网友写的博客，发现写的都很散，所以今天想写一篇总结下，以后方便查阅      前提：idea连接上了内置数据库，至于如何连接，我这里不多做介绍了，网上很多资料，或者可以看下面的链接教程                http://www.cnblogs.com/jeffen/p/6027970.html     第一种：i
IntelliJ IDEA中有什么让你相见恨晚的好用插件?
jakpopc的博客
06-22 4793
我是Jackpop，我们交个朋友吧！IntelliJ IDEA在开发者中的受欢迎程度，想必不用多说。开箱即用、功能强大，备受欢迎。虽然IDEA已经提供了丰富的功能，但是，我们仍然可以通过配置插件来满足个性化的需求。JetBrains Marketplace 上有着大量实用插件，但是，并非每个都称得上优秀。今天，就给大家精选12款非常好用的IDEA插件。命名规范对于编程语言重要，而且不同编程语言有不同的命名规范要求。有的要求驼峰命名，有的要求下划线命名。通过CamelCase这款插件，我们可以在CamelCa
idea自动生成实体类
09-14
2. 在Database窗口中，右键点击要生成实体类的表，并选择"Scripted Extensions"，然后选择"Generate POJOs.groovy"。 3. 在弹出的对话框中，选择要创建实体类的包，并点击"OK"。这将自动运行Generate POJOs.groovy...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值