使用nio对磁盘下的文件进行过滤

　 上篇博文讲到为了解决tomcat日志自动清理的问题，翻看了tomcat-juli这个jar包。在FileHandler类下有一个利用nio完成对磁盘下过期文件进行过滤的功能实现，正好这段时间正在学习nio框架，所以特意抽离源码做了一下测试。以下将提供测试代码(主要代码摘抄自tomcat-juli.jar源码)，与君共勉。

package nio;
 
import java.io.File;
import java.io.IOException;
import java.nio.file.DirectoryStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.regex.Pattern;
 
public class FilesFilter {
    private static int maxDays = 0;
    private static String directory = "E:\\Program Files\\Apache Software Foundation\\Tomcat 8.5.24\\logs";
    private static String prefix = "localhost.";
    private static String suffix = ".log";
    private static Pattern pattern = null;
 
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        pattern = Pattern
                .compile("^(" + Pattern.quote(prefix) + ")\\d{4}-\\d{1,2}-\\d{1,2}(" + Pattern.quote(suffix) + ")$");
 
        try {
            DirectoryStream<Path> files = streamFilesForDelete();
            System.out.println("准备输出:");
            for (Path file : files) {
                // Files.delete(file);
                System.out.println(file.toRealPath());
            }
        } catch (IOException e) {
            System.out.println("Unable to delete log files older than [" + maxDays + "] days");
        }
 
    }
    /**
     * 查询过期文件
     * @return
     * @throws IOException
     */
    private static DirectoryStream<Path> streamFilesForDelete() throws IOException {
        final Date maxDaysOffset = getMaxDaysOffset();
        final SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd");
        return Files.newDirectoryStream(new File(directory).toPath(), new DirectoryStream.Filter<Path>() {
            public boolean accept(Path path) throws IOException {
                boolean result = false;
                String date = obtainDateFromPath(path);
                if (date != null) {
                    try {
                        Date dateFromFile = formatter.parse(date);
                        result = dateFromFile.before(maxDaysOffset);
                    } catch (ParseException e) {
                        // no-op
                    }
                }
                return result;
            }
        });
    }
 
    private static String obtainDateFromPath(Path path) {
        Path fileName = path.getFileName();
        if (fileName == null) {
            return null;
        }
        String date = fileName.toString();
        if (pattern.matcher(date).matches()) {
            date = date.substring(prefix.length());
            return date.substring(0, date.length() - suffix.length());
        } else {
            return null;
        }
    }
 
    private static Date getMaxDaysOffset() {
        Calendar cal = Calendar.getInstance();
        cal.set(Calendar.HOUR_OF_DAY, 0);
        cal.set(Calendar.MINUTE, 0);
        cal.set(Calendar.SECOND, 0);
        cal.set(Calendar.MILLISECOND, 0);
        cal.add(Calendar.DATE, -maxDays);
        return cal.getTime();
    }
}

　　代码重点只有一个： java.nio.file.Files.newDirectoryStream(Path dir,DirectoryStream.Filter<? super Path> filter)，nio的Files类下newDirectoryStream方法用于获取指定磁盘下的物理文件列表Stream，该方法第二个参数可接收一个用于过滤的功能函数。实现accept方法，针对自己的场景自定义过滤条件。个人理解这种方式，可以在一次磁盘扫描中，完成无效文件过滤与符合条件文件的获取操作。比起传统的io对文件逐个扫描的操作，可以减少磁盘读写开销，有助于软件性能提升。当然，没有做过试验，在我这里还不算是一个结论。有这方面性能验证的朋友，可以给我留言解惑，谢谢！