author | ohair |
Wed, 06 Apr 2011 22:06:11 -0700 | |
changeset 9035 | 1255eb81cc2f |
parent 8158 | 77d9c0f1c19f |
child 17170 | 9f33b89c7978 |
permissions | -rw-r--r-- |
6695
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
1 |
/* |
9035
1255eb81cc2f
7033660: Update copyright year to 2011 on any files changed in 2011
ohair
parents:
8158
diff
changeset
|
2 |
* Copyright (c) 2010, 2011, Oracle and/or its affiliates. All rights reserved. |
6695
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
4 |
* |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
5 |
* This code is free software; you can redistribute it and/or modify it |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
7 |
* published by the Free Software Foundation. |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
8 |
* |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
9 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
10 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
11 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
12 |
* version 2 for more details (a copy is included in the LICENSE file that |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
13 |
* accompanied this code). |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
14 |
* |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
15 |
* You should have received a copy of the GNU General Public License version |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
16 |
* 2 along with this work; if not, write to the Free Software Foundation, |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
17 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
18 |
* |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
19 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
20 |
* or visit www.oracle.com if you need additional information or have any |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
21 |
* questions. |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
22 |
*/ |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
23 |
|
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
24 |
import java.nio.file.*; |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
25 |
import java.nio.file.attribute.*; |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
26 |
import java.io.IOException; |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
27 |
import java.util.*; |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
28 |
|
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
29 |
/** |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
30 |
* Unit test for Files.walkFileTree to test maxDepth parameter |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
31 |
*/ |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
32 |
|
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
33 |
public class MaxDepth { |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
34 |
public static void main(String[] args) throws Exception { |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
35 |
final Path top = Paths.get(args[0]); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
36 |
|
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
37 |
for (int i=0; i<5; i++) { |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
38 |
Set<FileVisitOption> opts = Collections.emptySet(); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
39 |
final int maxDepth = i; |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
40 |
Files.walkFileTree(top, opts, maxDepth, new SimpleFileVisitor<Path>() { |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
41 |
// compute depth based on relative path to top directory |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
42 |
private int depth(Path file) { |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
43 |
Path rp = file.relativize(top); |
8158 | 44 |
return (rp.getFileName().toString().equals("")) ? 0 : rp.getNameCount(); |
6695
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
45 |
} |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
46 |
|
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
47 |
@Override |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
48 |
public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) { |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
49 |
int d = depth(dir); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
50 |
if (d == maxDepth) |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
51 |
throw new RuntimeException("Should not open directories at maxDepth"); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
52 |
if (d > maxDepth) |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
53 |
throw new RuntimeException("Too deep"); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
54 |
return FileVisitResult.CONTINUE; |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
55 |
} |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
56 |
|
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
57 |
@Override |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
58 |
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) { |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
59 |
int d = depth(file); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
60 |
if (d > maxDepth) |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
61 |
throw new RuntimeException("Too deep"); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
62 |
return FileVisitResult.CONTINUE; |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
63 |
} |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
64 |
}); |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
65 |
} |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
66 |
} |
0e91810597f9
6907737: (file) FileVisitor and Files.walkFileTree issues
alanb
parents:
diff
changeset
|
67 |
} |