Java简单拼接代码生成器

2019.6.27 java

实现:在com.fx.zy.app.下生成相应模块的Controller类,用分号隔开的多个方法

controller_meta.txt

AppLogin,login,appLogin;changePwd;getUserByCode
Mold,mod,mold;signIn;specSwap

GenerateControllerByContact.java

package test;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;

public class GenerateControllerByContact {
	public static void main(String[] args) throws Exception {
		File directory = new File("");
		
		// E:\\sts-4.2.1.RELEASE\\workspace\\fxdev\\src\\main\\java
		String baseDir = directory.getAbsolutePath() + "\\src\\main\\java\\";
		
		String resourceDir = (new GenerateControllerByContact().getClass().getResource("").getPath()).substring(1);
		String basePkg = "com.fx.zy.app.";
		String requestMappingUrl = "/app";
		
		System.out.println("baseDir=" + baseDir);
		System.out.println("basePkg=" + basePkg);
		System.out.println("requestMappingUrl=" + requestMappingUrl);
		System.out.println("resourceDir=" + resourceDir);
		
		String fileName = resourceDir + "controller_meta.txt";
		readFileByLines(fileName, baseDir, basePkg, requestMappingUrl);
	}

	private static void readFileByLines(String fileName, String baseDir, String basePkg, String requestMappingUrl) {
		File file = new File(fileName);
		BufferedReader reader = null;
		try {
			reader = new BufferedReader(new FileReader(file));
			// 生成实体类
			getEntity(reader, baseDir, basePkg, requestMappingUrl);
			reader.close();
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (reader != null)
				try {
					reader.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
		}
	}

	private static void getEntity(BufferedReader reader, String baseDir, String basePkg, String requestMappingUrl) throws IOException {
		// Login,com.fx.zy.app.login,login;changePwd;getUserInfo
		
		String tempString = null;
		String stringEntity = "";
		
		while ((tempString = reader.readLine()) != null) {
			stringEntity = "";
			String[] values = tempString.split(",");
			String className = values[0] + "Controller";
			String pkgName = basePkg + values[1];
			String methods = values[2];
			String destDir = baseDir + pkgName.replace(".", "\\");
			
			File dir = new File(destDir);
			if (!dir.exists()) {
				dir.mkdirs();
			}
			
			stringEntity += "package " + pkgName + ";\r\n";
			stringEntity += "\r\n";
			stringEntity += "import org.springframework.web.bind.annotation.RequestBody;\r\n";
			stringEntity += "import org.springframework.web.bind.annotation.RequestMapping;\r\n";
			stringEntity += "import org.springframework.web.bind.annotation.RestController;\r\n";
			stringEntity += "\r\n";
			stringEntity += "import com.alibaba.fastjson.JSONObject;\r\n";
			stringEntity += "import com.jfinal.kit.Ret;\r\n";
			stringEntity += "import jf.kit.RetKit;\r\n";
			
			stringEntity += "\r\n";
			stringEntity += "@RestController\r\n";
			stringEntity += "@RequestMapping(\"" + requestMappingUrl + "\")\r\n";
			stringEntity += "public class " + className + " {\r\n";
			stringEntity += "\r\n";
			
			String[] methodsArr = methods.split(";");
			
			for (int i = 0; i < methodsArr.length; i++) {
				stringEntity += "    @RequestMapping(value=\"/" + methodsArr[i] + "\")\r\n";
				stringEntity += "    public Ret " + methodsArr[i] + "(@RequestBody JSONObject params) {\r\n";
				stringEntity += "\r\n";
				stringEntity += "        return RetKit.dataOk(\"\").set(\"flag\", true);\r\n";
				stringEntity += "    }\r\n\r\n";
			}
			
			stringEntity += "}";
			saveStringToTxt(stringEntity, destDir + "\\" + className + ".java");
		}
	}

	private static void saveStringToTxt(String string, String filePath) {
		try {
			FileOutputStream fos = new FileOutputStream(filePath);
			fos.write(string.getBytes());
			fos.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

}

更新列表:

*

参考文章:

[2]: [3]:

相关阅读