summaryrefslogtreecommitdiff
path: root/BJC-Utils2/src/main/java/bjc/utils/funcutils/FunctionalFileVisitor.java
blob: 6d3336a7943a4f753f9e068fa2b8bc6df4786d7e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package bjc.utils.funcutils;

import java.io.IOException;
import java.nio.file.FileVisitResult;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.function.BiPredicate;

final class FunctionalFileVisitor
		extends SimpleFileVisitor<Path> {
	private BiPredicate<Path, BasicFileAttributes>	traversalPredicate;
	private BiPredicate<Path, BasicFileAttributes>	traversalAction;

	public FunctionalFileVisitor(
			BiPredicate<Path, BasicFileAttributes> traversalPredicate,
			BiPredicate<Path, BasicFileAttributes> traversalAction) {
		this.traversalPredicate = traversalPredicate;
		this.traversalAction = traversalAction;
	}

	@Override
	public FileVisitResult preVisitDirectory(Path dir,
			BasicFileAttributes attrs) throws IOException {
		if (traversalPredicate.test(dir, attrs)) {
			return FileVisitResult.CONTINUE;
		}

		return FileVisitResult.SKIP_SUBTREE;
	}

	@Override
	public FileVisitResult visitFile(Path file,
			BasicFileAttributes attrs) throws IOException {
		if (traversalAction.test(file, attrs)) {
			return FileVisitResult.CONTINUE;
		}

		return FileVisitResult.TERMINATE;
	}
}