org.apache.dubbo.common.bytecode.CustomizedLoaderClassPath Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of dubbo Show documentation
The all in one project of dubbo
There is a newer version: 3.3.0-beta.2
/*
 * 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.dubbo.common.bytecode;

import javassist.ClassPath;

import java.io.InputStream;
import java.lang.ref.WeakReference;
import java.net.URL;

/**
 * A class search-path representing a class loader.
 *
 * It is used for obtaining a class file from the given
 * class loader by getResourceAsStream().
 * The LoaderClassPath refers to the class loader through
 * WeakReference.  If the class loader is garbage collected,
 * the other search pathes are examined.
 *
 * The given class loader must have both getResourceAsStream()
 * and getResource().
 *
 * @author Bill Burke
 * @author Shigeru Chiba
 */
public class CustomizedLoaderClassPath implements ClassPath {
    private WeakReference clref;

    /**
     * Creates a search path representing a class loader.
     */
    public CustomizedLoaderClassPath(ClassLoader cl) {
        clref = new WeakReference<>(cl);
    }

    public String toString() {
        ClassLoader cl = null;
        if (clref != null) {
            cl = clref.get();
        }

        return cl == null ? "" : cl.toString();
    }

    /**
     * Obtains a class file from the class loader.
     * This method calls getResourceAsStream(String)
     * on the class loader.
     */
    public InputStream openClassfile(String classname) {
        String cname = classname.replace('.', '/') + ".class";
        ClassLoader cl = clref.get();
        if (cl == null) {
            return null;        // not found
        } else {
            InputStream result = cl.getResourceAsStream(cname);
            if (result == null && (cl != this.getClass().getClassLoader())) {
                return this.getClass().getClassLoader().getResourceAsStream(cname);
            }
            return result;
        }
    }

    /**
     * Obtains the URL of the specified class file.
     * This method calls getResource(String)
     * on the class loader.
     *
     * @return null if the class file could not be found.
     */
    public URL find(String classname) {
        String cname = classname.replace('.', '/') + ".class";
        ClassLoader cl = clref.get();
        if (cl == null) {
            return null;        // not found
        } else {
            URL url = cl.getResource(cname);
            if (url == null && (cl != this.getClass().getClassLoader())) {
                return this.getClass().getClassLoader().getResource(cname);
            }
            return url;
        }
    }

    /**
     * Closes this class path.
     */
    public void close() {
        clref = null;
    }
}