第2-3-6章打包批量下载附件的接口开发-文件存储服务系统-nginx/fastDFS/minio/阿里云oss/七牛云oss

假装文艺范儿 2023-04-18 原文

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<Attachment> list = 
        (List<Attachment>) 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<Attachment> list) throws Exception {
    if (list.isEmpty()) {
        throw BizException.wrap("您下载的文件不存在");
    }
    List<FileDO> 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<FileDO> 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<String, String> map = new LinkedHashMap<>(list.size());
        Map<String, Integer> 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<Attachment> list = super.list(
        Wraps.<Attachment>lbQ()
        .in(Attachment::getBizType, bizTypes)
        .in(Attachment::getBizId, bizIds));

    down(request, response, list);
}

5.7.3 接口测试