org.seedstack.maven.classloader.DisposableClassLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of seedstack-maven-plugin

The newest version!

/*
 * Copyright © 2013-2021, The SeedStack authors 
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.seedstack.maven.classloader;

import java.net.URL;
import java.net.URLClassLoader;
import java.util.HashSet;
import java.util.Set;

class DisposableClassLoader extends URLClassLoader {
    private final ReloadingClassLoader reloadingClassLoader;
    private final Set names = new HashSet<>();

    DisposableClassLoader(ReloadingClassLoader reloadingClassLoader, URL[] urLs) {
        super(urLs, null);
        this.reloadingClassLoader = reloadingClassLoader;
    }

    void addName(String name) {
        names.add(name);
    }

    @Override
    public Class loadClass(String name, boolean resolve) throws ClassNotFoundException {
        if (this.names.contains(name)) {
            // Only load the class specific to this classloader
            synchronized (getClassLoadingLock(name)) {
                // First, check if the class has already been loaded
                Class c = findLoadedClass(name);

                if (c == null) {
                    // If not, directly load it (no delegation to parent)
                    c = findClass(name);
                    if (resolve) {
                        resolveClass(c);
                    }
                }

                return c;
            }
        } else {
            // Otherwise delegate back to reloading classloader
            return reloadingClassLoader.loadClass(name, resolve);
        }
    }
}