com.gitee.starblues.loader.launcher.classpath.JarOutClasspathResource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-brick-loader Show documentation
加载插件模块
There is a newer version: 3.1.2
/**
 * Copyright [2019-Present] [starBlues]
 *
 * Licensed 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 com.gitee.starblues.loader.launcher.classpath;

import com.gitee.starblues.loader.archive.Archive;
import com.gitee.starblues.loader.archive.ExplodedArchive;
import com.gitee.starblues.loader.archive.JarFileArchive;
import com.gitee.starblues.loader.utils.FilesUtils;
import com.gitee.starblues.loader.utils.ObjectUtils;
import lombok.AllArgsConstructor;

import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.net.URL;
import java.util.*;
import java.util.jar.Manifest;

import static com.gitee.starblues.loader.LoaderConstant.*;

/**
 * jar out 类型的 classpath 获取者
 *
 * @author starBlues
 * @version 3.1.0
 * @since 3.0.4
 */
@AllArgsConstructor
public class JarOutClasspathResource implements ClasspathResource{


    private final static Archive.EntryFilter ENTRY_FILTER = (entry)->{
        String name = entry.getName();
        return name.startsWith(PROD_CLASSES_PATH);
    };

    private final static Archive.EntryFilter INCLUDE_FILTER = (entry) -> {
        if (entry.isDirectory()) {
            return entry.getName().equals(PROD_CLASSES_PATH);
        }
        return false;
    };

    private final File rootJarFile;


    @Override
    public List getClasspath() throws Exception {
        Archive archive = getArchive();
        Iterator archives = archive.getNestedArchives(ENTRY_FILTER, INCLUDE_FILTER);
        List urls = getEntryResource(archives);
        urls.addAll(getLibResource(archive));
        return urls;
    }

    private Archive getArchive() throws IOException {
        return (rootJarFile.isDirectory() ? new ExplodedArchive(rootJarFile) : new JarFileArchive(rootJarFile));
    }

    private List getEntryResource(Iterator archives) throws Exception{
        List urls =  new ArrayList<>();
        while (archives.hasNext()){
            Archive archive = archives.next();
            URL url = archive.getUrl();
            String path = url.getPath();
            if(path.contains(PROD_CLASSES_URL_SIGN)){
                urls.add(url);
            }
        }
        return urls;
    }

    private List getLibResource(Archive archive) throws Exception{
        Manifest manifest = archive.getManifest();
        String libDir = manifest.getMainAttributes().getValue(MAIN_LIB_DIR);
        String relativePath = rootJarFile.isDirectory() ? rootJarFile.getPath() : rootJarFile.getParent();
        libDir = FilesUtils.resolveRelativePath(relativePath, libDir);
        File libJarDir = new File(libDir);
        if(!libJarDir.exists()){
            throw new IllegalStateException("主程序依赖目录不存在: " + libDir);
        }
        List libIndexes = getLibIndexes(manifest);
        File[] libJarFile = getLibJarFile(libJarDir, libIndexes);
        List urls = new ArrayList<>(libJarFile.length);
        for (File file : libJarFile) {
            urls.add(file.toPath().toUri().toURL());
        }
        return urls;
    }

    private List getLibIndexes(Manifest manifest){
        String libIndexes = manifest.getMainAttributes().getValue(MAIN_LIB_INDEXES);
        if(ObjectUtils.isEmpty(libIndexes)){
            return Collections.emptyList();
        }
        String[] indexSplit = libIndexes.split(MAIN_LIB_INDEXES_SPLIT);
        List indexes = new ArrayList<>(indexSplit.length);
        for (String index : indexSplit) {
            if(ObjectUtils.isEmpty(index)){
                continue;
            }
            indexes.add(index);
        }
        if(indexes.isEmpty()){
            throw new IllegalStateException("主程序依赖包未发现!");
        }
        return indexes;
    }

    private File[] getLibJarFile(File rootFile, List libIndexes) {
        Set linIndexes = new HashSet<>(libIndexes);
        File[] listFiles = rootFile.listFiles(new FileFilter() {
            @Override
            public boolean accept(File pathname) {
                return linIndexes.contains(pathname.getName());
            }
        });
        if(listFiles == null || listFiles.length == 0){
            return new File[0];
        }
        return listFiles;
    }


}