前端代码:
//1. 响应下载的输出流
fileDown(detail.caseId).then(response => {
const aLink =document.createElement('a')
const blob = new Blob([response], { type: 'application/zip' })
let fileName =`${detail.caseId}`
aLink.href = window.URL.createObjectURL(blob)
aLink.setAttribute('download', fileName)
document.body.appendChild(aLink)
aLink.click()
document.body.appendChild(aLink)
}).catch(error => this.$message.error(error))
//下载地址
export function fileDown(caseId) {
return request({
url: '/case/zipFileDown/filedown',
method: 'post',
params:{caseId:caseId},
responseType:'blob'
})
}
后台代码:
package com.ruoyi.lawcase.controller;
import cn.hutool.Hutool;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.ZipUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.lawcase.entity.LawFileEntity;
import com.ruoyi.lawcase.service.LawFileService;
import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.xml.transform.Result;
import java.io.*;
import java.net.*;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
@RestController
@RequestMapping("zipFileDown")
public class UrlFilesToZipController{
private static final Logger logger = LoggerFactory.getLogger(UrlFilesToZipController.class);
@Resource
private LawFileService lawFileService;
@Value("${file.path}")
private String FILE_ROOT_PATH;
@PostMapping("filedown")
public void downTogether(@RequestParam("caseId") String caseId) {
HttpServletResponse response = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getResponse();
HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
// 根据caseId 查询出文件URL地址
QueryWrapper<LawFileEntity> ew = new QueryWrapper<>();
ew.lambda().eq(LawFileEntity::getFileSendId,caseId);
List<LawFileEntity> list = lawFileService.list(ew);
for(int i=0; i<list.size(); i++){
//从服务器下载文件存储到本地
downloadFile(list.get(i).getFilePath(),FILE_ROOT_PATH+"/" + caseId,list.get(i).getName());
}
try {
//将下载的文件打成压缩包(ZipUtil工具类,可去官网查看参数的意思)
ZipUtil.zip(FILE_ROOT_PATH+"/"+caseId,FILE_ROOT_PATH+"/"+caseId+".zip");
//压缩文件下载,输出给前端
myDownLoad(FILE_ROOT_PATH,caseId+".zip",response,request);
//删除下载的临时文件夹
FileUtil.del(new File(FILE_ROOT_PATH));
} catch (Exception e) {
logger.error("打包文件接口异常",e);
}
}
// 从服务器下载文件到本地的方法
public void downloadFile(String url, String saveAddress,String fileName) {
//创建不同的文件夹目录
File file=new File(saveAddress);
//判断文件夹是否存在
if (!file.exists())
{
//如果文件夹不存在,则创建新的的文件夹
file.mkdirs();
}
FileOutputStream fileOut = null;
HttpURLConnection conn = null;
InputStream inputStream = null;
try
{
// 建立链接
URL httpUrl=new URL(url);
conn=(HttpURLConnection) httpUrl.openConnection();
//以Post方式提交表单,默认get方式
//conn.setRequestMethod("Post");
conn.setDoInput(true);
conn.setDoOutput(true);
// post方式不能使用缓存
conn.setUseCaches(false);
//连接指定的资源
conn.connect();
//获取网络输入流
inputStream=conn.getInputStream();
BufferedInputStream bis = new BufferedInputStream(inputStream);
//写入到文件(注意文件保存路径的后面一定要加上文件的名称)
fileOut = new FileOutputStream(saveAddress + "/" +fileName);
BufferedOutputStream bos = new BufferedOutputStream(fileOut);
byte[] buf = new byte[4096];
int length = bis.read(buf);
//保存文件
while(length != -1)
{
bos.write(buf, 0, length);
length = bis.read(buf);
}
bos.close();
bis.close();
conn.disconnect();
} catch (Exception e)
{
e.printStackTrace();
System.out.println("抛出异常!!");
}
}
// 把压缩文件响应给前端
public void myDownLoad(String filePath,String fileName, HttpServletResponse response, HttpServletRequest request){
String filePathF = filePath + "/" + fileName ;
File f = new File(filePathF);
if (!f.exists()) {
try {
response.sendError(404, "File not found!");
} catch (IOException e) {
e.printStackTrace();
}
return;
}
try {
InputStream fis = new BufferedInputStream(new FileInputStream(filePathF));
byte[] buffer = new byte[fis.available()];
int index = fis.read(buffer); // 不能删除,否则压缩包解压不开(能正常下载)
fis.close();
//清空response
response.reset();
//设置response的Header
response.setContentType("application/octet-stream;charset=UTF-8");
response.addHeader("Content-disposition", "attachment; filename=" + fileName);
response.addHeader("Content-Length",""+f.length());
OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
toClient.write(buffer);
toClient.flush();
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
}