若依框架单文件直接下载,多文件打包下载

一、前端

handleDownload(row){
  if (row.xz == 1) {
    this.$message.warning('禁止下载')
    return
  }
  this.download('/packDownload', {'id': id})
},

二、后台

/**
 * 打包下载
 * @param response
 * @param zbJszlQueryDTO
 */
@Override
public void packDownload(HttpServletResponse response, UserDTO userDTO) throws Exception {
    // 获取文件的保存位置
    List<String> paths = 文件路径
    if (paths.size() == 1) {  // 直接下载
        String filePath = paths.get(0);
        filePath = filePath.replace("/profile/", RuoYiConfig.getProfile() + "/");

        File file = new File(filePath);
        if (!file.exists()) {
            throw new Exception("文件不存在");
        }
        String suffix = filePath.substring(filePath.lastIndexOf("."));
        // 输出文件流
        writeFileToRes(response, "user" + "-" + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + suffix, file);
    } else {  // 压缩之后在进行下载
        //压缩包名称(会拼上当前时间)
        String datumName = "user";
        //压缩文件
        List<String> filePathList = paths;
        File file = compressedFileToZip(datumName, filePathList);
        //输出文件流
        writeFileToRes(response, file.getName(), file);
        //删除压缩包
        if(file.exists()){
            file.delete();
        }
    }
}
// 输出文件流到response
private void writeFileToRes(HttpServletResponse response, String fileName, File file) throws IOException {
    FileInputStream inputStream = new FileInputStream(file);
    //1.设置文件ContentType类型,这样设置,会自动判断下载文件类型.
    response.setContentType("application/octet-stream");
    //2.设置文件头:最后一个参数是设置下载文件名
    response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
    response.addHeader("Content-Length", "" + file.length());

    //3.通过response获取ServletOutputStream对象(out)
    ServletOutputStream out = response.getOutputStream();

    int b = 0;
    byte[] buffer = new byte[1024];
    while (b != -1) {
        b = inputStream.read(buffer);
        //4.写到输出流(out)中
        out.write(buffer, 0, b);
    }
    out.flush();
    out.close();
    inputStream.close();
}
// 压缩文件
private File compressedFileToZip(String datumName, List<String> filePathList) throws Exception {
    //压缩包具体名称(拼接时间戳防止重名)
    String zipFileName = datumName + "-" + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + ".zip";
    //生成压缩包存储地址(最后会删掉)
    String fileZip = RuoYiConfig.getProfile() + "/" + zipFileName;
    OutputStream os=null;
    ZipOutputStream zos = null ;
    File file = new File(fileZip);
    try {
        if (!file.getParentFile().exists()) {
            file.getParentFile().mkdirs();
        }
        os=new FileOutputStream(file);
        //压缩文件
        zos = new ZipOutputStream(os);
        byte[] buf = new byte[1024];
        for (String filePath : filePathList) {
            filePath = filePath.replace("/profile/", RuoYiConfig.getProfile() + "/");
            File tempFile = new File(filePath);
            //在压缩包中添加文件夹
            //zos.putNextEntry(new ZipEntry("测试/"+tempFile.getName()));
            //直接在压缩包中添加文件
            zos.putNextEntry(new ZipEntry(tempFile.getName()));
            int len;
            FileInputStream in = new FileInputStream(tempFile);
            while ((len = in.read(buf)) != -1){
                zos.write(buf, 0, len);
            }
            zos.closeEntry();
            in.close();
        }
    } catch (Exception e) {
        e.printStackTrace();
        System.out.println(e.toString());
        throw new Exception("文件打包:"+e.getMessage());
    }finally {
        //关闭流
        if(zos != null){
            try {
                zos.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        //关闭流
        if(os!= null){
            try {
                os.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    return file;
}
  • 7
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Monly21

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值