1 /*

     2  * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    12  * version 2 for more details (a copy is included in the LICENSE file that

    13  * accompanied this code).

    14  *

    15  * You should have received a copy of the GNU General Public License version

    16  * 2 along with this work; if not, write to the Free Software Foundation,

    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    18  *

    19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    20  * or visit www.oracle.com if you need additional information or have any

    21  * questions.

    22  */

    23 

    24 import java.nio.file.*;

    25 import java.nio.file.attribute.*;

    26 import java.io.IOException;

    27 import java.util.*;

    28 

    29 /**

    30  * Unit test for Files.walkFileTree to test maxDepth parameter

    31  */

    32 

    33 public class MaxDepth {

    34     public static void main(String[] args) throws Exception {

    35         final Path top = Paths.get(args[0]);

    36 

    37         for (int i=0; i<5; i++) {

    38             Set<FileVisitOption> opts = Collections.emptySet();

    39             final int maxDepth = i;

    40             Files.walkFileTree(top, opts, maxDepth, new SimpleFileVisitor<Path>() {

    41                 // compute depth based on relative path to top directory

    42                 private int depth(Path file) {

    43                     Path rp = file.relativize(top);

    44                     return (rp == null) ? 0 : rp.getNameCount();

    45                 }

    46 

    47                 @Override

    48                 public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) {

    49                     int d = depth(dir);

    50                     if (d == maxDepth)

    51                         throw new RuntimeException("Should not open directories at maxDepth");

    52                     if (d > maxDepth)

    53                         throw new RuntimeException("Too deep");

    54                     return FileVisitResult.CONTINUE;

    55                 }

    56 

    57                 @Override

    58                 public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {

    59                     int d = depth(file);

    60                     if (d > maxDepth)

    61                         throw new RuntimeException("Too deep");

    62                     return FileVisitResult.CONTINUE;

    63                 }

    64             });

    65         }

    66     }

    67 }