一直以来使用原生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()即可启动。

项目文件结构截图

演示效果

最终可以看到，实现了我想要的效果。