springbootJpa生成样板代码
前言
公司自定义的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