将网络上的多个资源进行打包下载,主要是为了记录如何将多个文件打包到一个接口里进行下载,也可修改为打包本地文件等。
代码:
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.net.URLEncoder;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
public class ResourceUtil {
private static final String DEFAULT_FILENAME = "resources";
private static final String UTF8 = "UTF-8";
/**
* 将网络上的多个资源进行打包下载
*
* @param resources 资源<资源名(包括文件后缀), 资源url>
* @param title 打包的压缩包名
* @param response 响应
*/
public static void packageResource(Map<String, String> resources, String title, HttpServletResponse response) {
if (isBlank(title)) {
title = DEFAULT_FILENAME;
}
//设置返回格式
response.reset();
response.setContentType("application/octet-stream");
try (ZipOutputStream zos = new ZipOutputStream(response.getOutputStream())) {
String fileName = URLEncoder.encode(title.concat(".zip"), UTF8);
response.setHeader("Content-Disposition", "attachment; filename=".concat(fileName));
for (Map.Entry<String, String> entry : resources.entrySet()) {
// 跳过没有完整信息的资源
if (isBlank(entry.getKey()) || isBlank(entry.getValue())) {
continue;
}
// 根据url获取资源
URL url = new URL(entry.getValue());
InputStream is = url.openStream();
BufferedInputStream bis = new BufferedInputStream(is);
zos.putNextEntry(new ZipEntry(entry.getKey()));
byte[] buffer = new byte[1024];
int r;
while ((r = bis.read(buffer)) != -1) {
zos.write(buffer, 0, r);
}
}
zos.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 判断字符串是否为空
* 从StringUtils中复制出来的isBlank方法
*
* @param cs 字符串
* @return true-空字符串/false-非空字符串
*/
private static boolean isBlank(CharSequence cs) {
int strLen;
if (cs != null && (strLen = cs.length()) != 0) {
for (int i = 0; i < strLen; ++i) {
if (!Character.isWhitespace(cs.charAt(i))) {
return false;
}
}
}
return true;
}
}