业务需求,需要做一个导出用户单据图片的功能,需求为:按单据单号分文件夹分别存储单据的图片!最终导出zip包给用户。通过网上各种找代码,写出一个小demo,供各位看官学习!
// 假装是一个单据集合
List<String> imgUrls = new ArrayList<>();
imgUrls.add("http://img1.imgtn.bdimg.com/it/u=2171717409,2602158110&fm=26&gp=0.jpg");
imgUrls.add("http://images.ofweek.com/Upload/News/2016-5/Finn/53/20160503055505730.jpg");
imgUrls.add("http://y1.ifengimg.com/a/2016_04/4855d9fcb2ccade.jpg");
try {
//zip包的名称
String downloadFilename = "图片.zip";
downloadFilename = URLEncoder.encode(downloadFilename, "UTF-8");
// 指明response的返回对象是文件流
response.setContentType("application/octet-stream");
// 设置在下载框默认显示的文件名
response.setHeader("Content-Disposition", "attachment;filename=" + downloadFilename);
@Cleanup ZipOutputStream zos = new ZipOutputStream(response.getOutputStream());
imgUrls.forEach(i -> {
InputStream fis = null;
try {
// 跨平台建文件夹和文件名字
// File.separator 根据操作系统来判断用什么文件分隔符
// 这里只建立了一个文件夹,如果想分类的话,在后面加上想要的文件夹名称即可
ZipEntry zipEntry = new ZipEntry("img_download" + File.separator +imgUrls.indexOf(i)+ ".jpg");
// 放到压缩包里
zos.putNextEntry(zipEntry);
// 获得图片流
fis = getInputStreamByGet(i);
if (null != fis) {
byte[] buffer = new byte[1024];
int r = 0;
// 从输入流读取一定数量的字节,存到缓存区中
while ((r = fis.read(buffer)) != -1) {
zos.write(buffer, 0, r);
}
}
} catch (Exception e) {
e.printStackTrace();
}
});
zos.flush();
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
/**
* 通过get请求得到读取器响应数据的数据流
* @param url
* @return
*/
private static InputStream getInputStreamByGet(String url) {
try {
HttpURLConnection conn = (HttpURLConnection) new URL(url).openConnection();
conn.setReadTimeout(5000);
conn.setConnectTimeout(5000);
conn.setRequestMethod("GET");
if (conn.getResponseCode() == HttpURLConnection.HTTP_OK) {
return conn.getInputStream();
}
} catch (IOException e) {
e.printStackTrace();
}
return null;
}
写到这里就联想到用这个方法可以从网上爬一些照片到本地!