io.microsphere.net.classpath.Handler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of microsphere-java-core Show documentation
Microsphere Java Core
The newest version!
/*
 * 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 io.microsphere.net.classpath;

import io.microsphere.net.ExtendableProtocolURLStreamHandler;
import io.microsphere.util.ClassLoaderUtils;

import java.io.IOException;
import java.net.Proxy;
import java.net.URL;
import java.net.URLConnection;
import java.net.URLStreamHandler;

import static io.microsphere.constants.PathConstants.SLASH_CHAR;
import static io.microsphere.util.ClassLoaderUtils.getClassLoader;

/**
 * The "classpath" protocol {@link URLStreamHandler} based on {@link ClassLoader}
 *
 * @author Mercy
 * @see ClassLoader#getResource(String)
 * @see ClassLoaderUtils#getDefaultClassLoader()
 * @since 1.0.0
 */
public class Handler extends ExtendableProtocolURLStreamHandler {

    @Override
    protected URLConnection openConnection(URL u, Proxy proxy) throws IOException {
        String authority = u.getAuthority();
        String path = u.getPath();

        String classPath = path;

        if (authority != null && !authority.isEmpty()) {
            classPath = authority + path;
        }

        while (classPath.indexOf(SLASH_CHAR) == 0) {
            classPath = classPath.substring(1);
        }

        ClassLoader classLoader = getClassLoader(Handler.class);
        URL url = classLoader.getResource(classPath);
        if (url == null) {
            throw new IOException("No Resource[classpath='" + classPath + "'] was not found!");
        }
        return url.openConnection();
    }
}