idea默认代码生成脚本修改

修改了下idea自带的代码生成脚本,增加了脚本代码的注释,生成了controller,service,impl,mapper,里面都是空的,具体可以根据自己的代码习惯增加
代码生成脚本的使用可以看下使用 idea 生成实体类代码

修改后的脚本代码

import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
import com.intellij.database.model.ObjectKind
import java.io.*
import java.text.SimpleDateFormat/** Available context bindings:*   SELECTION   Iterable<DasObject>*   PROJECT     project*   FILES       files helper*/
/*** 对应类型转换配置* 数据库类型:java类型*/
typeMapping = [(~/(?i)tinyint|smallint|mediumint/)  : "Integer",(~/(?i)int/)                         : "Long",(~/(?i)bool|bit/)                    : "Boolean",(~/(?i)float|double|decimal|real/)   : "Double",(~/(?i)datetime|timestamp|date|time/): "Date",(~/(?i)/)                            : "String"
]/*** 程序入口*/
FILES.chooseDirectoryAndSave("选择文件夹", "选择controller、domain,service,mapper的上级目录,生成的代码将会保存到对应文件夹。") {dir -> SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}/*** 代码生成* @param table 表的对象* @param dir 选择的文件夹路径*/
def generate(table, dir) {String domainDir = "${dir}/domain"String domainClassName = javaName(table.getName(), true)String domainPackageName = getPackageName(domainDir)// 生成实体类// 这个utf8是解决乱码new File(domainDir, "${domainClassName}.java").withPrintWriter("UTF-8") {out -> generateDomain(out, domainClassName, table, domainPackageName)}// 生成mapperString mapperDir = "${dir}/mapper"String mapperClassName = "${domainClassName}Mapper"String mapperPackageName = getPackageName(mapperDir)new File(mapperDir, "${mapperClassName}.java").withPrintWriter("UTF-8") {out ->generateMapper(out, mapperClassName, mapperPackageName,domainClassName, domainPackageName)}// 生成serviceString serviceDir = "${dir}/service"String serviceClassName = "${domainClassName}Service"String servicePackageName = getPackageName(serviceDir)new File(serviceDir, "${serviceClassName}.java").withPrintWriter("UTF-8") { out -> generateService(out, serviceClassName, servicePackageName) }// 生成serviceImplString servicImplDir = "${serviceDir}/impl"String serviceImplClassName = "${serviceClassName}Impl"String serviceImplPackageName = getPackageName(servicImplDir)new File(servicImplDir, "${serviceImplClassName}.java").withPrintWriter("UTF-8") {out ->generateServiceImpl(out, serviceImplClassName, serviceImplPackageName,serviceClassName, servicePackageName, mapperPackageName,mapperClassName, domainClassName, domainPackageName)}// 生成ControllerString controllerDir = "${dir}/controller"String controllerClassName = "${domainClassName}Controller"String controllerPackageName = getPackageName(controllerDir)new File(controllerDir, "${controllerClassName}.java").withPrintWriter("UTF-8") {out ->generateController(out, controllerClassName, controllerPackageName,serviceImplClassName, serviceImplPackageName, serviceClassName,servicePackageName, domainClassName, domainPackageName)}}/*** 文件夹路径转换为包名* @param dir 文件夹路径* @return 包名*/
static String getPackageName(dir) {return dir.toString().replaceAll("\\\\", ".").replaceAll("/", ".").replaceAll("^.*src(\\.main\\.java\\.)?", "")
}
/*** 生成实体类* @param out 输出流* @param className 类名* @param table 数据库表对象* @param dir 文件夹路径*/
def generateDomain(out, className, table, domainPackageName) {def fields = calcFields(table)String tableName = table.getName();def comment = table.getComment();out.println "package $domainPackageName;"out.println ""out.println "import com.baomidou.mybatisplus.annotation.IdType;"out.println "import com.baomidou.mybatisplus.annotation.TableField;"out.println "import com.baomidou.mybatisplus.annotation.TableId;"out.println "import com.baomidou.mybatisplus.annotation.TableName;"out.println "import java.util.Date;"out.println "import com.fasterxml.jackson.annotation.JsonFormat;"out.println "import java.io.Serializable;"out.println "import lombok.Data;"out.println "import lombok.AllArgsConstructor;"out.println "import lombok.Builder;"out.println "import lombok.NoArgsConstructor;"out.println ""out.println "/**"out.println "* $comment"out.println "* @author zjx"out.println "*/"out.println "@Data"out.println "@TableName(\"$tableName\")"out.println "@NoArgsConstructor"out.println "@AllArgsConstructor"out.println "@Builder"out.println "public class $className implements Serializable{"out.println ""out.println "\t@TableField(exist = false)"out.println genSerialID()out.println ""fields.each() {// 输出注释if (isNotEmpty(it.commoent)) {out.println "\t/**"out.println "\t * ${it.commoent}"out.println "\t */"}if (it.type == "Date") {out.println "\t@JsonFormat(pattern = \"yyyy-MM-dd HH:mm:ss\")"}if (it.annos != "") out.println "  ${it.annos}"out.println "\tprivate ${it.type} ${it.name};"}out.println ""out.println "}"
}/*** 生成mapper* @param out 输出流* @param mapperClassName mapper类名* @param mapperPackageName mapper包名* @param domainClassName 实体类类名* @param domainPackageName 实体类包名*/
void generateMapper(out, mapperClassName, mapperPackageName, domainClassName,domainPackageName) {out.println "package $mapperPackageName;"out.println ""out.println "import $domainPackageName.$domainClassName;"out.println "import com.baomidou.mybatisplus.core.mapper.BaseMapper;"out.println "import org.apache.ibatis.annotations.Mapper;"out.println ""out.println "/**"out.println "* @author zjx"out.println "*/"out.println "@Mapper"out.println "public interface $mapperClassName extends BaseMapper<$domainClassName>{"out.println ""out.println "}"
}
/*** 生成service* @param out 输出流* @param serviceClassName service类名* @param servicePackageName service包名*/
void generateService(out, serviceClassName, servicePackageName) {out.println "package $servicePackageName;"out.println ""out.println "/**"out.println "* @author zjx"out.println "*/"out.println "public interface $serviceClassName {"out.println ""out.println "}"
}
/*** 生成ServiceImpl* @param out 输出流* @param serviceImplClassName serviceImpl类名* @param serviceImplPackageName serviceImpl包名* @param serviceClassName service类名* @param service包名* @param mapperPackageName mapper类名* @param mapperClassName mapper包名* @param domainClassName 实体类类名* @param domainPackageName 实体类包名*/
void generateServiceImpl(out, serviceImplClassName, serviceImplPackageName,serviceClassName, servicePackageName, mapperPackageName,mapperClassName, domainClassName, domainPackageName) {String lowerMapperClassName =mapperClassName.length() == 1 ? mapperClassName : mapperClassName[0].toLowerCase() + mapperClassName[1..-1];out.println "package $serviceImplPackageName;"out.println ""out.println "import $servicePackageName.$serviceClassName;"out.println "import $mapperPackageName.$mapperClassName;"out.println "import org.springframework.stereotype.Service;"out.println "import javax.annotation.Resource;"out.println ""out.println "/**"out.println "* @author zjx"out.println "*/"out.println "@Service"out.println "public class $serviceImplClassName implements $serviceClassName {"out.println "\t@Resource"out.println "\tprivate $mapperClassName $lowerMapperClassName;"out.println ""out.println "}"
}/**** 生成ServiceImpl* @param out 输出流* @param serviceImplClassName serviceImpl类名* @param serviceImplPackageName serviceImpl包名* @param serviceClassName service类名* @param service包名* @param controllerClassName controller类名* @param controllerPackageName controller包名* @param domainClassName 实体类类名* @param domainPackageName 实体类包名*/
void generateController(out, controllerClassName, controllerPackageName,serviceImplClassName, serviceImplPackageName, serviceClassName,servicePackageName, domainClassName, domainPackageName){String lowerServiceClassName =serviceClassName.length() == 1 ? serviceClassName : serviceClassName[0].toLowerCase() + serviceClassName[1..-1];String lowerDomainClassName =domainClassName.length() == 1 ? domainClassName : domainClassName[0].toLowerCase() + domainClassName[1..-1];out.println "package $serviceImplPackageName;"out.println ""out.println "import $servicePackageName.$serviceClassName;"out.println "import javax.annotation.Resource;"out.println "import org.springframework.web.bind.annotation.RestController;"out.println "import org.springframework.web.bind.annotation.RequestMapping;"out.println ""out.println "/**"out.println "* @author zjx"out.println "*/"out.println "@RestController"out.println "@RequestMapping(\"/${lowerDomainClassName}\")"out.println "public class $controllerClassName {"out.println "\t@Resource"out.println "\tprivate $serviceClassName $lowerServiceClassName;"out.println ""out.println "}"
}/*** 读取处理表的数据* @param table 表的对象* @return 表的数据*/
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() }.valuedef comm = [colName : col.getName(),name    : javaName(col.getName(), false),type    : typeStr,commoent: col.getComment(),annos   : ""]if ("主键".equals(col.getComment().toString()))comm.annos += "\t@TableId(type = IdType.AUTO)"fields += [comm]}
}/*** 表名转换为类名* @param tableName 表名* @param capitalize 是否将首字母转大写* @return 转换后的类名*/
String javaName(tableName, capitalize) {String s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(tableName).collect { Case.LOWER.apply(it).capitalize() }.join("").replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}static def isNotEmpty(content) {return content != null && content.toString().trim().length() > 0
}/*** 生成serialVersionUID*/
static String genSerialID() {return "\tprivate static final long serialVersionUID =  " + Math.abs(new Random().nextLong()) + "L;"
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/777181.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

CAS中的ABA问题

ABA 问题是指在并发编程中&#xff0c;一个值从 A 变成 B&#xff0c;然后再变回 A&#xff0c;而导致误判的问题。在 CAS&#xff08;Compare and Swap&#xff09;等原子操作中&#xff0c;由于线程间的竞争&#xff0c;可能会发生 ABA 问题。 为了解决 ABA 问题&#xff0c…

Pycharm服务器配置python解释器并结合内网穿透实现公网远程开发

&#x1f49d;&#x1f49d;&#x1f49d;欢迎来到我的博客&#xff0c;很高兴能够在这里和您见面&#xff01;希望您在这里可以感受到一份轻松愉快的氛围&#xff0c;不仅可以获得有趣的内容和知识&#xff0c;也可以畅所欲言、分享您的想法和见解。 推荐:kwan 的首页,持续学…

spring boot3自定义注解+拦截器+Redis实现高并发接口限流

⛰️个人主页: 蒾酒 &#x1f525;系列专栏&#xff1a;《spring boot实战》 &#x1f30a;山高路远&#xff0c;行路漫漫&#xff0c;终有归途 目录 写在前面 内容简介 实现思路 实现步骤 1.自定义限流注解 2.编写限流拦截器 3.注册拦截器 4.接口限流测试 写在前…

sql中使用了in直接卡死

项目场景&#xff1a; 最近在做数据迁移的工作&#xff0c;会员库有200多万数据&#xff0c;组织机构有4万数据。要把这些数据迁移到新系统的数据库中&#xff0c;开发新系统之前完全没考虑过数据量。当数据导入之后&#xff0c;页面就各种报错。。。。。。。。。。。。。。。…

外包干了8天,技术退步明显.......

先说一下自己的情况&#xff0c;大专生&#xff0c;19年通过校招进入杭州某软件公司&#xff0c;干了接近4年的功能测试&#xff0c;今年年初&#xff0c;感觉自己不能够在这样下去了&#xff0c;长时间呆在一个舒适的环境会让一个人堕落! 而我已经在一个企业干了四年的功能测…

【ZZULIOJ】1009: 求平均分(Java)

目录 题目描述 输入 输出 样例输入 Copy 样例输出 Copy code 题目描述 已知某位学生的数学、英语和计算机课程的成绩&#xff0c;求该生三门课程的平均分。 输入 输入三个整数&#xff0c;数据之间由空格隔开。 输出 输出占一行&#xff0c;包含一个实数&#xff0c…

Redis命令介绍

一、redis启动&#xff1a; 本地启动&#xff1a;redis-cli 远程启动&#xff1a;redis-cli -h host -p port -a password Redis 连接命令 1 AUTH password 验证密码是否正确 2 ECHO message 打印字符串 3 PING 查看服务是否运行 4 QUIT 关闭当前连接 5 SELECT index 切换…

2024.3.28abc晚训题解

VJ晚训网址点击传送晚训 A题 比较巧的输入方式就是循环n&#xff0c;内层套一个循环7 #include<bits/stdc.h> using namespace std; int main(){int n;cin>>n;for(int i1;i<n;i){int ans0;for(int j1;j<7;j){int x;cin>>x;ansansx; }cout<<ans…

JUC内容概述

复习概念 Sleep和Wait的区别 Sleep是Thread的静态方法&#xff0c;wait是Object的方法&#xff0c;任何对象实例都可以使用sleep不会释放锁&#xff0c;他也不需要占用锁&#xff0c;暂停。wait会释放锁&#xff0c;但是调用他的前提是线程占有锁他们都可以被Interrupted方法…

SQLite数据库文件损坏的可能几种情况(一)

返回&#xff1a;SQLite—系列文章目录 上一篇&#xff1a;SQLiteC/C接口详细介绍sqlite3_stmt类&#xff08;十三&#xff09; 下一篇&#xff1a;SQLite使用的临时文件&#xff08;二&#xff09; 概述 SQLite数据库具有很强的抗损坏能力。如果应用程序崩溃&#xff0c…

鸿蒙hdc使用指导

简介 hdc&#xff08;HarmonyOS Device Connector&#xff09;是HarmonyOS为开发人员提供的用于调试的命令行工具&#xff0c;通过该工具可以在windows/linux/mac系统上与真实设备或者模拟器进行交互。 环境准备 hdc工具通过HarmonyOS SDK获取&#xff0c;存放于SDK的toolch…

2024-3-28 市场情绪强修复

这一轮退潮负反馈都修复了&#xff0c; 艾艾精工 博信股份 安奈尔 永悦科技 大理药业 &#xff0c;高新发展 也补跌了&#xff0c;收尸队也干活了&#xff0c;情绪不修复不接力得最好写照。这轮周期 宁科生物 已经7板&#xff0c;已经追平了 博信股份7板&#xff0c;看明天溢…

Mac安装minio

Mac安装minio 本文介绍使用 mac 安装 MinIO。 所有软件安装优先参考官网&#xff1a;MinIO Object Storage for MacOS — MinIO Object Storage for MacOS #使用 brew 安装 minio brew install minio/stable/minio#找到 minio tong ~ $ brew list minio /opt/homebrew/Cella…

银行监管报送系统介绍(十二):非居民金融账户涉税信息报送

国家税务总局、财政部、中国人民银行、中国银行业监督管理委员会、中国证券监督管理委员会、国家金融监督管理总局2017年5月9日发布、2017年7月1日起施行的《非居民金融账户涉税信息尽职调查管理办法》。 一、《管理办法》出台的背景是什么&#xff1f;   受二十国集团&…

math模块篇(四)

文章目录 math.ldexp(x, i)math.modf(x)math.nextafter(x, y, steps1)math.perm(n, kNone)math.prod(iterable, *, start1) math.ldexp(x, i) math.ldexp(x, i) 是 Python 中 math 模块的一个函数&#xff0c;用于执行二进制标度操作。它接受两个参数&#xff1a;尾数&#xf…

【unity】如何汉化unity Hub

相信大家下载安装unity后看着满操作栏的英文&#xff0c;英文不好的小伙伴们会一头雾水。但是没关系你要记住你要怎么高速运转的机器进入中国&#xff0c;请记住我给出的原理&#xff0c;不懂不代表不会用啊。现在我们就来把编译器给进行汉化。 第一步&#xff1a;我们打开Uni…

【VTKExamples::Points】第十一期 RadiusOutlierRemoval

很高兴在雪易的CSDN遇见你 VTK技术爱好者 QQ:870202403 公众号:VTK忠粉 前言 本文分享VTK样例RadiusOutlierRemoval,并解析接口vtkRadiusOutlierRemoval,希望对各位小伙伴有所帮助! 感谢各位小伙伴的点赞+关注,小易会继续努力分享,一起进步! 你的点赞就是我的…

【消息队列开发】 实现 MqClientTests 类——测试客户端

文章目录 &#x1f343;前言&#x1f333;所需属性&#x1f334;BeforeEach&#x1f332;AfterEach&#x1f38d;API测试⭕总结 &#x1f343;前言 本次开发任务 测试客户端接口 &#x1f333;所需属性 所需要一共三个属性 BrokerServer&#xff1a;服务器 ConnectionFa…

考研数学|武忠祥学习包搭配《660》和《880》

一、660、880、三大计算简单分析 660题 这本题册具有高难度、综合度和深度&#xff0c;属于高质量的题材。我建议不要在基础阶段就着手解决其中的660题&#xff0c;因为这可能会影响你的信心。相反&#xff0c;你可以在基础阶段完成一轮学习后&#xff0c;将这些题目留到强化…

安泰ATA-2161高压放大器在生物传感器研究中的应用

生物传感器&#xff08;biosensor&#xff09;&#xff0c;是一种对生物物质敏感并将其浓度转换为电信号进行检测的仪器&#xff0c;它利用固定化生物成分或生物体作为敏感元件。生物传感器并不专指用于生物技术领域的传感器&#xff0c;它的应用领域还包括环境监测、医疗卫生和…