xxl.mathematica.io.FileNames Maven / Gradle / Ivy

Go to download
package xxl.mathematica.io;

import io.vavr.control.Try;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.filefilter.RegexFileFilter;
import org.apache.commons.io.filefilter.TrueFileFilter;
import xxl.mathematica.functional.Map;

import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;

/**
 * 文件名称
 */
public class FileNames {
    /**
     * 查找文件
     *
     * @param regex
     * @param dir
     * @param recursive 是否递归
     * @return
     */
    public static List fileNames(String regex, String dir, boolean recursive) {
        return Try.ofCallable(() -> {
            File directory = new File(dir);
            if (directory.isDirectory()) {
                Collection files = FileUtils.listFiles(directory, new RegexFileFilter(regex), recursive ? TrueFileFilter.TRUE : null);
                return Map.map(File::getAbsolutePath, new ArrayList<>(files));
            } else {
                return Arrays.asList(directory.getAbsolutePath());
            }
        }).getOrNull();

    }

    /**
     * 默认递归
     *
     * @param regex
     * @param dir
     * @return
     */
    public static List fileNames(String regex, String dir) {
        return fileNames(regex, dir, true);
    }

}