1 /*
2  * Copyright (c) 2017, 2018, 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 
25 
26 package jdk.tools.jaotc.collect.directory;
27 
28 import jdk.tools.jaotc.collect.ClassSource;
29 import jdk.tools.jaotc.collect.FileSystemFinder;
30 
31 import java.nio.file.Path;
32 import java.util.function.BiConsumer;
33 
34 public final class DirectorySource implements ClassSource {
35     private final Path directoryPath;
36     private final ClassLoader classLoader;
37 
DirectorySource(Path directoryPath, ClassLoader classLoader)38     DirectorySource(Path directoryPath, ClassLoader classLoader) {
39         this.directoryPath = directoryPath;
40         this.classLoader = classLoader;
41     }
42 
43     @Override
eachClass(BiConsumer<String, ClassLoader> consumer)44     public void eachClass(BiConsumer<String, ClassLoader> consumer) {
45         FileSystemFinder finder = new FileSystemFinder(directoryPath, ClassSource::pathIsClassFile);
46 
47         for (Path path : finder) {
48             consumer.accept(ClassSource.makeClassName(directoryPath.relativize(path).normalize()), classLoader);
49         }
50     }
51 
52     @Override
toString()53     public String toString() {
54         return "directory:" + directoryPath.toString();
55     }
56 }
57