概述
一直以来使用原生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+"
";
}
//找到类名
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(";"))+";
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()即可启动。
项目文件结构截图
演示效果
最终可以看到，实现了我想要的效果。