banner
Matt

Matt

Matt-xlog
telegram
github

【JAVA】【工具类】ランタイムでjarファイル内のファイルを読み取り、フォルダにコピーする

import lombok.SneakyThrows;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.util.AbstractMap;
import java.util.List;
import java.util.Map;
import java.util.jar.JarFile;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import java.util.zip.ZipEntry;
import java.net.URI;
/**
 * jarファイルの内容を読み取るためのユーティリティクラス
 */
public class readJarUtils {
    private static final Logger logger = LoggerFactory.getLogger(readJarUtils.class);
    public void test(){
        URI uri = null;
        try {
            uri = readJarUtils.class.getProtectionDomain().getCodeSource().getLocation().toURI();
        } catch (URISyntaxException e) {
            e.printStackTrace();
        }
        String tempPath = "/deployments";
        String sourceDir = "krb";  //リソースフォルダ
        if (uri.toString().startsWith("file")) {
            // IDEA実行時、リソースをコピーする
            copyLocalResourcesFileToTemp(sourceDir + "/", "*", tempPath + "/" + sourceDir);
        } else {
            // jarファイルのパスを取得
            String jarPath = uri.toString();
            uri = URI.create(jarPath.substring(jarPath.indexOf("file:"),jarPath.indexOf(".jar") + 4));
            // jarファイルにした後、リソースをコピーする
            readJarUtils.copyJarResourcesFileToTemp(uri, tempPath, "BOOT-INF/classes/" + sourceDir);
        }

    }
/**
     * 指定したディレクトリにローカルリソースファイルをコピーする
     * @param fileRoot      コピーするリソースディレクトリ
     * @param regExpStr     リソースファイルのマッチ正規表現、*はすべてにマッチ
     * @param tempParent    保存先
     */
    public static void copyLocalResourcesFileToTemp(String fileRoot, String regExpStr, String tempParent) {
        try {
            ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
            Resource[] resources = resolver.getResources(fileRoot + regExpStr);
            for (Resource resource : resources) {
                File newFile = new File(tempParent, resource.getFilename());
                if (newFile.exists()) {
                    newFile.delete();
                }
                InputStream stream = null;
                try {
                    stream = resource.getInputStream();
                } catch (Exception e) {
                    // resourceがフォルダの場合、例外が発生するため、この例外を無視する
                }
                if (stream == null) {
                    newFile.mkdirs();
                    copyLocalResourcesFileToTemp(fileRoot + resource.getFilename()  + "/", regExpStr, tempParent + "/" + resource.getFilename());
                } else {
                    if (!newFile.getParentFile().exists()) {
                        newFile.getParentFile().mkdirs();
                    }
                    org.apache.commons.io.FileUtils.copyInputStreamToFile(stream, newFile);
                }
            }
        } catch (Exception e) {
            logger.error("ローカルソーステンプレートのコピーに失敗しました", e);
        }
    }
/**
     * jarファイル内のリソースファイルを指定したディレクトリにコピーする
     * @param path          jarファイルのパス
     * @param tempPath      保存先ディレクトリ
     * @param filePrefix    コピーするリソースファイルのディレクトリ:BOOT-INF/classes/で始まる
     */
    public static void copyJarResourcesFileToTemp(URI path, String tempPath, String filePrefix) {
        try {
            List<Map.Entry<ZipEntry, InputStream>> collect =
                    readJarFile(new JarFile(path.getPath()), filePrefix).collect(Collectors.toList());
            for (Map.Entry<ZipEntry, InputStream> entry : collect) {
                // ファイルの相対パス
                String key = entry.getKey().getName();
                // ファイルストリーム
                InputStream stream = entry.getValue();
                File newFile = new File(tempPath + key.replaceAll("BOOT-INF/classes", ""));
                if (!newFile.getParentFile().exists()) {
                    newFile.getParentFile().mkdirs();
                }
                org.apache.commons.io.FileUtils.copyInputStreamToFile(stream, newFile);
            }
        } catch (IOException e) {
            logger.error("jarソーステンプレートのコピーに失敗しました", e);
        }
    }
 @SneakyThrows
    public static Stream<Map.Entry<ZipEntry, InputStream>> readJarFile(JarFile jarFile, String prefix) {
        Stream<Map.Entry<ZipEntry, InputStream>> readingStream =
                jarFile.stream().filter(entry -> !entry.isDirectory() && entry.getName().startsWith(prefix))
                        .map(entry -> {
                            try {
                                return new AbstractMap.SimpleEntry<>(entry, jarFile.getInputStream(entry));
                            } catch (IOException e) {
                                return new AbstractMap.SimpleEntry<>(entry, null);
                            }
                        });
        return readingStream.onClose(() -> {
            try {
                jarFile.close();
            } catch (IOException e) {
                logger.error("jarFileのクローズに失敗しました", e);
            }
        });
    }
}   
読み込み中...
文章は、創作者によって署名され、ブロックチェーンに安全に保存されています。