一、需求:
今天碰到了一个需求,就是将COS对象存储的图片和后端的数据库查询出来的用户的信息的excel表格一起打包成一个压缩包提供给前端下载。
二、分析:
作者的思路是这样的: 从COS的存储地址拿到对应图片的流(这个可以参考腾讯云的COS对象的下载 腾讯云COS操作文档) 再拿到一个此用户excel表格的流,将这些流转换成byte[] 数组,供压缩类进行打包成同一个文件时进行二次使用,也就是将流转换成字节数组,供二次使用,压缩类再由此字节数组转换成输入流,再由压缩类的输出流,将此输入流一个一个的刷新到前端。
三、环境搭建:
- 引入处理excel表格的依赖、hutool工具包、cos对象存储的依赖(这里我就不一一介绍如何配置对象存储了,具体的可以参考上诉的腾讯云文档)
<properties>
<poi.version>4.1.0</poi.version>
<poi-ooxml.version>4.1.0</poi-ooxml.version>
<cos_api.version>5.6.15</cos_api.version>
<hutool-all.version>5.1.0</hutool-all.version>
</properties>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>${poi.version}</version>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>${poi-ooxml.version}</version>
</dependency>
<dependency>
<groupId>com.qcloud</groupId>
<artifactId>cos_api</artifactId>
<version>${cos_api.version}</version>
</dependency>
<dependency>
<groupId>cn.hutool</groupId>
<artifactId>hutool-all</artifactId>
<version>${hutool-all.version}</version>
</dependency>
四、功能实现:
- 通过cosClinet获取输入流(这里作者是使用了多个不同地区的存储桶):
/**
* 根据path 到相应地区的存储桶 获取图片的下载输入流和图片的名字
* @param path 数据库字段对应的图片url地址
* @return
*/
public InputStream downLoadResourceFromBucket(String path){
String[] split = path.split("\\.");
String[] split2 = path.split("/");
String bucketName = split[0].substring(8);
String area = split[2];
String key = "student-photo" + "/" +split2[split2.length - 1];
GetObjectRequest getObjectRequest = new GetObjectRequest(bucketName, key);
COSObject object = null;
if(area.equals(tenXunUtil.getAreaSH())){
object = cosClientSH.getObject(getObjectRequest);
}else if(area.equals(tenXunUtil.getAreaGZ())){
object = cosClientGZ.getObject(getObjectRequest);
}
COSObjectInputStream objectInputStream = object.getObjectContent();
return objectInputStream;
}
- 将图片和excel表格进行生成相应的字节数组存放到Map<String,byte[]>结构中,k:表示将要使用的文件名。
/**
* 对资源进行处理,并且处理成字节数组
* k:文件名 v:输入流转换成的字节数组.
* @return
*/
public Map<String,byte[]> addByteArrayToInputStreamMap(User user)