1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
| package com.document.zip;
import lombok.extern.slf4j.Slf4j;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.tools.zip.ZipOutputStream;
import java.io.*;
import java.util.Enumeration;
import java.util.List;
@Slf4j
public class AntZipUtil {
private static int BUFFER_SIZE = 2 << 10;
public static void toZip(String srcDir, String zipPath, boolean isDelSrcFile) {
long start = System.currentTimeMillis();
ZipOutputStream zos = null;
File sourceFile = new File(srcDir);
File zipFile = new File(zipPath);
if(!sourceFile.exists()){
throw new RuntimeException("需压缩文件或者文件夹不存在");
}
try {
zos = new ZipOutputStream(zipFile);
compress(sourceFile, zos, sourceFile.getName());
if(isDelSrcFile){
sourceFile.delete();
}
log.info("原文件:{}. 压缩到:{}完成. 是否删除原文件:{}. 耗时:{}ms. ",srcDir,zipPath,isDelSrcFile, System.currentTimeMillis()-start);
} catch (IOException e) {
log.error("zip error from AntZipUtil: {}. ",e.getMessage());
throw new RuntimeException(e);
} finally {
try {
if (zos != null) {zos.finish();zos.close();}
} catch (Exception e) {}
}
}
private static void compress(File sourceFile, ZipOutputStream zos, String name) throws IOException {
if (sourceFile.isFile()) {
byte[] buf = new byte[BUFFER_SIZE];
zos.putNextEntry(new ZipEntry(sourceFile,name));
int len;
FileInputStream in = new FileInputStream(sourceFile);
while ((len = in.read(buf)) != -1) {
zos.write(buf, 0, len);
}
zos.closeEntry();
in.close();
} else {
File[] listFiles = sourceFile.listFiles();
if (listFiles == null || listFiles.length == 0) {
zos.putNextEntry(new ZipEntry(sourceFile,name + "/"));
zos.closeEntry();
} else {
for (File file : listFiles) {
compress(file, zos, name + "/" + file.getName());
}
}
}
}
public static void toZip(List<File> srcFiles , String zipPath) {
long start = System.currentTimeMillis();
File zipFile = new File(zipPath);
File parentFile = zipFile.getParentFile();
if(!parentFile.exists()) parentFile.mkdirs();
FileOutputStream fos = null;
ZipOutputStream zos = null;
try {
fos = new FileOutputStream(zipPath);
zos = new ZipOutputStream(new BufferedOutputStream(fos));
for (File srcFile : srcFiles) {
byte[] buf = new byte[BUFFER_SIZE];
zos.putNextEntry(new ZipEntry(srcFile.getName()));
int len;
FileInputStream in = new FileInputStream(srcFile);
while ((len = in.read(buf)) != -1){
zos.write(buf, 0, len);
}
in.close();
zos.closeEntry();
}
long end = System.currentTimeMillis();
log.info("压缩完成，耗时：" + (end - start) +" ms");
} catch (IOException e) {
throw new RuntimeException(e);
}finally{
try {
if(zos != null){
zos.finish();
zos.close();
}
if(fos != null) fos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
public static void unZip(File sourceFile, String saveFileDir) {
long start = System.currentTimeMillis();
if (!sourceFile.exists())
throw new RuntimeException("zip源文件不存在！");
File dir = new File(saveFileDir);
if(!dir.exists()){
dir.mkdirs();
}
File entryFile;
ZipFile zipFile = null;
InputStream inputStream = null;
try {
zipFile = new ZipFile(sourceFile);
for(Enumeration<ZipEntry> entries = zipFile.getEntries();
entries.hasMoreElements();){
ZipEntry zipEntry = entries.nextElement();
entryFile = new File(saveFileDir + "/" + zipEntry.getName());
if(zipEntry.isDirectory()){
entryFile.mkdirs();
} else {
File parent = entryFile.getParentFile();
if(parent!=null && !parent.exists()){
parent.mkdirs();
}
OutputStream os = new BufferedOutputStream(new FileOutputStream(
entryFile));
byte[] buffer = new byte[BUFFER_SIZE];
int len;
inputStream = zipFile.getInputStream(zipEntry);
while((len = inputStream.read(buffer)) != -1) {
os.write(buffer, 0, len);
}
inputStream.close();
os.close();
}
}
long end = System.currentTimeMillis();
log.info("解压完成，耗时：{} ms",end - start);
} catch (IOException e) {
throw new RuntimeException(e);
} finally {
try {
if(inputStream != null) inputStream.close();
if(zipFile != null) zipFile.close();
} catch (IOException e) {
}
}
}
}
|