/**
* 批量下载压缩
* @param policyId
* @param response
* @return
*/
@GetMapping(value = "batchDownLoad")
public WebResponse batchDownLoad(Integer policyId, HttpServletResponse response) {
//下载的压缩包的名
String downloadFilename = UUID.randomUUID().toString().replace("-", "");
logger.info("下载的压缩包的名:" +downloadFilename);
ZipOutputStream zos = null;
InputStream in = null;
try {
response.setContentType("application/octet-stream");// 指明response的返回对象是文件流
response.setHeader("Content-Disposition", "attachment; filename=" +downloadFilename + ".zip");
zos = new ZipOutputStream(response.getOutputStream());
List<PolicyDeclare> policyDeclares = policyDeclareMapper.selectByPolicyId(policyId);
for (PolicyDeclare policyDeclare : policyDeclares) {
HashMap<String, InputStream> fileMap = new HashMap<>();
if (StringUtil.isNotEmpty(policyDeclare.getAnnex())) {
JSONArray jsonArray = JSON.parseArray(policyDeclare.getAnnex());
for (int i =0; i<jsonArray.size(); i++) {
URL url = new URL(jsonArray.getJSONObject(i).getString("url"));
HttpURLConnection conn = (HttpURLConnection)url.openConnection();
//设置超时间为3秒
conn.setConnectTimeout(3*1000);
//防止屏蔽程序抓取而返回403错误
conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
//得到输入流
InputStream inputStream = conn.getInputStream();
fileMap.put(jsonArray.getJSONObject(i).getString("name"), inputStream);
}
}
for (Map.Entry<String, InputStream> fileMaps : fileMap.entrySet()) {
String newFileName = fileMaps.getKey();
//打压缩包
//entry.getKey()就是上述所说的最外层的文件夹,也是自己定义的。
zos.putNextEntry(new ZipEntry(policyDeclare.getEntName() + "\\" + newFileName));
//文件路径放入流
in = fileMaps.getValue();
byte[] buffer = new byte[1024];
int r = 0;
while ((r = in.read(buffer)) != -1) {
zos.write(buffer, 0, r);
}
in.close();
zos.flush();
}
}
} catch (Exception e) {
// log.error("zipFiles error!!", e);
} finally {
if (zos!=null) {
try {
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
try {
zos.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
if (in!=null) {
try {
in.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
return null;
}
}