IDEA:数据库表生成pojo(lombok版)和service/dao层

前言

公司自定义的eclipse插件能根据表自动生成mapper\controller\entity\service代码,也就是所谓的样板代码,用IDEA应该也有插件可以生成,不过不能自己定义相关内容,鉴于springBoot JPA对于CRUD来说开发量很少,很契合这种自动生成的样板代码。

方法

1.调出并配置项目所用的表

参考https://blog.csdn.net/sanqianweidu/article/details/106660590

2.找到所在位置并更改文件

  • 在这里插入图片描述
  • 替换文件,代码如下
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.teacher.manage.demo.bo;"    // 此处指定包路径,也就对应实体类中的package com.topex.admin.entity;
typeMapping = [    // 此处指定对应的类型映射,如下:数据库中bigint对应生成java的Long,int|tinyint生成Integer...
                   (~/(?i)bigint/)                      : "Long",
                   (~/(?i)int|tinyint/)                 : "Integer",
                   (~/(?i)float|double|decimal|real/): "java.math.BigDecimal",
                   (~/(?i)date|datetime|timestamp/)       : "java.util.Date",
                   (~/(?i)time/)                     : "java.sql.Time",
                   (~/(?i)/)                         : "String"
]

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 PrintWriter(new OutputStreamWriter(new FileOutputStream( new File(dir, className + ".java")), "utf-8")).withPrintWriter { out -> generate(out, className, fields) }
}

def generate(out, className, fields) {   // 从这里开始,拼实体类的具体逻辑代码
    out.println "package $packageName"
    out.println ""
    out.println "import lombok.Data;"     // 因为我使用了lombok插件,使用到了Data注解,所以在引包时加了这一行
    //out.println "import io.swagger.annotations.ApiModelProperty;"   // 同上,使用了swagger文档,所以引入到需要的注解
    out.println "import org.hibernate.annotations.DynamicUpdate;\n" +
            "\n" +
            "import javax.persistence.Entity;\n" +
            "import org.hibernate.annotations.GenericGenerator;\n" +
            "" +
            "import javax.persistence.Column;\n" +
            "import javax.persistence.GeneratedValue;\n" +
            "import javax.persistence.GenerationType;"

    out.println "import javax.persistence.Id;"  // tk.mybatis插件需用时需要@id注解,所以引入,不需要就去掉
    out.println ""
    out.println "@Entity"
    out.println "@DynamicUpdate"
    out.println "//@Proxy(lazy = false)"
    out.println "@Data"
    out.println "public class $className {"
    out.println ""
    int i = 0
    fields.each() {   // 遍历字段,按下面的规则生成
        // 输出注释,这里唯一的是id特殊判断了一下,如果判断it.name == id, 则多添加一行@Id
        if(i==0 ){
            if (it.type == "Integer") {
                out.println "\t@Id"
                out.println "\t@GeneratedValue(strategy = GenerationType.IDENTITY)"
                out.println "\t@Column(name = \"${it.realname}\")"
            }else if (it.type == "String") {
                out.println "\t@Id"
                out.println "\t@GenericGenerator(name=\"idGenerator\", strategy=\"uuid\") \n" +
                        "    @GeneratedValue(generator=\"idGenerator\")\n" +
                        "    @Column(name=\"${it.realname}\",length = 32)"
            }
        } else {
            if (isNotEmpty(it.commoent)) {
                out.println "\t/**"
                out.println "\t * ${it.commoent}"
                out.println "\t */"
            }
            if (it.annos != "") out.println "  ${it.annos}"
            out.println "\t@Column(name = \"${it.realname}\")"
        }
        out.println "\tprivate ${it.type} ${it.name};"
        out.println ""
        i++
    }
    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),
                           type : typeStr,
                           commoent: col.getComment(),
                           annos: "",
                           realname: col.getName()
                   ]]
    }
}

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

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

3.选择所有的表,右键选择这个文件后,会弹窗,选择生成的位置

在这里插入图片描述
所有表的bo就生成啦~
同理 service\serviceImp\dao层的脚本如下

  • service
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.teacher.manage.demo"    // 此处指定包路径,也就对应实体类中的package com.topex.admin
typeMapping = [    // 此处指定对应的类型映射,如下:数据库中bigint对应生成java的Long,int|tinyint生成Integer...
                   (~/(?i)bigint/)                      : "Long",
                   (~/(?i)int|tinyint/)                 : "Integer",
                   (~/(?i)float|double|decimal|real/): "java.math.BigDecimal",
                   (~/(?i)date|datetime|timestamp/)       : "java.util.Date",
                   (~/(?i)time/)                     : "java.sql.Time",
                   (~/(?i)/)                         : "String"
]

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 PrintWriter(new OutputStreamWriter(new FileOutputStream( new File(dir, className + "Service.java")), "utf-8")).withPrintWriter { out -> generate(out, className, fields) }
}

def generate(out, className, fields) {   // 从这里开始,拼实体类的具体逻辑代码
    out.println "package ${packageName}.service;"
    out.println ""
    out.println "import ${packageName}.bean.*;\n" +
            "import ${packageName}.vo.*;\n"

    out.println "import java.util.List;"
    out.println "public interface ${className}Service {"

    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),
                           type : typeStr,
                           commoent: col.getComment(),
                           annos: ""]]
    }
}

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

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]
}
  • serviceImp
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.teacher.manage.demo"    // 此处指定包路径,也就对应实体类中的package com.topex.admin
typeMapping = [    // 此处指定对应的类型映射,如下:数据库中bigint对应生成java的Long,int|tinyint生成Integer...
                   (~/(?i)bigint/)                      : "Long",
                   (~/(?i)int|tinyint/)                 : "Integer",
                   (~/(?i)float|double|decimal|real/): "java.math.BigDecimal",
                   (~/(?i)date|datetime|timestamp/)       : "java.util.Date",
                   (~/(?i)time/)                     : "java.sql.Time",
                   (~/(?i)/)                         : "String"
]

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 PrintWriter(new OutputStreamWriter(new FileOutputStream( new File(dir, className + "ServiceImpl.java")), "utf-8")).withPrintWriter { out -> generate(out, className, fields) }
}

def generate(out, className, fields) {   // 从这里开始,拼实体类的具体逻辑代码
    out.println "package ${packageName}.service.impl;"
    out.println "import org.springframework.beans.BeanUtils;\n" +
            "import org.springframework.beans.factory.annotation.Autowired;\n" +
            "import org.springframework.data.domain.Page;\n" +
            "import org.springframework.data.domain.PageRequest;\n" +
            "import org.springframework.data.domain.Pageable;\n" +
            "import org.springframework.data.domain.Sort;\n" +
            "import org.springframework.stereotype.Service;\n" +
            "import org.springframework.transaction.annotation.Transactional;\n" +
            "import lombok.extern.slf4j.Slf4j;"
    out.println "import ${packageName}.bean.*;\n" +
            "import ${packageName}.vo.*;\n" +
            "import ${packageName}.enums.*;\n" +
            "import ${packageName}.exception.*;\n" +
            "import ${packageName}.dao.${className}Dao;\n" +
            "import ${packageName}.service.${className}Service;"
    out.println "@Service\n" +
            "@Slf4j\n" +
            "//@Transactional"
    out.println "public class ${className}ServiceImpl implements ${className}Service {"
    out.println "   @Autowired"
    out.println "   private ${className}Dao ${Case.LOWER.apply(className[0]) + className[1..-1]}Dao;"
    out.println "   \n" +
            "   /**\n" +
            "   @Override\n" +
            "    public List<UserMsgVo> getAllUserMsgByUserId(Integer pageNum, Integer pageLimit, String userId) {\n" +
            "        Sort sort = new Sort(Sort.Direction.DESC, \"date\");\n" +
            "        Pageable pageable = PageRequest.of(pageNum - 1, pageLimit, sort);\n" +
            "        Page<Message> msgPage = messageDao.findByFromId(userId, pageable);\n" +
            "    }\n" +
            "   */"
    out.println "   //BeanUtils.copyProperties(messageParam, msg);"
    out.println "   //log.info(\"保存/更新用户={}\", user);"
    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),
                       type : typeStr,
                       commoent: col.getComment(),
                       annos: ""]]
    }
}

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

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

  • dao
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.teacher.manage.demo"    // 此处指定包路径,也就对应实体类中的package com.topex.admin
typeMapping = [    // 此处指定对应的类型映射,如下:数据库中bigint对应生成java的Long,int|tinyint生成Integer...
                   (~/(?i)bigint/)                      : "Long",
                   (~/(?i)int|tinyint/)                 : "Integer",
                   (~/(?i)float|double|decimal|real/): "java.math.BigDecimal",
                   (~/(?i)date|datetime|timestamp/)       : "java.util.Date",
                   (~/(?i)time/)                     : "java.sql.Time",
                   (~/(?i)/)                         : "String"
]

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 PrintWriter(new OutputStreamWriter(new FileOutputStream( new File(dir, className + "Dao.java")), "utf-8")).withPrintWriter { out -> generate(out, className, fields) }
}

def generate(out, className, fields) {   // 从这里开始,拼实体类的具体逻辑代码
    out.println "package ${packageName}.dao;"
    out.println ""
    out.println "import org.springframework.data.jpa.repository.JpaRepository;\n" +
            "import org.springframework.data.jpa.repository.Query;\n" +
            "import org.springframework.stereotype.Repository;"

    out.println "import org.springframework.data.domain.Page;\n" +
            "import org.springframework.data.domain.Pageable;" +
            "import java.util.List;" +
            "import ${packageName}.bo.${className};"
    out.println "@Repository"
    out.print "public interface ${className}Dao extends JpaRepository<$className, "
    //首列是主键
    if(fields[0].type == "String"){
        out.print "String"
    } else if(fields[0].type == "Integer"){
        out.print "Integer"
    }

    out.println "> {"
    out.println "    /**\n" +
            "  List<${className}> findAll();\n" +
            "  List<${className}> findAll(Sort sort);\n" +
            "  <S extends ${className}> List<S> save(Iterable<S> entities);\n" +
            "  void flush(); \n" +
            "  ${className} saveAndFlush(${className} entity); \n" +
            "  void deleteInBatch(Iterable<${className}> entities);\n" +
            "  void deleteAllInBatch();\n"+
            "  Page<${className}> findAll(Pageable pageable);\n" +
            "  <S extends ${className}> S save(S entity);\n" +
            "  <S extends ${className}> Iterable<S> save(Iterable<S> entities);\n" +
            "  ${className} findOne(ID id);\n" +
            "  boolean exists(ID id);\n" +
            "  Iterable<${className}> findAll();\n" +
            "  Iterable<${className}> findAll(Iterable<ID> ids);\n" +
            "  long count();\n" +
            "  void delete(ID id);\n" +
            "  void delete(${className} entity);\n" +
            "  void delete(Iterable<? extends ${className}> entities);\n" +
            "  void deleteAll();" +
            "  @Query(value = \"select new ${packageName}.dto.xx(jud.teachId, jud.courseId, jud.judgeType, jud.judgeId) from ${className} jud where jud.courseId = :courseId \" +\n" +
            "            \"and jud.teachId = :teacherId and jud.valid = :valid\"\n" +
            "            )\n" +
            "   List<${className}> findByTeachIdAndCourseIdAndValid(String teacherId, String courseId, String valid);\n" +
            "   Page<${className}> findByFromId(String fromId, Pageable pageable);\n" +
            "    @Query(value = \"select info.category_Type, info.answer from judge_info info where info.judge_Id = :judgeId\",\n" +
            "            nativeQuery = true)\n" +
            "    List<Object[]> findCategoryAndAnswerByJudgeId(@Param(\"judgeId\") String judgeId);\n" +
            "    @Query(value = \"update Token set valid = '0' where userId = :userId and valid = '1'\")\n" +
            "    @Modifying\n" +
            "    void updateValidOfUser(@Param(\"userId\") String userId);\n"+
            "*/"

    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),
                           type : typeStr,
                           commoent: col.getComment(),
                           annos: ""]]
    }
}

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

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

我的目录结构如下
在这里插入图片描述

tips:

  • 按需更改。但是中文注释乱码没解决。。
  • bo类字段名爆红?没关联上表啦 虽说没关系但是看着不舒服,参考如下博客就行

https://blog.csdn.net/weixin_46822367/article/details/116916643

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值