查询指定目录下文件并输出到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());

    }
}