經常用ibator工具生成VO和DAO，但是有個麻煩的就是沒有manager類，昨天有空手工寫了一個根據DAO和DAOImp生成Manager和ManagerImpl的工具，工具還可以把通過資料庫生成的xml檔案裡的DATE型別替換成TIMESTAMP，因為xml裡的DATE型別只有日期，沒分秒，Example裡的好多Date型別，都是java.sql.Date也是沒有分秒的。故全部替換成java.util.Date，這樣就省去不好功夫了。一鍵全部生成。

package test; import java.io.BufferedReader; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.io.IOException; /** * @Description: ibator補丁工具，生成Manager類，替換xml，java裡的Date資料型別 * @author: 尹濱 * @E-mail: yinbinhome@163.com * @date 2011-3-7 上午09:55:26 * @version 1.0 * */ public class IbatorConfig { /** * @param args */ public static void main(String[] args) { String path="D://workspace//portal//src//com//bfb//portal//vo//ab"; new IbatorConfig().start(path); } /** * 按目錄查詢檔案， * 1.換檔案中的指定字串 * 2.生成Manager介面 * 3.生成ManagerImpl類 */ public void start(String path){ File file=new File(path); File[] files = file.listFiles(); int i=0; int j=0; int m=0; int n=0; if(files!=null && files.length>0){ for(File f:files){ System.out.print("/n第 "+(++i)+" 個檔案"); String filename=f.getName(); if(filename.toLowerCase().indexOf("_sqlmap.xml")>0){ System.out.println("替換sqlMap.xml檔案開始......"); String fileALLname=f.getPath(); readFileByLines(fileALLname,"DATE","TIMESTAMP"); }else if(filename.toLowerCase().indexOf("example.java")>0){ System.out.println("替換Example.java檔案開始......"); String fileALLname=f.getPath(); readFileByLines(fileALLname,"java.sql.Date","Date"); }else if(filename.toLowerCase().indexOf("daoimpl.java")>0){ System.out.println("替換DAO.java檔案開始......"); String fileALLname=f.getPath(); chgDAOImpl(fileALLname); }else if(filename.toLowerCase().indexOf("dao.java")>0){ System.out.println("生成Manager.java介面檔案開始......"); addManager(f); System.out.println("生成ManagerImple.java介面檔案開始......"); File f1=new File(f.getPath()); newManagerImpl(f1); } } } } /** * 按行讀取，替換指定字串 */ public void readFileByLines(String fileName,String oldstring,String newstring) { File file = new File(fileName); BufferedReader reader = null; StringBuffer sb=new StringBuffer(""); try { reader = new BufferedReader(new FileReader(file)); String tempString = null; int line = 1; // 一次讀入一行，直到讀入null為檔案結束 while ((tempString = reader.readLine()) != null) { line++; String temp=tempString.replace(oldstring, newstring); sb.append(temp+"/n"); } reader.close(); } catch (IOException e) { e.printStackTrace(); } finally { if (reader != null) { try { reader.close(); } catch (IOException e1) { } } } //System.out.println(sb); appendMethodB(fileName,sb.toString()); } /** * 按給定內容，更新檔案內容 */ public void appendMethodB(String fileName, String content) { try { //開啟一個寫檔案器，建構函式中的第二個引數true表示以追加形式寫檔案 FileWriter writer = new FileWriter(fileName, false); writer.write(content); writer.close(); } catch (IOException e) { e.printStackTrace(); } } /** * 更新DAO為SqlMapClient增加setter getter * @param fileName */ public void chgDAOImpl(String fileName){ File file = new File(fileName); BufferedReader reader = null; StringBuffer sb=new StringBuffer(""); try { reader = new BufferedReader(new FileReader(file)); String tempString = null; int line = 1; // 一次讀入一行，直到讀入null為檔案結束 while ((tempString = reader.readLine()) != null) { line++; if(tempString.toLowerCase().indexOf("private sqlmapclient sqlmapclient;")>0){ sb.append(tempString+"/n/n"); sb.append(" public void setSqlMapClient(SqlMapClient sqlMapClient) {/n"+ " this.sqlMapClient = sqlMapClient;/n"+ " }/n/n"+ " public SqlMapClient getSqlMapClient() {/n"+ " return sqlMapClient;/n"+ " }/n"); }else{ sb.append(tempString+"/n"); } } reader.close(); } catch (IOException e) { e.printStackTrace(); } finally { if (reader != null) { try { reader.close(); } catch (IOException e1) { } } } //System.out.println(sb); appendMethodB(fileName,sb.toString()); } /** * 增加Manager介面 * @param fileIn */ public void addManager(File fileIn){ BufferedReader reader = null; StringBuffer sb=new StringBuffer(""); try { reader = new BufferedReader(new FileReader(fileIn)); String tempString = null; int line = 1; // 一次讀入一行，直到讀入null為檔案結束 while ((tempString = reader.readLine()) != null) { line++; if(tempString.indexOf("interface")>0 && tempString.indexOf("DAO")>0){ String temp=tempString.replace("DAO", "Manager"); sb.append(temp+"/n"); }else{ sb.append(tempString+"/n"); } } reader.close(); } catch (IOException e) { e.printStackTrace(); } finally { if (reader != null) { try { reader.close(); } catch (IOException e1) { } } } //System.out.println(sb); String name=fileIn.getName().replace("DAO", "Manager"); String newFilename=fileIn.getParent()+"//"+name; appendMethodB(newFilename,sb.toString()); } /** * 更改DAO的名字 * @param name * @return */ public String chgName(String name){ String firstC=name.substring(0, 1).toLowerCase(); String otherC=name.substring(1); return firstC+otherC; } /** * 第一個字母大寫 * @param word * @return */ private String firstWordToUpper(String word) { String first = word.substring(0, 1); return first.toUpperCase() + word.substring(1); } /** * setter getter * @param name */ public String getset(String name){ StringBuffer sb=new StringBuffer(""); sb.append(" public void set"+name+"("+name+" "+chgName(name)+"){/n"); sb.append(" this."+chgName(name)+"="+chgName(name)+";/n"); sb.append(" }/n"); sb.append(" public "+name+" get"+name+"("+name+" "+chgName(name)+"){/n"); sb.append(" return "+chgName(name)+";/n"); sb.append(" }/n"); return sb.toString(); } /** * 取方法名 * @param ss * @return */ public String getMethodName(String ss){ int s1=ss.indexOf("("); String temp=ss.substring(0,s1); int s2=temp.lastIndexOf(" "); return (ss.substring(s2,s1)); } /** * 得到變數名字的串 * @param line * @return */ public String getParam(String line){ int s1=line.indexOf("("); int s2=line.indexOf(")"); String params=line.substring(s1+1,s2); String p1=""; String p2=""; String paramStr=""; if(params.indexOf(",")>0){ int s3=params.indexOf(" "); int s4=params.indexOf(","); int s5=params.lastIndexOf(" "); p1=params.substring(s3,s4); p2=params.substring(s5); paramStr="("+p1+","+p2+")"; }else{ int s6=params.lastIndexOf(" "); p1=params.substring(s6); paramStr="("+p1+")"; } return paramStr; } /** * 更具介面方法宣告生成方法體 * @param line * @param daoName * @return */ public String getMethod(String line,String daoName){ String methodname=getMethodName(line); StringBuffer sb=new StringBuffer(""); if(line.indexOf("SQLException;")>0 && line.indexOf("import")<0){ if(line.indexOf("void")>0 && line.indexOf(";")>0){ String temp=line.replace(";", "{/n"); sb.append(" public "+temp); sb.append(" "+chgName(daoName)+"."+methodname+getParam(line)+";/n"); sb.append(" }/n/r"); }else{ String temp=line.replace(";", "{/n"); sb.append(" public "+temp); sb.append(" return "+chgName(daoName)+"."+methodname+getParam(line)+";/n"); sb.append(" }/n/r"); } } return sb.toString(); } /** * 傳入DAO介面的File，生成ManagerImpl * @param file */ public void newManagerImpl(File file){ String oldname=file.getName(); int s1=oldname.indexOf("."); oldname=oldname.substring(0,s1); String name=oldname.replace("DAO", "ManagerImpl"); String filename=file.getParent()+"//"+name+".java"; file.getName(); BufferedReader reader = null; StringBuffer sb=new StringBuffer(""); try { reader = new BufferedReader(new FileReader(file)); String tempString = null; int line = 1; // 一次讀入一行，直到讀入null為檔案結束 while ((tempString = reader.readLine()) != null) { line++; if(tempString.indexOf("interface")>0){ sb.append(""); String temp=tempString.replace("interface", "class"); temp=temp.replace(oldname, name); temp=temp.replace("{", ""); temp=temp+" implements "+oldname.replace("DAO", "Manager")+" {/n"; sb.append(temp); sb.append("/n"); sb.append(" private "+oldname+" "+chgName(oldname)+";/n/n"); sb.append(" public "+name+"() {/n"); sb.append(" super();/n"); sb.append(" }/n"); sb.append(getset(oldname)); }else if(tempString.indexOf("SQLException;")>0 && tempString.indexOf("import")<0){ String methodBody=getMethod(tempString,oldname); sb.append(methodBody); }else{ sb.append(tempString+"/n"); } } reader.close(); } catch (IOException e) { e.printStackTrace(); } finally { if (reader != null) { try { reader.close(); } catch (IOException e1) { } } } appendMethodB(filename,sb.toString()); } }