/*
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.runtime.module.artifact.classloader;
import static org.mule.runtime.api.util.Preconditions.checkArgument;
import org.mule.runtime.module.artifact.descriptor.ArtifactDescriptor;
/**
* Tracks {@link ArtifactClassLoader} created by {@link ArtifactClassLoaderFactory}
*/
public class TrackingArtifactClassLoaderFactory<T extends ArtifactDescriptor> implements ArtifactClassLoaderFactory<T> {
private final ArtifactClassLoaderManager artifactClassLoaderManager;
private final ArtifactClassLoaderFactory<T> artifactClassLoaderFactory;
/**
* Tracks the classloader created by another factory
*
* @param artifactClassLoaderManager tracks each created class loader. Non null.
* @param artifactClassLoaderFactory factory that creates the class loaders to be tracked. Non null.
*/
public TrackingArtifactClassLoaderFactory(ArtifactClassLoaderManager artifactClassLoaderManager,
ArtifactClassLoaderFactory<T> artifactClassLoaderFactory) {
checkArgument(artifactClassLoaderManager != null, "artifactClassLoaderManager cannot be null");
checkArgument(artifactClassLoaderFactory != null, "artifactClassLoaderFactory cannot be null");
this.artifactClassLoaderManager = artifactClassLoaderManager;
this.artifactClassLoaderFactory = artifactClassLoaderFactory;
}
@Override
public ArtifactClassLoader create(String artifactId, T descriptor, ClassLoader parent,
ClassLoaderLookupPolicy baseLookupPolicy) {
ArtifactClassLoader artifactClassLoader = artifactClassLoaderFactory.create(artifactId, descriptor, parent, baseLookupPolicy);
track(artifactClassLoader);
return artifactClassLoader;
}
private void track(ArtifactClassLoader artifactClassLoader) {
artifactClassLoaderManager.register(artifactClassLoader);
artifactClassLoader.addShutdownListener(() -> artifactClassLoaderManager.unregister(artifactClassLoader.getArtifactId()));
}
}