Github user ottobackwards commented on a diff in the pull request: https://github.com/apache/metron/pull/530#discussion_r131967849 --- Diff: bundles-lib/src/main/java/org/apache/metron/bundles/BundleClassLoaders.java --- @@ -0,0 +1,353 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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.apache.metron.bundles; + +import java.net.URISyntaxException; +import java.util.*; + +import org.apache.commons.collections.CollectionUtils; +import org.apache.commons.vfs2.FileObject; +import org.apache.commons.vfs2.FileSystemException; +import org.apache.commons.vfs2.FileSystemManager; +import org.apache.metron.bundles.bundle.Bundle; +import org.apache.metron.bundles.bundle.BundleCoordinate; +import org.apache.metron.bundles.bundle.BundleDetails; +import org.apache.metron.bundles.util.BundleProperties; +import org.apache.metron.bundles.util.BundleSelector; +import org.apache.metron.bundles.util.FileUtils; +import org.apache.metron.bundles.util.BundleUtil; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +/** + * A singleton class used to initialize the extension and framework classloaders. + */ +public final class BundleClassLoaders { + + private static volatile BundleClassLoaders bundleClassLoaders; + private volatile InitContext initContext; + private static final Logger logger = LoggerFactory.getLogger(BundleClassLoaders.class); + + private final static class InitContext { + + private final List<FileObject> extensionDirs; + private final Map<String, Bundle> bundles; + + private InitContext( + final List<FileObject> extensionDirs, + final Map<String, Bundle> bundles) { + this.extensionDirs = extensionDirs; + this.bundles = bundles; + } + } + + private BundleClassLoaders() { + } + + /** + * @return The singleton instance of the BundleClassLoaders + */ + public static BundleClassLoaders getInstance() { + BundleClassLoaders result = bundleClassLoaders; + if (result == null) { + synchronized (BundleClassLoaders.class) { + result = bundleClassLoaders; + if (result == null) { + bundleClassLoaders = result = new BundleClassLoaders(); + } + } + } + return result; + } + + public static void reset() { + getInstance().unInit(); + } + + private void unInit() { + synchronized (this) { + initContext = null; + } + } + + /** + * Initializes and loads the BundleClassLoaders. This method must be called before the rest of the + * methods to access the classloaders are called and it can be safely called any number of times + * provided the same framework and extension working dirs are used. + * + * @param fileSystemManager the FileSystemManager + * @param extensionsDirs where to find extension artifacts + * @param props BundleProperties + * @throws FileSystemException if any issue occurs while exploding bundle working directories. + * @throws java.lang.ClassNotFoundException if unable to load class definition + * @throws IllegalStateException already initialized with a given pair of directories cannot + * reinitialize or use a different pair of directories. + */ + public void init(final FileSystemManager fileSystemManager, final List<FileObject> extensionsDirs, + BundleProperties props) + throws FileSystemException, ClassNotFoundException, URISyntaxException { + if (extensionsDirs == null || fileSystemManager == null) { + throw new NullPointerException("cannot have empty arguments"); + } + InitContext ic = initContext; + if (ic == null) { + synchronized (this) { + ic = initContext; + if (ic == null) { + initContext = ic = load(fileSystemManager, extensionsDirs, props); + } + } + } else { + boolean matching = CollectionUtils + .isEqualCollection(initContext.extensionDirs, extensionsDirs); + if (!matching) { + throw new IllegalStateException( + "Cannot reinitialize and extension/framework directories cannot change"); + } + } + } + + /** + * Should be called at most once. + */ + private InitContext load(final FileSystemManager fileSystemManager, + final List<FileObject> extensionsDirs, BundleProperties props) + throws FileSystemException, ClassNotFoundException, URISyntaxException { + // get the system classloader + final ClassLoader systemClassLoader = ClassLoader.getSystemClassLoader(); + + // find all bundle files and create class loaders for them. + final Map<String, Bundle> directoryBundleLookup = new LinkedHashMap<>(); + final Map<String, ClassLoader> coordinateClassLoaderLookup = new HashMap<>(); + final Map<String, Set<BundleCoordinate>> idBundleLookup = new HashMap<>(); + + for (FileObject extensionsDir : extensionsDirs) { + // make sure the bundle directory is there and accessible + FileUtils.ensureDirectoryExistAndCanRead(extensionsDir); + + final List<FileObject> bundleDirContents = new ArrayList<>(); + FileObject[] dirFiles = extensionsDir.findFiles(new BundleSelector(props.getArchiveExtension())); + if (dirFiles != null) { + List<FileObject> fileList = Arrays.asList(dirFiles); + bundleDirContents.addAll(fileList); + } + + if (!bundleDirContents.isEmpty()) { + final List<BundleDetails> bundleDetails = new ArrayList<>(); + final Map<String, String> bundleCoordinatesToBundleFile = new HashMap<>(); + + // load the bundle details which includes bundle dependencies + for (final FileObject bundleFile : bundleDirContents) { + if(!bundleFile.exists() || !bundleFile.isFile()) { + continue; + } + BundleDetails bundleDetail = null; + try { + bundleDetail = getBundleDetails(bundleFile, props); + } catch (IllegalStateException e) { + logger.warn("Unable to load BUNDLE {} due to {}, skipping...", + new Object[]{bundleFile.getURL(), e.getMessage()}); + } + + // prevent the application from starting when there are two BUNDLEs with same group, id, and version + final String bundleCoordinate = bundleDetail.getCoordinate().getCoordinate(); + if (bundleCoordinatesToBundleFile.containsKey(bundleCoordinate)) { + final String existingBundleWorkingDir = bundleCoordinatesToBundleFile + .get(bundleCoordinate); + throw new IllegalStateException( + "Unable to load BUNDLE with coordinates " + bundleCoordinate + + " and working directory " + bundleDetail.getBundleFile() + + " because another BUNDLE with the same coordinates already exists at " + + existingBundleWorkingDir); + } + + bundleDetails.add(bundleDetail); + bundleCoordinatesToBundleFile.put(bundleCoordinate, + bundleDetail.getBundleFile().getURL().toURI().toString()); + } + + // attempt to locate the jetty bundle --- End diff -- I missed this. I have been getting away from nifi-isms as I go along.
--- If your project is set up for it, you can reply to this email and have your reply appear on GitHub as well. If your project does not have this feature enabled and wishes so, or if the feature is enabled but not working, please contact infrastructure at infrastruct...@apache.org or file a JIRA ticket with INFRA. ---