1. 程式人生 > >使用Java從分層目錄中提取所有檔名

使用Java從分層目錄中提取所有檔名

1.建立名為TestRecursiveDirectoryTraversal的主類

package testrecursivedirectorytraversal;

import java.io.File;
import java.util.HashSet;
import java.util.Set;

public class TestRecursiveDirectoryTraversal {
	public static void main(String[] args){
		System.out.println(listFiles(new File("Path")).size());//Path替換為你所要提取檔名的主目錄
	}
	
	public static Set<File> listFiles(File rootDir) {
	    Set<File> fileSet = new HashSet<File>();//Set用來儲存檔案資訊
	    if(rootDir == null || rootDir.listFiles()==null){
	        return fileSet;
	    }
	    for (File fileOrDir : rootDir.listFiles()) {
             if (fileOrDir.isFile()){
               fileSet.add(fileOrDir);
               System.out.println(fileOrDir.getPath());//輸出檔案路徑
             }
             else{
               fileSet.addAll(listFiles(fileOrDir));//遞迴地呼叫listFiles方法
             }
     }

	    return fileSet;
	}
}

2.建立資料夾,以及資料夾下包含的檔案(名稱、路徑自定義)

3.將Path替換為資料夾的根目錄