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);
}
});
}
}