5.6 接口开发-根据文件id打包下载附件
第2-1-2章 传统方式安装FastDFS-附FastDFS常用命令
第2-1-3章 docker-compose安装FastDFS,实现文件存储服务
第2-1-5章 docker安装MinIO实现文件存储服务-springboot整合minio-minio全网最全的资料
5.6.1 接口文档
根据文件id打包下载附件接口分两种情况进行下载:
1、如果客户端提交的文件id只有一个,则下载对应的原始文件
2、如果客户端提交的文件id有多个,则将对应的多个原始文件进行压缩,最终下载的是压缩后的文件
接口文档如下:
5.6.2 代码实现
第一步:在AttachmentController中提供根据文件id打包下载文件的方法
/**
* 下载一个文件或多个文件打包下载
*
* @param ids 文件id
* @param response
* @throws Exception
*/
@ApiOperation(value = "根据文件id打包下载", notes = "根据附件id下载多个打包的附件")
@GetMapping(value = "/download", produces = "application/octet-stream")
public void download(
@ApiParam(name = "ids[]", value = "文件id 数组")
@RequestParam(value = "ids[]") Long[] ids,
HttpServletRequest request, HttpServletResponse response) throws Exception {
BizAssert.isTrue(ArrayUtils.isNotEmpty(ids),
BASE_VALID_PARAM.build("附件id不能为空"));
//根据文件id下载文件
attachmentService.download(request, response, ids);
}
第二步:在AttachmentService接口中扩展download方法
/**
* 根据文件id下载附件
*
* @param request
* @param response
* @param ids
* @throws Exception
*/
void download(HttpServletRequest request,
HttpServletResponse response,
Long[] ids) throws Exception;
第三步:在AttachmentServiceImpl实现类中实现download方法
@Autowired
private FileBiz fileBiz;
/**
* 根据文件id下载文件
* @param request
* @param response
* @param ids
* @throws Exception
*/
@Override
public void download(HttpServletRequest request,
HttpServletResponse response,
Long[] ids) throws Exception {
//根据文件id查询数据库
List list =
(List) super.listByIds(Arrays.asList(ids));
down(request, response, list);
}
/**
* 文件下载
* @param request
* @param response
* @param list
* @throws Exception
*/
private void down(HttpServletRequest request, HttpServletResponse response,
List list) throws Exception {
if (list.isEmpty()) {
throw BizException.wrap("您下载的文件不存在");
}
List fileDOList =
list.stream().map((file) ->FileDO.builder()
.url(file.getUrl())
.submittedFileName(file.getSubmittedFileName())
.size(file.getSize())
.dataType(file.getDataType())
.build())
.collect(Collectors.toList());
fileBiz.down(fileDOList, request, response);
}
第四步:创建FileBiz,统一进行文件下载
package com.itheima.pinda.file.biz;
import cn.hutool.core.util.StrUtil;
import com.itheima.pinda.file.domain.FileDO;
import com.itheima.pinda.file.enumeration.DataType;
import com.itheima.pinda.file.utils.ZipUtils;
import com.itheima.pinda.utils.NumberHelper;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
/**
* 文件和附件的一些公共方法
*/
@Component
@Slf4j
public class FileBiz {
/**
* 构建新文件名称
* @param filename
* @param order
* @return
*/
private static String buildNewFileName(String filename, Integer order) {
return StrUtil.strBuilder(filename).
insert(filename.lastIndexOf("."), "(" + order + ")").toString();
}
/**
* 下载文件
* @param list
* @param request
* @param response
* @throws Exception
*/
public void down(List list,
HttpServletRequest request,
HttpServletResponse response) throws Exception {
int fileSize = list.stream().filter(
(file) -> file != null &&
!DataType.DIR.eq(file.getDataType()) &&
StringUtils.isNotEmpty(file.getUrl()))
.mapToInt(
(file) -> NumberHelper.intValueOf0(file.getSize())).sum();
String extName = list.get(0).getSubmittedFileName();
if (list.size() > 1) {
extName = StringUtils.substring(extName, 0,
StringUtils.lastIndexOf(extName, ".")) +
"等.zip";
}
Map map = new LinkedHashMap<>(list.size());
Map duplicateFile = new HashMap<>(list.size());
list.stream()
//过滤不符合要求的文件
.filter((file) -> file != null && !DataType.DIR.eq(file.getDataType()) && StringUtils.isNotEmpty(file.getUrl()))
//循环处理相同的文件名
.forEach((file) -> {
String submittedFileName = file.getSubmittedFileName();
if (map.containsKey(submittedFileName)) {
if (duplicateFile.containsKey(submittedFileName)) {
duplicateFile.put(submittedFileName, duplicateFile.get(submittedFileName) + 1);
} else {
duplicateFile.put(submittedFileName, 1);
}
submittedFileName = buildNewFileName(submittedFileName, duplicateFile.get(submittedFileName));
}
map.put(submittedFileName, file.getUrl());
});
ZipUtils.zipFilesByInputStream(map, Long.valueOf(fileSize), extName, request, response);
}
}
5.6.3 接口测试
第一步:启动Nacos配置中心
第二步:启动Nginx服务
第三步:启动文件服务
第四步:访问接口文档,地址为http://localhost:8765/doc.html
5.7 接口开发-根据业务类型/业务id打包下载
5.7.1 接口文档
根据业务类型/业务id打包下载文件接口分两种情况进行下载:
1、如果根据业务类型和业务id匹配到的文件只有一个,则下载对应的原始文件
2、如果根据业务类型和业务id匹配到的文件有多个,则将对应的多个原始文件进行压缩,最终下载的是压缩后的文件
接口文档如下:
5.7.2 代码实现
第一步:在AttachmentController中提供根据业务类型和业务id打包下载的方法
/**
* 根据业务类型或者业务id其中之一,或者2个同时打包下载文件
*
* @param bizIds 业务id
* @param bizTypes 业务类型
*
*/
@ApiImplicitParams({
@ApiImplicitParam(name = "bizIds[]", value = "业务id数组", dataType = "array", paramType = "query"),
@ApiImplicitParam(name = "bizTypes[]", value = "业务类型数组", dataType = "array", paramType = "query"),
})
@ApiOperation(value = "根据业务类型/业务id打包下载", notes = "根据业务id下载一个文件或多个文件打包下载")
@GetMapping(value = "/download/biz", produces = "application/octet-stream")
public void downloadByBiz(
@RequestParam(value = "bizIds[]", required = false) String[] bizIds,
@RequestParam(value = "bizTypes[]", required = false) String[] bizTypes,
HttpServletRequest request, HttpServletResponse response) throws Exception {
BizAssert.isTrue(!(ArrayUtils.isEmpty(bizTypes) && ArrayUtils.isEmpty(bizIds)), BASE_VALID_PARAM.build("附件业务id和业务类型不能同时为空"));
attachmentService.downloadByBiz(request, response, bizTypes, bizIds);
}
第二步:在AttachmentService接口中扩展downloadByBiz方法
/**
* 根据业务id和业务类型下载附件
*
* @param request
* @param response
* @param bizTypes
* @param bizIds
* @throws Exception
*/
void downloadByBiz(HttpServletRequest request, HttpServletResponse response,
String[] bizTypes, String[] bizIds) throws Exception;
第三步:在AttachmentServiceImpl实现类中实现downloadByBiz方法
/**
* 根据业务id和业务类型下载附件
*
* @param request
* @param response
* @param bizTypes
* @param bizIds
* @throws Exception
*/
@Override
public void downloadByBiz(HttpServletRequest request, HttpServletResponse response, String[] bizTypes, String[] bizIds) throws Exception {
List list = super.list(
Wraps.lbQ()
.in(Attachment::getBizType, bizTypes)
.in(Attachment::getBizId, bizIds));
down(request, response, list);
}
5.7.3 接口测试
第一步:启动Nacos配置中心
第二步:启动Nginx服务
第三步:启动文件服务
第四步:访问接口文档,地址为http://localhost:8765/doc.html
第2-1-2章 传统方式安装FastDFS-附FastDFS常用命令
第2-1-3章 docker-compose安装FastDFS,实现文件存储服务
第2-1-5章 docker安装MinIO实现文件存储服务-springboot整合minio-minio全网最全的资料