基于mybatis-generator代码生成工具改(链式方法实体版)
2023-09-11 14:18:03 时间
概述
一直以来使用原生mybatis-generator的我发现有一个地方很不方便,即它生成的实体类的set方法返回值是void,而目前比较流行的则是链式set的写法,即set方法返回值不再是void而是这个类本身,这样的好处是你可以连写,比如demo.setA().setB().setC(),对此,我对它进行了改动,改动的思路比较简单,它底层代码我看不太懂,但我采用了一个简单的方法,即将它生成后的原生实体类通过文件操作改为链式实体类。
详细
代码实现过程
首先说下我的思路,我是将其生成的java文件读取其内容,然后查找到相应的位置,找到目标代码,替换为自己的代码,最后将其写入该java文件。
其核心代码可以说就是一个类处理器,不依赖任何框架(当然也不依赖mybatis-generator)。
下面来看几个核心的类。
1、文件工具类
package com.zs.tools; public class FileUtils { /** * 得到项目根路径 * 比如:E:/Workspaces/eclipse/mybatis-zs-tool/ * @return */ public static String getRootPath(){ String str=FileUtils.class.getResource("/").getFile().toString().substring(1); String rootPath=str.replaceAll("target/classes/", ""); return rootPath; } }
这个比较简单,就是取得项目根路径。
2、实体处理工具类
package com.zs.tools; import java.io.BufferedReader; import java.io.File; import java.io.FileNotFoundException; import java.io.FileReader; import java.io.IOException; import java.io.PrintWriter; import java.util.ArrayList; import java.util.List; /** * 2018-1-25 * 该类的作用是将一个生成的类的setget方法中的set方法改为:返回值为该类 * 比如: * public As setB(String b) { this.b = b; return this; } * @author it023 * */ public class EntityUtils { public static String handleSetTransVoidToClass(File file) throws IOException{ if (!file.isDirectory()) { String tmp=""; FileReader fr=new FileReader(file); BufferedReader br=new BufferedReader(fr); String str; while ((str = br.readLine()) != null) { tmp=tmp+str+"\r\n"; } //找到类名 String className=tmp.substring(tmp.indexOf("class")+5+1, tmp.indexOf("{")-1); //先切开 String ss[]=tmp.split("set"); List<String> litmp=new ArrayList<String>();//容器备用,之所用用它是因为它是变长的 //如果不满足这个条件,说明找错了,那么就将相邻的两个合并 for (int i=0;i<ss.length-1;i++) { /*条件1:切了之后最后一个字符是空格 *条件2:最后一个字符之前的四个字符是“void” *如果满足上面两个条件,说明我们找到了,将其处理好,然后放入容器 *如果不满足上面两个条件,就将它与之后的一个合并,并放入容器 */ String s=ss[i]; String snext=ss[i+1]; if (s.substring(s.length()-1, s.length()).equals(" ") && s.substring(s.length()-1-4, s.length()-1).equals("void")) { //开始处理 String a1=s.substring(0, s.length()-5)+className+" "; ss[i+1]=snext.substring(0, snext.indexOf(";"))+";\r\n return this;"+snext.substring(snext.indexOf(";")+1, snext.length()); litmp.add(a1); }else{ litmp.add(s); } } litmp.add(ss[ss.length-1]); String result=""; for (String s : litmp) { result=result+s+"set"; } result=result.substring(0, result.length()-3); PrintWriter out=new PrintWriter(file); out.write(result); //-关闭流- fr.close(); br.close(); out.flush(); out.close(); } return "修改成功"; } }
其重点就在于如何准确找到set方法的地方,代码注释中有写。
3、启动类
package com.zs.main; import java.io.File; import java.io.IOException; import org.mybatis.generator.api.ShellRunner; import com.zs.tools.EntityUtils; import com.zs.tools.FileUtils; public class App { public static void main( String[] args ) throws IOException{ App.work(); } public static void work() throws IOException { String[] args = new String[] { "-configfile", "resources\\generatorConfig.xml", "-overwrite" }; ShellRunner.main(args); String beanPath=FileUtils.getRootPath()+"resources/com/zs/entity"; File filedir=new File(beanPath); File files[]=filedir.listFiles(); for (File file : files) { EntityUtils.handleSetTransVoidToClass(file); } } }
调用App.work()即可启动。
项目文件结构截图
演示效果
最终可以看到,实现了我想要的效果。
注:本文著作权归作者,由demo大师发表,拒绝转载,转载需要作者授权
相关文章
- Mybatis in 条件传参三种实现方法(直接$,List,[])
- vue打包后出现一些map文件的解决方法
- mybatis返回map类型数据空值字段不显示(三种解决方法)(转载)
- MyBatis——mybatis基本搭建
- Mybatis-Plus 之BaseMapper 方法详解
- 21 面向对象编程 方法回顾和加深 代码:方法的定义 代码:方法的调用
- python在不同层级目录import模块的方法
- Tensorflows安装(cpu版最简安装方法)
- 快速压缩图片方法(小白篇)
- 提升研发团队战斗力的方法
- 前段元素居中的N种方法
- mybatis的分页插件使用方法
- Mybatis 中的 @Mapper, @MapperScan, @Repository,mybatis.mapper-locations 的用法与区别
- 《深入理解mybatis原理》 Mybatis数据源与连接池
- Windows常用维护方法-随时更新
- 2016 系统设计第一期 (档案一)jQuery ajax serialize()方法form提交数据
- Mybatis_mybatis常用jdbcType数据类型
- mybatis方法参数报错:Parameter 'xxx' not found - @requestParam 和@param的区别、desc关键字作为了字段名称查询问题、id取值错乱问题、ERROR:syntax error at or near "$8" in postgres
- 在 Flutter 和 Dart 中取消 Future 的 3 种方法
- Mybatis通用操作数据库方法类总结
- 使用Guava报错NoSuchMethodError的解决方法
- Mybatis自动生成Xml文件,针对字段类型为text等会默认产生XXXXWithBlobs的方法问题
- SQLServer 报错:用户、组或角色'XXX' 在当前数据库中已存在. 错误:15023解决方法
- nodejs jade 模板 引擎的使用方法
- spring,配置文件applictionContext.xml,Mybatis mybatis.xml,springMVC spring整合springMVC mybatis
- Mybatis-plus逆向工程使用方法