ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

add method

Browse Source
This commit is contained in:
申劭明 2021-10-15 01:37:16 +00:00 committed by Gitee
parent 6b3fc153d7
commit 612c2d8a98
1 changed files with 60 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
hutool-core/src/main/java/cn/hutool/core/util/ZipUtil.java
View File

@ -9,17 +9,14 @@ import cn.hutool.core.io.FastByteArrayOutputStream;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IORuntimeException;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.io.file.FileSystemUtil;
import cn.hutool.core.io.resource.Resource;
import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.*;
import java.nio.charset.Charset;
import java.nio.file.FileSystem;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
@ -75,6 +72,61 @@ public class ZipUtil {
}
}
/**
* 在zip文件中添加新文件, 如果已经存在则不会有效果
*
* @param zipFilePathStr zip文件存储路径
* @param appendFilePathStr 待添加文件路径(可以是文件夹)
*/
public static void addFile(String zipFilePathStr, String appendFilePathStr) throws IOException {
Path zipPath = Paths.get(zipFilePathStr);
Path appendFilePath = Paths.get(appendFilePathStr);
try (FileSystem zipFileSystem = FileSystemUtil.createZip(zipPath.toString())) {
Path root = zipFileSystem.getPath("/");
Path dest = zipFileSystem.getPath(root.toString(), appendFilePath.getFileName().toString());
if (!Files.isDirectory(appendFilePath)) {
Files.copy(appendFilePath, dest, StandardCopyOption.COPY_ATTRIBUTES);
} else {
Files.walkFileTree(appendFilePath, new SimpleFileVisitor<Path>() {
/**
* 用于保证文件夹拷贝后的效果跟常见压缩软件的效果相同
*/
private String dirRoot = null;
@Override
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
final Path dest;
if (dirRoot != null) {
dest = zipFileSystem.getPath(root.toString(), dirRoot, File.separator, StrUtil.subAfter(file.toString(), dirRoot, false));
} else {
dest = zipFileSystem.getPath(root.toString(), file.toString());
}
Files.copy(file, dest, StandardCopyOption.COPY_ATTRIBUTES);
return FileVisitResult.CONTINUE;
}
@Override
public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
final Path dirToCreate;
if (dirRoot == null) {
dirToCreate = zipFileSystem.getPath(root.toString(), dir.getFileName().toString());
dirRoot = dir.getFileName().toString();
} else {
dirToCreate = zipFileSystem.getPath(root.toString(), dirRoot, File.separator, StrUtil.subAfter(dir.toString(), dirRoot, false));
}
if (Files.notExists(dirToCreate)) {
Files.createDirectories(dirToCreate);
}
return FileVisitResult.CONTINUE;
}
});
}
} catch (FileAlreadyExistsException ignored) {
// 文件已存在, 跳过
}
}
/**
* 打包到当前目录，使用默认编码UTF-8
*