学步园

abator补丁工具 V4.0

增加自动把dao，daImpl，service，serviceImpl，移动到指定目录

经常用ibator工具生成VO和DAO，但是有个麻烦的就是没有manager类，昨天有空手工写了一个根据DAO和DAOImp生成Manager和ManagerImpl的工具，工具还可以把通过数据库生成的xml文件里的DATE类型替换成TIMESTAMP，因为xml里的DATE类型只有日期，没分秒，Example里的好多Date类型，都是java.sql.Date也是没有分秒的。故全部替换成java.util.Date，这样就省去不好功夫了。一键全部生成。

package test;</p> <p>import java.io.BufferedReader;<br /> import java.io.File;<br /> import java.io.FileReader;<br /> import java.io.FileWriter;<br /> import java.io.IOException;<br /> import java.util.HashMap;<br /> import java.util.Map;</p> <p>/**<br /> * 增加自动写入application-Context.xml功能，把生成的dao与service对应的spring配置文件写入里面<br /> * @Description: ibator补丁工具，生成Manager类，替换xml，java里的Date数据类型<br /> * @author: 尹滨<br /> * @E-mail: yinbinhome@163.com<br /> * @date 2011-3-7 上午09:55:26<br /> * @version 2.0<br /> *<br /> */<br /> public class IbatorConfig {</p> <p> /**<br /> * @param args<br /> */<br /> public static void main(String[] args) {<br /> String path_dao="D://workspace//rcportalTEST//src//com//rc//portal//dao//impl";<br /> String path_sqlmap="D://workspace//rcportalTEST//src//com//rc//portal//vo//sqlmap";<br /> String path_vo="D://workspace//rcportalTEST//src//com//rc//portal//vo";<br /> String path_service="D://workspace//rcportalTEST//src//com//rc//portal//service//impl";<br /> String path_app_dao="D://workspace//rcportalTEST//conf_spring//applicationContext-dao.xml";<br /> String path_app_service="D://workspace//rcportalTEST//conf_spring//applicationContext-service.xml";</p> <p> new IbatorConfig().readyGO(path_dao,path_sqlmap,path_vo,path_service,path_app_dao,path_app_service); </p> <p> }</p> <p> public void readyGO(String daopath,String sqlpath,String vopath,String serivepath,String app_dao,String app_service){<br /> start(daopath,sqlpath,vopath,serivepath,app_dao,app_service);<br /> moveDAO(daopath);<br /> } </p> <p> public void start(String daopath,String sqlpath,String vopath,String serivepath,String app_dao,String app_service){<br /> System.out.println("处理dao......");<br /> startDo(daopath);<br /> System.out.println("处理sqlmap......");<br /> startDo(sqlpath);<br /> System.out.println("处理vo......");<br /> startDo(vopath);<br /> System.out.println("移动dao......");<br /> startDo2(daopath,serivepath,app_dao,app_service);<br /> }<br /> /**<br /> * 按目录查找文件，<br /> * 1.换文件中的指定字符串<br /> * 2.生成Manager接口<br /> * 3.生成ManagerImpl类<br /> */<br /> public void startDo(String path){<br /> File file=new File(path);<br /> File[] files = file.listFiles();<br /> int i=0;<br /> //System.out.println("文件路径："+path);<br /> if(files!=null && files.length>0){<br /> for(File f:files){<br /> //System.out.print("/n第 "+(++i)+" 个文件");<br /> String filename=f.getName();<br /> if(filename.toLowerCase().indexOf("_sqlmap.xml")>0){<br /> //System.out.println("替换sqlMap.xml文件开始......");<br /> String fileALLname=f.getPath();<br /> readFileByLines(fileALLname,"DATE","TIMESTAMP");<br /> }else if(filename.toLowerCase().indexOf("example.java")>0){<br /> //System.out.println("替换Example.java文件开始......");<br /> String fileALLname=f.getPath();<br /> readFileByLines(fileALLname,"java.sql.Date","Date");<br /> }else if(filename.toLowerCase().indexOf("daoimpl.java")>0){<br /> //System.out.println("替换DAO.java文件开始......");<br /> String fileALLname=f.getPath();<br /> chgDAOImpl(fileALLname);<br /> }else if(filename.toLowerCase().indexOf("dao.java")>0){<br /> //System.out.println("生成Manager.java接口文件开始......");<br /> addManager(f);<br /> //System.out.println("生成ManagerImple.java接口文件开始......");<br /> File f1=new File(f.getPath());<br /> newManagerImpl(f1);<br /> }<br /> }<br /> }<br /> } </p> <p> /**<br /> * 往application.xml里追加dao与service的bean<br /> * @param dao_path dao的文件夹<br /> * @param service_path sevice文件夹<br /> * @param app_dao_path application-dao.xml 全路径<br /> * @param app_service_path application-service.xml 全路径<br /> */<br /> public void startDo2(String dao_path,String service_path,String app_dao_path,String app_service_path){<br /> File file=new File(dao_path);<br /> File[] files = file.listFiles();<br /> int i=0;<br /> //System.out.println("文件路径："+dao_path);<br /> if(files!=null && files.length>0){<br /> for(File f:files){<br /> //System.out.print("/n第 "+(++i)+" 个文件");<br /> String filename=f.getName();<br /> if(filename.toLowerCase().indexOf("daoimpl.java")>0){<br /> //System.out.println("开始生成dao的bean......");<br /> String fileALLname=f.getPath();<br /> String daobean=getDaoBean(fileALLname);<br /> daobean+="/n";<br /> daobean+="</beans>";<br /> readFileByLines(app_dao_path,"</beans>",daobean);<br /> }<br /> }<br /> }<br /> File file2=new File(service_path);<br /> File[] files2 = file2.listFiles();<br /> int i2=0;<br /> //System.out.println("文件路径："+service_path);<br /> if(files2!=null && files2.length>0){<br /> for(File f:files2){<br /> //System.out.print("/n第 "+(++i2)+" 个文件");<br /> String filename=f.getName();<br /> if(filename.toLowerCase().indexOf("managerimpl.java")>0){<br /> //System.out.println("开始生成service的bean......");<br /> String fileALLname=f.getPath();<br /> String servicebean=getServiceBean(fileALLname);<br /> servicebean+="/n";<br /> servicebean+="</beans>";<br /> readFileByLines(app_service_path,"</beans>",servicebean);<br /> }<br /> }<br /> }<br /> }</p> <p> /**<br /> * 按行读取，替换指定字符串<br /> */<br /> public void readFileByLines(String fileName,String oldstring,String newstring) {<br /> File file = new File(fileName);<br /> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(file));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> String temp=tempString.replace(oldstring, newstring);<br /> sb.append(temp+"/n");</p> <p> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }<br /> ////System.out.println(sb);<br /> appendMethodB(fileName,sb.toString());<br /> }</p> <p> /**<br /> * 按给定内容，更新文件内容<br /> */<br /> public void appendMethodB(String fileName, String content) {<br /> try {<br /> //打开一个写文件器，构造函数中的第二个参数true表示以追加形式写文件<br /> FileWriter writer = new FileWriter(fileName, false);<br /> writer.write(content);<br /> writer.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> }<br /> }</p> <p> /**<br /> * 更新DAO为SqlMapClient增加setter getter<br /> * @param fileName<br /> */<br /> public void chgDAOImpl(String fileName){<br /> File file = new File(fileName);<br /> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(file));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> if(tempString.toLowerCase().indexOf("private sqlmapclient sqlmapclient;")>0){<br /> sb.append(tempString+"/n/n");<br /> sb.append(" public void setSqlMapClient(SqlMapClient sqlMapClient) {/n"+<br /> " this.sqlMapClient = sqlMapClient;/n"+<br /> " }/n/n"+<br /> " public SqlMapClient getSqlMapClient() {/n"+<br /> " return sqlMapClient;/n"+<br /> " }/n");<br /> }else{<br /> sb.append(tempString+"/n");<br /> }</p> <p> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }<br /> ////System.out.println(sb);<br /> appendMethodB(fileName,sb.toString());<br /> }</p> <p> /**<br /> * 增加Manager接口<br /> * @param fileIn<br /> */<br /> public void addManager(File fileIn){<br /> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(fileIn));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> if(tempString.indexOf("interface")>0 && tempString.indexOf("DAO")>0){<br /> String temp=tempString.replace("DAO", "Manager");<br /> sb.append(temp+"/n");<br /> }else if(tempString.indexOf("package")!=-1){<br /> String temp=tempString.replace("dao.impl", "service");<br /> sb.append(temp+"/n");<br /> }else{<br /> sb.append(tempString+"/n");<br /> }<br /> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }<br /> ////System.out.println(sb);<br /> String name=fileIn.getName().replace("DAO", "Manager");<br /> String newFilename=fileIn.getParent().replace("dao//impl", "service")+"//"+name;<br /> appendMethodB(newFilename,sb.toString());<br /> }</p> <p> /**<br /> * 更改DAO的名字<br /> * @param name<br /> * @return<br /> */<br /> public String chgName(String name){<br /> String firstC=name.substring(0, 1).toLowerCase();<br /> String otherC=name.substring(1);<br /> return firstC+otherC;<br /> }</p> <p> /**<br /> * 第一个字母大写<br /> * @param word<br /> * @return<br /> */<br /> private String firstWordToUpper(String word) {<br /> String first = word.substring(0, 1);<br /> return first.toUpperCase() + word.substring(1);<br /> }</p> <p> /**<br /> * setter getter<br /> * @param name<br /> */<br /> public String getset(String name){<br /> StringBuffer sb=new StringBuffer("");<br /> sb.append(" public void set"+name+"("+name+" "+chgName(name)+"){/n");<br /> sb.append(" this."+chgName(name)+"="+chgName(name)+";/n");<br /> sb.append(" }/n");</p> <p> sb.append(" public "+name+" get"+name+"("+name+" "+chgName(name)+"){/n");<br /> sb.append(" return "+chgName(name)+";/n");<br /> sb.append(" }/n");<br /> return sb.toString();<br /> }</p> <p> /**<br /> * 取方法名<br /> * @param ss<br /> * @return<br /> */<br /> public String getMethodName(String ss){<br /> int s1=ss.indexOf("(");<br /> String temp=ss.substring(0,s1);<br /> int s2=temp.lastIndexOf(" ");</p> <p> return (ss.substring(s2,s1));<br /> }</p> <p> /**<br /> * 得到变量名字的串<br /> * @param line<br /> * @return<br /> */<br /> public String getParam(String line){<br /> int s1=line.indexOf("(");<br /> int s2=line.indexOf(")");<br /> String params=line.substring(s1+1,s2);<br /> String p1="";<br /> String p2="";<br /> String paramStr="";<br /> if(params.indexOf(",")>0){<br /> int s3=params.indexOf(" ");<br /> int s4=params.indexOf(",");<br /> int s5=params.lastIndexOf(" ");<br /> p1=params.substring(s3,s4);<br /> p2=params.substring(s5);<br /> paramStr="("+p1+","+p2+")";<br /> }else{<br /> int s6=params.lastIndexOf(" ");<br /> p1=params.substring(s6);<br /> paramStr="("+p1+")";<br /> }<br /> return paramStr;<br /> }</p> <p> /**<br /> * 更具接口方法声明生成方法体<br /> * @param line<br /> * @param daoName<br /> * @return<br /> */<br /> public String getMethod(String line,String daoName){<br /> String methodname=getMethodName(line);<br /> StringBuffer sb=new StringBuffer("");<br /> if(line.indexOf("SQLException;")>0 && line.indexOf("import")<0){<br /> if(line.indexOf("void")>0 && line.indexOf(";")>0){<br /> String temp=line.replace(";", "{/n");<br /> sb.append(" public "+temp);<br /> sb.append(" "+chgName(daoName)+"."+methodname+getParam(line)+";/n");<br /> sb.append(" }/n/r");<br /> }else{<br /> String temp=line.replace(";", "{/n");<br /> sb.append(" public "+temp);<br /> sb.append(" return "+chgName(daoName)+"."+methodname+getParam(line)+";/n");<br /> sb.append(" }/n/r");<br /> }<br /> }<br /> return sb.toString();</p> <p> }</p> <p> /**<br /> * 传入DAO接口的File，生成ManagerImpl<br /> * @param file<br /> */<br /> public void newManagerImpl(File file){<br /> String oldname=file.getName();<br /> int s1=oldname.indexOf(".");<br /> oldname=oldname.substring(0,s1);<br /> String name=oldname.replace("DAO", "ManagerImpl");<br /> String filename=file.getParent().replace("dao", "service")+"//"+name+".java";<br /> file.getName();</p> <p> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(file));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> if(tempString.indexOf("interface")>0){<br /> sb.append("");<br /> String temp=tempString.replace("interface", "class");<br /> temp=temp.replace(oldname, name);<br /> temp=temp.replace("{", "");<br /> temp=temp+" implements "+oldname.replace("DAO", "Manager")+" {/n";<br /> sb.append(temp);</p> <p> sb.append("/n");<br /> sb.append(" private "+oldname+" "+chgName(oldname)+";/n/n");<br /> sb.append(" public "+name+"() {/n");<br /> sb.append(" super();/n");<br /> sb.append(" }/n");<br /> sb.append(getset(oldname));<br /> }else if(tempString.indexOf("SQLException;")>0 && tempString.indexOf("import")<0){<br /> String methodBody=getMethod(tempString,oldname);<br /> sb.append(methodBody);<br /> }else if(tempString.indexOf("package")!=-1){<br /> String tempIMP=tempString.replace("package", "import");<br /> tempIMP=tempIMP.replace("dao.impl", "dao");<br /> tempIMP=tempIMP.replace(";", ".")+oldname+";/n";</p> <p> String tempmanager=tempString.replace("package", "import");<br /> tempmanager=tempmanager.replace("dao", "service");<br /> String managername=name.substring(0,name.indexOf("Impl"));<br /> tempmanager=tempmanager.replace("impl", managername);</p> <p> String temp=tempString.replace("dao", "service");<br /> sb.append(temp+"/n/n");<br /> sb.append(tempIMP);<br /> sb.append(tempmanager);<br /> }else{<br /> sb.append(tempString+"/n");<br /> }<br /> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }<br /> appendMethodB(filename,sb.toString());<br /> }</p> <p> public void addSpringDao(String filename){</p> <p> }</p> <p> public String getDaoModel(){<br /> String daoM=<br /> " <bean id=/"#id#/" class=/"#classname#/">/n"+<br /> " <property name=/"sqlMapClient/" ref=/"sqlMapClient/" />/n"+<br /> " </bean>/n";<br /> return daoM;<br /> }</p> <p> public String getServiceModel(){<br /> String serviceM=<br /> " <bean id=/"#id#/" class=/"#classname#/">/n"+<br /> " <property name=/"#daoname#/" ref=/"#daoref#/" />/n"+<br /> " </bean>/n";<br /> return serviceM;<br /> }</p> <p> /**<br /> * 取类得全路径名,dao属性名<br /> * @param fileName<br /> * @param oldstring<br /> * @param newstring<br /> */<br /> public Map getDaoAllName(String fileName) {<br /> String daoname="";<br /> //String name="";<br /> Map<String,String> map=new HashMap<String,String>();<br /> File file = new File(fileName);<br /> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(file));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> if(tempString.indexOf("package")!=-1){<br /> daoname=tempString.substring(tempString.indexOf("package")+7,tempString.indexOf(";")).trim()+".";<br /> }else if(tempString.indexOf("public class")!=-1){<br /> daoname+=tempString.substring(tempString.indexOf("public class")+12,tempString.indexOf("implements")).trim();<br /> }</p> <p> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }<br /> String oldname=file.getName();<br /> int s1=oldname.indexOf(".");<br /> oldname=oldname.substring(0,s1);<br /> String daoAttrname=this.chgName(oldname);<br /> daoAttrname=daoAttrname.substring(0,daoAttrname.indexOf("Impl"));<br /> map.put("daoname", daoname);<br /> map.put("daoAttrname", daoAttrname);<br /> return map;<br /> }</p> <p> /**<br /> * 取service类全路径名和dao属性的名字<br /> * @param fileName<br /> * @return<br /> */<br /> public Map getServiceAllName(String fileName) {<br /> String servicename="";<br /> String daoname="";<br /> Map<String,String> map=new HashMap<String,String>();<br /> File file = new File(fileName);<br /> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(file));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> if(tempString.indexOf("package")!=-1){<br /> servicename=tempString.substring(tempString.indexOf("package")+7,tempString.indexOf(";")).trim()+".";<br /> }else if(tempString.indexOf("public class")!=-1){<br /> servicename+=tempString.substring(tempString.indexOf("public class")+12,tempString.indexOf("implements")).trim();<br /> }else if(tempString.indexOf("private")>0 && tempString.indexOf("DAO")>0){<br /> daoname=tempString.substring(tempString.indexOf("DAO ")+4,tempString.indexOf(";")).trim();<br /> }<br /> String temp=tempString;</p> <p> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }</p> <p> String oldname=file.getName();<br /> int s1=oldname.indexOf(".");<br /> oldname=oldname.substring(0,s1);<br /> String serviceAttrname=this.chgName(oldname);<br /> serviceAttrname=serviceAttrname.substring(0,serviceAttrname.indexOf("Impl"));<br /> map.put("servicename", servicename);<br /> map.put("daoname", daoname);<br /> map.put("serviceAttrname", serviceAttrname);<br /> return map;<br /> }</p> <p> /**<br /> * 生成dao bean<br /> * @param filename<br /> * @return<br /> */<br /> public String getDaoBean(String filename){<br /> Map<String,String> map=getDaoAllName(filename);<br /> String daoname=map.get("daoname");<br /> String daoAttrname=map.get("daoAttrname");<br /> String daobean=getDaoModel();<br /> daobean=daobean.replace("#classname#", daoname);<br /> daobean=daobean.replace("#id#", daoAttrname);<br /> return daobean;<br /> }</p> <p> /**<br /> * 生成service bean<br /> * @param filename<br /> * @return<br /> */<br /> public String getServiceBean(String filename){<br /> Map<String,String> map=getServiceAllName(filename);<br /> String servicename=map.get("servicename");<br /> String daoname=map.get("daoname");<br /> String serviceAttrname=map.get("serviceAttrname");<br /> String servicebean=getServiceModel();<br /> servicebean=servicebean.replace("#classname#", servicename);<br /> servicebean=servicebean.replace("#id#", serviceAttrname);<br /> servicebean=servicebean.replace("#daoname#", daoname);<br /> servicebean=servicebean.replace("#daoref#", daoname);<br /> return servicebean;<br /> }</p> <p> public void moveDAO(String path){</p> <p> File file=new File(path);<br /> File[] files = file.listFiles();<br /> int i=0;<br /> //System.out.println("文件路径："+path);<br /> if(files!=null && files.length>0){<br /> for(File f:files){</p> <p> String filename=f.getName();<br /> if(filename.toLowerCase().indexOf("daoimpl.java")>0){<br /> String newDaoname=filename.replace("Impl.java", ".java");<br /> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(f.getPath()));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> if(tempString.indexOf("package")!=-1){<br /> String temp=tempString.replace("package", "import");<br /> temp=temp.replace("dao.impl;", "dao.")+newDaoname.replace(".java", ";");<br /> sb.append(tempString+"/n/n");<br /> sb.append(temp);<br /> }else{<br /> sb.append(tempString+"/n");<br /> }<br /> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }</p> <p> appendMethodB(f.getPath(),sb.toString());</p> <p> }else if(filename.toLowerCase().indexOf("dao.java")>0){<br /> BufferedReader reader = null;<br /> StringBuffer sb=new StringBuffer("");<br /> try {<br /> reader = new BufferedReader(new FileReader(f.getPath()));<br /> String tempString = null;<br /> int line = 1;<br /> // 一次读入一行，直到读入null为文件结束<br /> while ((tempString = reader.readLine()) != null) {<br /> line++;<br /> if(tempString.indexOf("package")!=-1){<br /> tempString=tempString.replace("dao.impl", "dao");<br /> sb.append(tempString);<br /> }else{<br /> sb.append(tempString+"/n");<br /> }<br /> }<br /> reader.close();<br /> } catch (IOException e) {<br /> e.printStackTrace();<br /> } finally {<br /> if (reader != null) {<br /> try {<br /> reader.close();<br /> } catch (IOException e1) {<br /> }<br /> }<br /> }</p> <p> appendMethodB(f.getPath().replace("dao//impl", "dao"),sb.toString());<br /> f.delete();<br /> }<br /> }<br /> }<br /> }<br /> }