123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148 |
- /*
- * Copyright 2015 Decebal Suiu
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- package org.pf4j;
-
- import org.pf4j.processor.ServiceProviderExtensionStorage;
- import org.slf4j.Logger;
- import org.slf4j.LoggerFactory;
-
- import java.io.IOException;
- import java.io.Reader;
- import java.net.URISyntaxException;
- import java.net.URL;
- import java.nio.charset.StandardCharsets;
- import java.nio.file.FileSystem;
- import java.nio.file.FileSystems;
- import java.nio.file.FileVisitOption;
- import java.nio.file.FileVisitResult;
- import java.nio.file.Files;
- import java.nio.file.Path;
- import java.nio.file.Paths;
- import java.nio.file.SimpleFileVisitor;
- import java.nio.file.attribute.BasicFileAttributes;
- import java.util.Collections;
- import java.util.HashSet;
- import java.util.LinkedHashMap;
- import java.util.List;
- import java.util.Map;
- import java.util.Set;
-
- /**
- * The {@link java.util.ServiceLoader} base implementation for {@link ExtensionFinder}.
- * This class lookup extensions in all extensions index files {@code META-INF/services}.
- *
- * @author Decebal Suiu
- */
- public class ServiceProviderExtensionFinder extends AbstractExtensionFinder {
-
- private static final Logger log = LoggerFactory.getLogger(ServiceProviderExtensionFinder.class);
-
- public ServiceProviderExtensionFinder(PluginManager pluginManager) {
- super(pluginManager);
- }
-
- @Override
- public Map<String, Set<String>> readClasspathStorages() {
- log.debug("Reading extensions storages from classpath");
- Map<String, Set<String>> result = new LinkedHashMap<>();
-
- final Set<String> bucket = new HashSet<>();
- try {
- URL url = getClass().getClassLoader().getResource(getExtensionsResource());
- if (url != null) {
- Path extensionPath;
- if (url.toURI().getScheme().equals("jar")) {
- FileSystem fileSystem = FileSystems.newFileSystem(url.toURI(), Collections.<String, Object>emptyMap());
- extensionPath = fileSystem.getPath(getExtensionsResource());
- } else {
- extensionPath = Paths.get(url.toURI());
- }
-
- bucket.addAll(readExtensions(extensionPath));
- }
-
- debugExtensions(bucket);
-
- result.put(null, bucket);
- } catch (IOException | URISyntaxException e) {
- log.error(e.getMessage(), e);
- }
-
- return result;
- }
-
- @Override
- public Map<String, Set<String>> readPluginsStorages() {
- log.debug("Reading extensions storages from plugins");
- Map<String, Set<String>> result = new LinkedHashMap<>();
-
- List<PluginWrapper> plugins = pluginManager.getPlugins();
- for (PluginWrapper plugin : plugins) {
- String pluginId = plugin.getDescriptor().getPluginId();
- log.debug("Reading extensions storages for plugin '{}'", pluginId);
- final Set<String> bucket = new HashSet<>();
-
- try {
- URL url = ((PluginClassLoader) plugin.getPluginClassLoader()).findResource(getExtensionsResource());
- if (url != null) {
- Path extensionPath;
- if (url.toURI().getScheme().equals("jar")) {
- FileSystem fileSystem = FileSystems.newFileSystem(url.toURI(), Collections.<String, Object>emptyMap());
- extensionPath = fileSystem.getPath(getExtensionsResource());
- } else {
- extensionPath = Paths.get(url.toURI());
- }
-
- bucket.addAll(readExtensions(extensionPath));
- } else {
- log.debug("Cannot find '{}'", getExtensionsResource());
- }
-
- debugExtensions(bucket);
-
- result.put(pluginId, bucket);
- } catch (IOException | URISyntaxException e) {
- log.error(e.getMessage(), e);
- }
- }
-
- return result;
- }
-
- private static String getExtensionsResource() {
- return ServiceProviderExtensionStorage.EXTENSIONS_RESOURCE;
- }
-
- private Set<String> readExtensions(Path extensionPath) throws IOException {
- final Set<String> result = new HashSet<>();
- Files.walkFileTree(extensionPath, Collections.<FileVisitOption>emptySet(), 1, new SimpleFileVisitor<Path>() {
-
- @Override
- public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
- log.debug("Read '{}'", file);
- try (Reader reader = Files.newBufferedReader(file, StandardCharsets.UTF_8)) {
- ServiceProviderExtensionStorage.read(reader, result);
- }
-
- return FileVisitResult.CONTINUE;
- }
-
- });
-
- return result;
- }
-
- }
|