java写的代码生成工具源码

package com.cxsoft.action; import java.io.BufferedReader; import java.io.BufferedWriter; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.io.IOException; import java.sql.Connection; import java.sql.SQLException; import java.util.ResourceBundle; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import com.cxsoft.base.TableForm; import com.cxsoft.common.DBConnection; import com.cxsoft.common.DBFactory; import com.cxsoft.tool.DBTool; import com.cxsoft.tool.FormatTool; import com.cxsoft.tool.StringUtil; /** * 文件生成器 * @author <a href= "mailto:chengzhiping@sinosoft.com.cn"> chengzhiping@sinosoft.com.cn</a> * 上午11:03:55 2012-5-15 */ /** * @author <a href= "mailto:chengzhiping@sinosoft.com.cn"> chengzhiping@sinosoft.com.cn</a> * 上午10:36:47 2012-5-20 */ public class FileGenerateAction { private static final Log log = LogFactory.getLog(FileGenerateAction.class); /** * 引擎入口 * @param tableForm * @throws Exception */ public static void engineEntry(TableForm tableForm) throws Exception{ String basePath = StringUtil.replaceSlash(tableForm.getBasePath()); tableForm.setBasePath(basePath); if(tableForm.isBeanBol()){ generateBean(tableForm); } if(tableForm.isServiceBol()){ generateService(tableForm); } if(tableForm.isJspBol()){ generateJsp(tableForm); } } /** * 生成Bean * @param tableForm * @throws Exception */ private static void generateBean(TableForm tableForm) throws Exception{ String writeFilePath = tableForm.getStorePath(); String tableName = tableForm.getSelecedTable(); String objectName = StringUtil.initialStrToUpper(tableName); objectName = StringUtil.convertUnderLine(objectName); String packageName = StringUtil.addPoint(tableForm.getBasePath(),"model;"); String fileName = StringUtil.addPoint(objectName,"java"); tableForm.setPackageName(packageName); packageName = StringUtil.convertPoint(packageName); tableForm.setSelecedTable(tableName); tableForm.setObjectName(objectName); tableForm.setFileName(fileName); writeFilePath +="/"+ packageName+"/"+fileName; tableForm.setWriteFilePath(writeFilePath); readAndWriteFile("../codeBuilder/template/Bean.temp",tableForm); } /** * 生成Service及其实现类 * @param tableForm * @throws Exception */ private static void generateService(TableForm tableForm) throws Exception{ String writeFilePath = tableForm.getStorePath(); String tableName = tableForm.getSelecedTable(); String objectName = StringUtil.initialStrToUpper(tableName); objectName = StringUtil.convertUnderLine(objectName); String packageName = StringUtil.addPoint(tableForm.getBasePath(),"service;");//tableForm.getBasePath()+".service;"; String fileName = "I"+objectName+"Service.java"; // String rootPath = tableForm.getBasePath(); tableForm.setPackageName(packageName); packageName = StringUtil.convertPoint(packageName); tableForm.setSelecedTable(tableName); tableForm.setObjectName(objectName); tableForm.setFileName(fileName); writeFilePath += "/"+packageName+"/"+fileName; tableForm.setWriteFilePath(writeFilePath); readAndWriteFile("../codeBuilder/template/Service.temp",tableForm); { String packageNameImple = StringUtil.addPoint(tableForm.getBasePath(),"service.imple;");//tableForm.getBasePath()+".service.imple;"; fileName = objectName + "ServiceImple.java"; tableForm.setPackageName(packageNameImple); packageNameImple = StringUtil.convertPoint(packageNameImple); tableForm.setSelecedTable(tableName); tableForm.setFileName(fileName); String writeFilePathImple = tableForm.getStorePath() +"/"+ packageNameImple+"/"+fileName; tableForm.setWriteFilePath(writeFilePathImple); readAndWriteFile("../codeBuilder/template/ServiceImple.temp",tableForm); } } /** * 生成Jsp页面 * @param tableForm * @throws Exception */ private static void generateJsp(TableForm tableForm) throws Exception{ String writeFilePath = tableForm.getStorePath(); String tableName = tableForm.getSelecedTable(); String objectName = StringUtil.initialStrToUpper(tableName); objectName = StringUtil.convertUnderLine(objectName); String basePath = tableForm.getStorePath() ; String fileName = "/"+objectName.toLowerCase()+"/"+"list.jsp"; tableForm.setSelecedTable(tableName); tableForm.setObjectName(objectName); tableForm.setFileName(fileName); writeFilePath = basePath+fileName; tableForm.setWriteFilePath(writeFilePath); readAndWriteFile("../codeBuilder/template/list.temp",tableForm); } /** * 根据不同模版生成所需文件 * @param readFilePath * @param tableForm * @throws Exception */ private static void readAndWriteFile(String readFilePath,TableForm tableForm) throws Exception{ try { FileReader fr = new FileReader(readFilePath);// 创建FileReader对象，用来读取字符流 BufferedReader br = new BufferedReader(fr); // 缓冲指定文件的输入 String writeFilePath = tableForm.getWriteFilePath(); if(createFile(writeFilePath)){ FileWriter fw = new FileWriter(writeFilePath);// 创建FileWriter对象，用来写入字符流 BufferedWriter bw = new BufferedWriter(fw); // 将缓冲对文件的输出 String myreadline; // 定义一个String类型的变量,用来每次读取一行 while (br.ready()) { myreadline = br.readLine();// 读取一行 if(tableForm.isBeanBol()){ if(!"".equals(myreadline)) myreadline = filteBeanFile(myreadline,tableForm); } if(tableForm.isServiceBol()){ if(!"".equals(myreadline)) myreadline = filteServiceFile(myreadline,tableForm); } if(tableForm.isJspBol()){ if(!"".equals(myreadline)) myreadline = filteJspFile(myreadline,tableForm); } bw.write(myreadline); // 写入文件 bw.newLine(); log.info(myreadline);// 在屏幕上输出 } bw.flush(); // 刷新该流的缓冲 bw.close(); fw.close(); } br.close(); fr.close(); } catch (IOException e) { e.printStackTrace(); throw new Exception(); } } /** * 过滤Service文件 * @param readLine * @param tableForm * @return */ private static String filteServiceFile(String readLine,TableForm tableForm){ return filterCommon(readLine,tableForm); } /** * 过滤Bean文件 * @param readLine * @param tableForm * @return * @throws Exception */ private static String filteBeanFile(String readLine,TableForm tableForm) throws Exception{ readLine = filterCommon(readLine,tableForm); Object columns[] = tableForm.getRightAllValues(); StringBuffer sb = new StringBuffer(); if(readLine.contains("@fieldDeclare")){ for (int i = 0; i < columns.length; i++) { String field = ""; String fieldDeclare = getField(columns[i].toString(),tableForm.getModel().getDBtype()); field += readLine.replaceAll("@fieldDeclare",fieldDeclare); sb.append(field); } return sb.toString(); }else if(readLine.contains("@method")){ for (int i = 0; i < columns.length; i++) { String columnType = columns[i].toString().split("-")[0]; String fieldType = translateToJavaType(tableForm.getModel().getDBtype(),columnType); String fieldName = columns[i].toString().split("-")[1].toLowerCase(); StringBuffer method = new StringBuffer(); method.append(readLine.replaceAll("@method",getMethod(tableForm,fieldType,fieldName)+setMethod(tableForm.getObjectName(),fieldType,fieldName))); sb.append(method); } return sb.toString(); }else{ return readLine; } } /** * 过滤Jsp文件 * @param readLine * @param tableForm * @return * @throws SQLException */ private static String filteJspFile(String readLine,TableForm tableForm) throws SQLException{ readLi