2
3 import lombok.extern.slf4j.Slf4j;
4 import org.apache.commons.io.FilenameUtils;
5
6 import java.io.*;
7 import java.nio.charset.Charset;
8 import java.util.zip.ZipEntry;
9 import java.util.zip.ZipInputStream;
10 import java.util.zip.ZipOutputStream;
11
12
13 @Slf4j
14 public class PfZipUtil {
15 public static final int BUFFER_SIZE = 2048;
16
17 private PfZipUtil() {
18 }
19
20 public static void zipFile(String toZipFilePath, String zipFilePath) throws IOException {
21 InputStream inputStream = null;
22 ZipOutputStream zipOutputStream = null;
23 File toZipFile = new File(toZipFilePath);
24 File zipFile = new File(zipFilePath);
25 try {
26 inputStream = new FileInputStream(toZipFile);
27 zipOutputStream = new ZipOutputStream(new FileOutputStream(zipFile));
28 zipOutputStream.putNextEntry(new ZipEntry(toZipFile.getName()));
29 int currentByte = -1;
30 log.debug(String.valueOf(currentByte));
31 byte[] data = new byte[BUFFER_SIZE];
32 while ((currentByte = inputStream.read(data, 0, BUFFER_SIZE)) != -1) {
33 zipOutputStream.write(data, 0, currentByte);
34 }
35 zipOutputStream.flush();
36 } finally {
37 SafeCloseUtil.safeClose(inputStream);
38 SafeCloseUtil.safeClose(zipOutputStream);
39 if (toZipFile.exists()) {
40 boolean isDeleted = toZipFile.delete();
41 if (!isDeleted) {
42 log.warn("delete file failed " + toZipFile.getAbsolutePath());
43 }
44 }
45 }
46 }
47
48 public static void zipFiles(File[] toZipFilePath, String zipFilePath) throws IOException {
49 InputStream inputStream = null;
50 zipFilePath = PfStringUtil.getLegalPath(zipFilePath);
51 File zipFile = new File(zipFilePath);
52 FileOutputStream outputStream = new FileOutputStream(zipFile);
53 ZipOutputStream zipOutputStream = new ZipOutputStream(outputStream);
54 try {
55 for (File file : toZipFilePath) {
56 inputStream = new FileInputStream(file.getAbsolutePath());
57 zipOutputStream.putNextEntry(new ZipEntry(file.getName()));
58 int currentByte = -1;
59 log.debug(String.valueOf(currentByte));
60 byte[] data = new byte[BUFFER_SIZE];
61 while ((currentByte = inputStream.read(data, 0, BUFFER_SIZE)) != -1) {
62 zipOutputStream.write(data, 0, currentByte);
63 }
64 zipOutputStream.closeEntry();
65 inputStream.close();
66 }
67 zipOutputStream.close();
68 } finally {
69 SafeCloseUtil.safeClose(inputStream);
70 SafeCloseUtil.safeClose(zipOutputStream);
71 for (File file : toZipFilePath) {
72 if (file.exists()) {
73 boolean isDeleted = file.delete();
74 if (!isDeleted) {
75 log.warn("delete file failed " + file.getAbsolutePath());
76 }
77 }
78 }
79 }
80 }
81
82 public static void createZip(String sourcePath, String zipPathNo) {
83 String zipPath = FileUtil.verifyNonNullString(zipPathNo);
84 if (zipPath == null) {
85 return;
86 }
87 FileOutputStream fos = null;
88 ZipOutputStream zos = null;
89 try {
90 fos = new FileOutputStream(zipPath);
91 zos = new ZipOutputStream(fos);
92 String sourcePath2 = FileUtil.verifyNonNullString(sourcePath);
93 if (sourcePath2 != null) {
94 writeZip(new File(sourcePath2), "", zos);
95 }
96 } catch (FileNotFoundException e) {
97 log.error("creat zipfile error", e);
98 } finally {
99 try {
100 if (zos != null) {
101 zos.close();
102 }
103 if (fos != null) {
104 fos.close();
105 }
106 } catch (IOException e) {
107 log.error("creat zipfile error", e);
108 }
109
110 }
111 }
112
113 private static void writeZip(File file, String parentPath, ZipOutputStream zos) {
114 if (file.exists()) {
115 if (file.isDirectory()) {// 处理文件夹
116 parentPath += file.getName() + File.separator;
117 File[] files = file.listFiles();
118 if (files != null && files.length != 0) {
119 for (File f : files) {
120 writeZip(f, parentPath, zos);
121 }
122 } else { // 空目录则创建当前目录
123 try {
124 zos.putNextEntry(new ZipEntry(parentPath));
125 } catch (IOException e) {
126 e.printStackTrace();
127 }
128 }
129 } else {
130 FileInputStream fis = null;
131 try {
132 fis = new FileInputStream(file);
133 ZipEntry ze = new ZipEntry(parentPath + file.getName());
134 zos.putNextEntry(ze);
135 byte[] content = new byte[1024];
136 int len = -1;
137 log.debug(String.valueOf(len));
138 while ((len = fis.read(content)) != -1) {
139 zos.write(content, 0, len);
140 zos.flush();
141 }
142
143 } catch (FileNotFoundException e) {
144 log.error("creat zipfile error", e);
145 } catch (IOException e) {
146 log.error("creat zipfile error", e);
147 } finally {
148 try {
149 if (fis != null) {
150 fis.close();
151 }
152 } catch (IOException e) {
153 log.error("creat zipfile error", e);
154 }
155 }
156 }
157 }
158 }
159
160 private static void createDirIfNotExists(String destPath, String entryName) {
161 String destPath2 = FileUtil.verifyNonNullString(destPath + File.separator + entryName);
162 if (destPath2 != null) {
163 File file = new File(destPath2);
164 if (!file.exists()) {
165 boolean success = file.mkdirs();
166 if (!success) {
167 log.info("mkdir error,path:" + file);
168 }
169 }
170 }
171 }
172
173 /**
174 * @param zipFileName
175 * @param destPath
176 * @return 解压文件的根目录
177 */
178 public static File unZip(String zipFileName, String destPath) {
179
180 zipFileName = FileUtil.verifyNonNullString(zipFileName);
181 destPath = FileUtil.verifyNonNullString(destPath);
182 if (zipFileName == null || zipFileName.length() <= 0) {
183 log.warn("zipFileName is empty!");
184 return null;
185 }
186 if (destPath == null || destPath.length() <= 0) {
187 log.warn("destPath is empty!");
188 return null;
189 }
190
191 FileInputStream fin = null;
192 ZipInputStream in = null;
193 FileOutputStream os = null;
194 try {
195 String zipFileName1 = FileUtil.verifyNonNullString(zipFileName);
196 fin = new FileInputStream(zipFileName1);
197 in = new ZipInputStream(fin);
198 // ZIP 文件条目
199 ZipEntry entry;
200
201 while ((entry = in.getNextEntry()) != null) {
202 String entryName = entry.getName();
203 // 判断是否为目录条目，目录条目定义为其名称以 '/'
204 // 结尾的条目。
205 entryName = FilenameUtils.normalize(entryName);
206 if (entry.isDirectory()) {
207 createDirIfNotExists(destPath, entryName);
208 } else {
209 // 如果压缩包存在路径conf/explain.xml，存在conf文件夹，但java读取不到conf文件夹，而是直接读取其中的文件。我们要先判断conf文件夹是否已经
210 // 创建，如果没有必须先建立conf，否则解压失败
211 String path = destPath + File.separator + entryName;
212 /**
213 * 下面两行代码是为了屏蔽掉路径中的"/"，统一为"\",
214 */
215 String path2 = FileUtil.verifyNonNullString(path);
216 if (path2 != null) {
217 File file = new File(path2);
218 String tempPath = file.getPath();
219
220 tempPath = tempPath.substring(0, tempPath.lastIndexOf(File.separator));
221 String tempPath2 = FileUtil.verifyNonNullString(tempPath);
222 if (tempPath2 != null) {
223 File tempFile = new File(tempPath2);
224 if (!tempFile.exists()) {
225 boolean success = tempFile.mkdirs();
226 if (!success) {
227 log.info("mkdir error,path:");
228 }
229 }
230 }
231 }
232 String destPath2 = FileUtil.verifyNonNullString(destPath + File.separator + entryName);
233 if (destPath2 != null) {
234 os = new FileOutputStream(destPath2);
235 // Transfer bytes from the ZIP
236 // file to the output file
237 byte[] buf = new byte[1024];
238 int len = 0;
239 log.debug(String.valueOf(len));
240 while ((len = in.read(buf)) > 0) {
241 os.write(buf, 0, len);
242 }
243 os.close();
244 in.closeEntry();
245 }
246 //fin.close();
247 }
248 }
249 return new File(destPath);
250 } catch (IOException e) {
251 log.error(e.getMessage(), e);
252 } finally {
253 closeFileResource(fin, in, os);
254 }
255 return null;
256 }
257
258 /**
259 * 计算ZIP文件的解压大小
260 *
261 * @param inputStream 输入流
262 * 注：输入流由调用方负责关闭
263 * @return 解压大小
264 */
265 public static long getUncompressedSize(InputStream inputStream) throws IOException {
266 ZipInputStream zipInputStream = new ZipInputStream(inputStream, Charset.forName("GBK"));
267 byte[] buffer = new byte[8 * 1024 * 1024];
268 long uncompressedSize = 0;
269 ZipEntry zipEntry;
270 while ((zipEntry = zipInputStream.getNextEntry()) != null) {
271 long size = zipEntry.getSize();
272 // ZipEntry的size可能为-1，表示未知
273 if (size == -1) {
274 int len;
275 while ((len = zipInputStream.read(buffer, 0, buffer.length)) != -1) {
276 uncompressedSize += len;
277 }
278 } else {
279 uncompressedSize += size;
280 }
281 zipInputStream.closeEntry();
282 }
283 return uncompressedSize;
284 }
285
286 private static void closeFileResource(FileInputStream fin, ZipInputStream in, FileOutputStream os) {
287 if (fin != null) {
288 try {
289 fin.close();
290 } catch (IOException e) {
291 log.error(e.getMessage(), e);
292 }
293
294 }
295 if (in != null) {
296 try {
297 in.close();
298 } catch (IOException e) {
299 log.error(e.getMessage(), e);
300 }
301
302 }
303 if (os != null) {
304 try {
305 os.close();
306 } catch (IOException e) {
307 log.error(e.getMessage(), e);
308 }
309
310 }
311 }
312
313 }
