Java获取目录下以及该目录下子目录所有的文件名

Java获取目录下以及该目录下子目录所有的文件名

package com.yangchen.test;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * Created by 杨晨 on 2020/7/22.
 */
public class GetFileName {

    public static void main(String[] args) {
        // 获取当前目录下所有的文件名，以及包括子目录下的文件名，该参数根据实际情况改动
        List<String> fileNames = getFileNames("E:\\testFile");
        try {
            // 将文件名写入到txt文件中
            writeTxtFile(fileNames);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     *
     * 获取目录下以及子目录的所有文件名
     */
    public static List<String> getFileNames(String path) {
        ArrayList<String> fileNames = new ArrayList<>();
        File file = new File(path);
        List<String> strings = getDirFileNames(fileNames, file);
        return strings;
    }

    /**
     *
     * 获取某个目录下所有直接下级文件
     */
    public static List<String> getDirFileNames(ArrayList<String> fileNames, File file) {
        File[] tempList = file.listFiles();

        for (int i = 0; i < tempList.length; i++) {
            // 判断是文件还是目录，是文件的话
            if (tempList[i].isFile()) {
                //文件名，不包含路径
                String fileName = tempList[i].getName();
                fileNames.add(fileName);
            }
            // 是目录的话，继续获取该目录下的文件
            if (tempList[i].isDirectory()) {
                //递归查询目录下文件
                getDirFileNames(fileNames, tempList[i]);
            }
        }
        return fileNames;
    }

    /**
     * 将目录以及子目录所有文件写入txt文件
     * @param fileNames
     */
    public static void writeTxtFile( List<String> fileNames) throws IOException {
        // 将文件名写入到D盘下Result.txt
        FileWriter fileWriter = new FileWriter("D:\\Result.txt");
        for (String fileName : fileNames) {
            // 获取最后一个“-”以前的内容
            int lastIndex = fileName.lastIndexOf("-");
            String fileShortName = null;
            if (lastIndex < 0) {
                fileShortName = fileName;
            } else {
                // 获取截取的文件名称
                fileShortName = fileName.substring(0, lastIndex);
            }
            fileWriter.write(fileName + "        " + fileShortName + "\n");
        }
        // 最后关闭下文件流
        fileWriter.flush();
        fileWriter.close();
    }
}