zip解压和压缩 - ToB企服应用市场:ToB评测及商务社交产业平台

import org.springframework.mock.web.MockMultipartFile; import org.springframework.web.multipart.MultipartFile; import java.io.*; import java.nio.charset.Charset; import java.util.*; import java.util.zip.ZipEntry; import java.util.zip.ZipInputStream; public class ZipUtil { // 解压并返回的key是全路径 public static Map<String, List<MultipartFile>> unzipAndConvertToMap(File file) throws IOException { Map<String, List<MultipartFile>> map = new HashMap<>(); // 存放解压后的图片 byte[] buffer = new byte[1024]; ZipInputStream zis = new ZipInputStream(new FileInputStream(file),Charset.forName("GBK")); try { ZipEntry entry; while ((entry = zis.getNextEntry()) != null) { if (!entry.isDirectory() && isImage(entry.getName())) { // 判断当前解压的是否是图片 // 获取到全路径名称 String allFileName = entry.getName(); String[] split = allFileName.split("/"); // // 图片名称 // String imageName = split[split.length - 1]; // // 第一层文件夹名称 // String firstFile = split[0]; StringBuffer sb = new StringBuffer(); for (int i = 1; i < split.length-1; i++) { sb.append(split[i]).append("/"); } String folderName = sb.toString(); List<MultipartFile> imgList = map.getOrDefault(folderName, new ArrayList<>()); ByteArrayOutputStream baos = new ByteArrayOutputStream(); int len; while ((len = zis.read(buffer)) > 0) { baos.write(buffer, 0, len); } MultipartFile multipartFile = new MockMultipartFile(entry.getName(), entry.getName(), "", baos.toByteArray()); imgList.add(multipartFile); map.put(folderName, imgList); } } } finally { zis.close(); } return map; } // 解压zip并取出文件夹中的所有image格式数据(仅为一层路径) public static Map<String, List<MultipartFile>> unzipAndExtractImages(File file) throws IOException { Map<String, List<MultipartFile>> map = new HashMap<>(); // 存放解压后的图片 byte[] buffer = new byte[1024]; ZipInputStream zis = new ZipInputStream(new FileInputStream(file),Charset.forName("GBK")); try { ZipEntry entry; while ((entry = zis.getNextEntry()) != null) { if (!entry.isDirectory() && isImage(entry.getName())) { // 判断当前解压的是否是图片 String folderName = new File(entry.getName()).getParentFile().getName(); // 获取图片所在的文件夹名称 List<MultipartFile> imgList = map.getOrDefault(folderName, new ArrayList<>()); ByteArrayOutputStream baos = new ByteArrayOutputStream(); int len; while ((len = zis.read(buffer)) > 0) { baos.write(buffer, 0, len); } MultipartFile multipartFile = new MockMultipartFile(entry.getName(), entry.getName(), "", baos.toByteArray()); imgList.add(multipartFile); map.put(folderName, imgList); } } } finally { zis.close(); } return map; } // 判断是否是图片格式 private static boolean isImage(String fileName) { // 判断当前解压中的文件是否是图片 String lowerCaseFileName = fileName.toLowerCase(); return lowerCaseFileName.endsWith(".jpg") || lowerCaseFileName.endsWith(".jpeg") || lowerCaseFileName.endsWith(".png"); } /** * MultipartFile 转 File * * @param file * @throws Exception */ public static File multipartFileToFile(MultipartFile file) throws Exception { File toFile = null; if (file.equals("") || file.getSize() <= 0) { file = null; } else { InputStream ins = null; ins = file.getInputStream(); toFile = new File(Objects.requireNonNull(file.getOriginalFilename())); inputStreamToFile(ins, toFile); ins.close(); } return toFile; } /** * 删除生成file时产生的临时文件 * @param file */ public static void delteTempFile(File file) { if (file != null) { File del = new File(file.toURI()); del.delete(); } } //获取流文件 private static void inputStreamToFile(InputStream ins, File file) { try { OutputStream os = new FileOutputStream(file); int bytesRead = 0; byte[] buffer = new byte[8192]; while ((bytesRead = ins.read(buffer, 0, 8192)) != -1) { os.write(buffer, 0, bytesRead); } os.close(); ins.close(); } catch (Exception e) { e.printStackTrace(); } }

复制代码