查詢指定目錄下文件並輸出到Excel

本文主要記錄一下平時用到的工具類
本類作用是查詢某個路徑下所有文件，獲取文件名輸出到Excel表格中

import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

public class FileTest {
    /**
     * 獲取路徑下的所有文件/文件夾
     * @param directoryPath 需要遍歷的文件夾路徑
     * @param isAddDirectory 是否將子文件夾的路徑也添加到list集合中
     * @return
     */
    public static List<String> getAllFile(String directoryPath,boolean isAddDirectory) {
        List<String> list = new ArrayList<>();
        File baseFile = new File(directoryPath);
        if (baseFile.isFile() || !baseFile.exists()) {
            return list;
        }
        File[] files = baseFile.listFiles();
        for (File file : files) {
            if (file.isDirectory()) {
                if(isAddDirectory){
                    list.add(file.getAbsolutePath());
                }
                list.addAll(getAllFile(file.getAbsolutePath(),isAddDirectory));
            } else {
//                list.add(file.getAbsolutePath());//保存文件所在路徑
                list.add(file.getName());//保存文件名
            }
        }
        return list;
    }
    /**
     * 針對路徑或文件名導出的操作
     *
     * @param list
     */
    public static void excleOut(List<String> list) {
        // 創建Excel文檔
        HSSFWorkbook hwb = new HSSFWorkbook();
        // 通過excle對象創建一個選項卡對象
        HSSFSheet sheet = hwb.createSheet("sheet1");
        String path = null;
        // 循環list創建行
        for (int i = 0; i < list.size(); i++) {
            // 新建一行
            HSSFRow row = sheet.createRow(i);
            path = list.get(i);
            // 設置i+1行第0列的數據
            row.createCell(0).setCellValue(path);
        }
        OutputStream out = null;
        try {
            out = new FileOutputStream("E:/demo.xls");
            hwb.write(out);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                out.close();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }
    public static void main(String[] args) {
        List<String> allFile = getAllFile("E:\\", false);
        List<String> list1 = new ArrayList<>();
        for (String s : allFile) {
            list1.add(s);
        }
        excleOut(list1);
        System.out.println(allFile.size());

    }
}