84 lines
1.7 KiB
Java
84 lines
1.7 KiB
Java
package com.imyeyu.io;
|
||
|
||
import lombok.Getter;
|
||
|
||
import java.io.File;
|
||
import java.io.FileNotFoundException;
|
||
import java.io.IOException;
|
||
import java.io.InputStream;
|
||
import java.util.Enumeration;
|
||
import java.util.HashMap;
|
||
import java.util.Map;
|
||
import java.util.jar.JarEntry;
|
||
import java.util.jar.JarFile;
|
||
|
||
/**
|
||
* Jar 文件读取
|
||
*
|
||
* @author 夜雨
|
||
* @version 2021-12-01 17:39
|
||
*/
|
||
@Getter
|
||
public class JarReader extends JarFile {
|
||
|
||
/** jar 所有文件的数据流映射列表，Map<路径, 数据流> */
|
||
private final Map<String, InputStream> files;
|
||
|
||
/**
|
||
* 默认构造
|
||
*
|
||
* @param file 文件
|
||
* @throws IOException 读取异常
|
||
*/
|
||
public JarReader(File file) throws IOException {
|
||
super(file);
|
||
|
||
files = new HashMap<>();
|
||
|
||
Enumeration<JarEntry> entries = entries();
|
||
while (entries.hasMoreElements()) {
|
||
JarEntry entry = entries.nextElement();
|
||
if (entry != null && !entry.isDirectory()) {
|
||
files.put(entry.getName(), getInputStream(entry));
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 该文件是否存在
|
||
*
|
||
* @param path 路径
|
||
* @return true 为存在
|
||
*/
|
||
public boolean has(String path) {
|
||
return files.containsKey(path);
|
||
}
|
||
|
||
/**
|
||
* 获取文件输入流
|
||
*
|
||
* @param path 路径
|
||
* @return 输入流
|
||
* @throws FileNotFoundException 文件不存在
|
||
*/
|
||
public InputStream getInputStream(String path) throws FileNotFoundException {
|
||
if (has(path)) {
|
||
return files.get(path);
|
||
} else {
|
||
throw new FileNotFoundException("Not found file: " + path);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 读取某文件为字节数据
|
||
*
|
||
* @param path 路径
|
||
* @return 字节数据
|
||
* @throws FileNotFoundException 文件不存在
|
||
*/
|
||
public byte[] getBytes(String path) throws IOException {
|
||
return IO.toBytes(getInputStream(path));
|
||
}
|
||
|
||
}
|