以前寫的一個文件夾遍歷代碼

很久前寫的了，用於查看文件夾下的所有文件大小和路徑，一般是在傳遞給別人一個打包文件的時候生成結構目錄用的。

生成的是一個xml文件。

 

其實說白了就是一個深度優先的回溯遍歷過程。

代碼如下：

 

traverse類：

 

package Folders; import java.io.File; import javax.xml.parsers.ParserConfigurationException; import org.w3c.dom.Element; public class traverse { private static WriteDao FolderDao; public static void main(String args[]){ traverse.init("out.xml",System.getProperty("user.dir")); } public static void init(String outputFileName,String strPath){ try { FolderDao=new WriteDao(outputFileName); File dir = new File(strPath); Element node = FolderDao.initXML("Dir"); getFileNames(node,dir); FolderDao.toSave(); } catch (ParserConfigurationException e) { // TODO Auto-generated catch block e.printStackTrace(); } } /* public static void getFileNames(Element node, File dir){ FolderDao.WriteTextNode(node,"dir", dir.getName()+" # "+dir.length()); File[] files = dir.listFiles(); if(files.length>0){ Element newNode = FolderDao.WriteNode(node,"dir"); for(File file : files){ if(file.isDirectory()){ getFileNames(newNode,file); }else{ FolderDao.WriteTextNode(newNode,"file", file.getName()+" # "+file.length()); } } } }*/ public static long getFileNames(Element node, File dir){ //System.out.println(dir.getName()); Element dirNode = FolderDao.WriteNode(node,"Dir"); String name; if((name=dir.getName()).equals("")){//針對根目錄做出的處理 name = dir.toString(); } FolderDao.AddAttribute(dirNode, "Name", name); File[] files = dir.listFiles(); if(files == null){ System.out.println(dir.toString()+"'s fileslist is NULL"); FolderDao.AddAttribute(dirNode, "Size", size(0)); return 0; } long length = 0; if(files.length>0){ for(File file : files){ //System.out.println(" "+file.toString()); if(file.isDirectory()){ length += getFileNames(dirNode,file); }else{ Element FileNode = FolderDao.WriteNode(dirNode,"File"); FolderDao.AddAttribute(FileNode, "Name", file.getName()); FolderDao.AddAttribute(FileNode, "Size", size(file.length())); length += file.length(); } } } FolderDao.AddAttribute(dirNode, "Size", size(length)); return length; } private static String size(long x){ int precision = 7; if ( x < 1000<<10 ){ return String.valueOf( ((x>>precision)<<precision)/1024.0)+"KB"; }else if ( x< 1000<<20 ){ return String.valueOf( ((x>>(10+precision))<<precision)/1024.0 )+"MB"; }else{ return String.valueOf( ((x>>(20+precision))<<precision)/1024.0 )+"GB"; } } }

 

WriteDao類，用於生成xml。

package Folders; import java.io.FileOutputStream; import java.io.OutputStreamWriter; import java.io.PrintWriter; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.parsers.ParserConfigurationException; import javax.xml.transform.OutputKeys; import javax.xml.transform.Transformer; import javax.xml.transform.TransformerFactory; import javax.xml.transform.dom.DOMSource; import javax.xml.transform.stream.StreamResult; import org.w3c.dom.Document; import org.w3c.dom.Element; public class WriteDao { private Document document; private String filename; public WriteDao(String name) throws ParserConfigurationException{ filename=name; DocumentBuilderFactory factory=DocumentBuilderFactory.newInstance(); DocumentBuilder builder=factory.newDocumentBuilder(); document=builder.newDocument(); } public Element initXML(String Ele){ Element root=document.createElement(Ele); document.appendChild(root); return root; } public Element WriteNode(Element node, String ClassName){ Element element=document.createElement(ClassName); node.appendChild(element); return element; } public Element WriteTextNode(Element node,String ContentName,String mycontent){ Element element=document.createElement(ContentName); element.appendChild(document.createTextNode(mycontent)); node.appendChild(element); return element; } public void AddAttribute(Element element,String Attribute,String Argu){ element.setAttribute(Attribute,Argu); } public void toSave(){ try{ TransformerFactory tf=TransformerFactory.newInstance(); Transformer transformer=tf.newTransformer(); DOMSource source=new DOMSource(document); transformer.setOutputProperty(OutputKeys.ENCODING,"UTF8"); transformer.setOutputProperty(OutputKeys.INDENT,"yes"); PrintWriter pw=new PrintWriter( new OutputStreamWriter(new FileOutputStream(filename), "UTF8")); StreamResult result=new StreamResult(pw); transformer.transform(source,result); }catch(Exception e){ e.printStackTrace(); } } }