- 创建DownLoadUrlStream承载数据
import lombok.Data;
import java.io.InputStream;
@Data
public class DownLoadUrlStream {
private String url;
private InputStream stream;
}
- 创建工具类,进行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 {
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();
}
}
}
- 封装数据,调用方法即可