java 批量下载文件 打包成zip包

  1. 创建DownLoadUrlStream承载数据
import lombok.Data;
import java.io.InputStream;

@Data
public class DownLoadUrlStream {
    
    
    //文件地址 比如/123/234/1.txt
    private String url;
    //文件流
    private InputStream stream;

}
  1. 创建工具类,进行zip包压缩

import com.valid.util.view.DownLoadUrlStream;
import com.valid.util.view.DownLoadUrlString;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletResponse;

import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
@Component
public class DownloadUtils {
    
    
	/**
     * 
     * @param downLoadUrlStreams 文件信息 地址 流
     * @param zipName 压缩包名称
     * @param response
     * @throws IOException
     */
    public void downloadFolder(List<DownLoadUrlStream> downLoadUrlStreams,String zipName, HttpServletResponse response) throws IOException {
    
    

        response.setContentType("application/octet-stream");

        response.setHeader("Content-Disposition", "attachment; filename=\"" + URLEncoder.encode(zipName,"UTF-8")  + "\"");
        Set<String> paths = new HashSet<>(); // 用于记录已经添加到压缩文件中的目录

        try {
    
    
            ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream());
            for (DownLoadUrlStream file : downLoadUrlStreams) {
    
    
                String virtualPath = file.getUrl().substring(1);
                String[] pathArray = virtualPath.split("/");
                //构建文件的目录结构
                StringBuilder sb = new StringBuilder();
                for (int i = 0; i < pathArray.length - 1; i++) {
    
    
                    sb.append(pathArray[i]).append("/");
                    String folderPath = sb.toString();
                    if (!paths.contains(folderPath)) {
    
     // 如果该目录还未被添加到压缩文件中,则添加目录
                        ZipEntry folderEntry = new ZipEntry(folderPath);
                        zipOut.putNextEntry(folderEntry);
                        zipOut.closeEntry();
                        paths.add(folderPath); // 将新添加的目录记录到集合中
                    }
                }
                ZipEntry entry = new ZipEntry(virtualPath);
                zipOut.putNextEntry(entry);
                //将文件流写入文件中
                InputStream inputStream =  file.getStream();
                byte[] buffer = new byte[1024];
                int len;
                while ((len = inputStream.read(buffer)) > 0) {
    
    
                    zipOut.write(buffer, 0, len);
                }
                inputStream.close();
                zipOut.closeEntry();
            }
            zipOut.flush();
            zipOut.close();
        } catch (IOException e) {
    
    
            e.printStackTrace();
        }
    }
  }
   
  1. 封装数据,调用方法即可

猜你喜欢

转载自blog.csdn.net/qq_42900469/article/details/130982230