1. 程式人生 > 其它 >post方法呼叫後臺,下載多個檔案打包成zip

post方法呼叫後臺,下載多個檔案打包成zip

前端
exportRecruitExcel: function () {
debugger
var id = vm.tableData.id;
//window.location.href = baseURL + 'sys/recruit/exportById?id='+id;
/*var exportData = [
{'id':id}
] //模擬後臺需要接收的引數*/
let form = $("<form>"); //建立form標籤

form.attr("style","display:none");
form.attr("method","post");//設定請求方式
form.attr("action",baseURL + 'sys/recruit/exportById'); //action屬性設定請求路徑
$("body").append(form); //頁面新增form標籤

let input1 = $("<input>") //建立input標籤
input1.attr("type","hidden") //設定隱藏域
input1.attr("name","data") //設定傳送後臺資料的引數名
//input1.attr("value",JSON.stringify({"id": id}));
input1.attr("value",id);
form.append(input1);

form.submit();//表單提交即可下載!
}
}
});

後臺
controller
@PostMapping("exportById")
public void exportById(@RequestParam(value = "data" ) String id,HttpServletResponse response) {
List<String> paths = recruitService.exportById(Long.parseLong(id));
ZipUtil.downloadMoreFile(response,paths);
}
ZipUtil
package com.allianity.common.utils;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
* @program: workweixin-mobile-api
* @ClassName GetExcelUtil
* @description:
* @author: Marlo
* @create: 2021-07-29 09:35
* @Version 1.0
**/
public class ZipUtil {
//多個檔案,壓縮成zip後下載
public static void downloadMoreFile(HttpServletResponse response, List<String> paths) {
List<File> files = new ArrayList<>();
boolean isExist = true;
for (String path : paths) {
File file = new File(path);
if (!file.exists()) {
isExist = false;
break;
}
files.add(file);
}
if (isExist) {
String zipTmp = "C:\\myselefzxq\\image\\userData.zip";
zipd(zipTmp, files, response);
}
}

public static void zipd(String zipTmp, List<File> files, HttpServletResponse response) {
File zipTmpFile = new File(zipTmp);
try {
if (zipTmpFile.exists()) {
zipTmpFile.delete();
}
zipTmpFile.createNewFile();

response.reset();
// 建立檔案輸出流
FileOutputStream fous = new FileOutputStream(zipTmpFile);
ZipOutputStream zipOut = new ZipOutputStream(fous);
zipFile(files, zipOut);
zipOut.close();
fous.close();
downloadZip(zipTmpFile, response);
} catch (IOException e) {
e.printStackTrace();
}
}


//files打成壓縮包
public static void zipFile(List files, ZipOutputStream outputStream) {
int size = files.size();
for (int i = 0; i < size; i++) {
File file = (File) files.get(i);
zipFile(file, outputStream);
}
}


public static void zipFile(File inputFile, ZipOutputStream ouputStream) {
try {
if (inputFile.exists()) {
if (inputFile.isFile()) {
FileInputStream IN = new FileInputStream(inputFile);
BufferedInputStream bins = new BufferedInputStream(IN, 512);
ZipEntry entry = new ZipEntry(inputFile.getName());
ouputStream.putNextEntry(entry);

int nNumber;
byte[] buffer = new byte[512];
while ((nNumber = bins.read(buffer)) != -1) {
ouputStream.write(buffer, 0, nNumber);
}

bins.close();
IN.close();
} else {
try {
File[] files = inputFile.listFiles();
for (int i = 0; i < files.length; i++) {
zipFile(files[i], ouputStream);
}
} catch (Exception e) {
e.printStackTrace();
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
}

public static HttpServletResponse downloadZip(File file, HttpServletResponse response) {
if (file.exists() == false) {
System.out.println("待壓縮的檔案目錄:" + file + "不存在.");
} else {
try {
// 以流的形式下載檔案。
InputStream fis = new BufferedInputStream(new FileInputStream(file.getPath()));
byte[] buffer = new byte[fis.available()];
fis.read(buffer);
fis.close();
// 清空response
response.reset();

OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
response.setContentType("application/octet-stream");

// 如果輸出的是中文名的檔案,在此處就要用URLEncoder.encode方法進行處理
response.setHeader("Content-Disposition",
"attachment;filename=" + new String(file.getName().getBytes("GB2312"), "ISO8859-1"));
toClient.write(buffer);
toClient.flush();
toClient.close();
} catch (Exception ex) {
ex.printStackTrace();
} finally {
try {
File f = new File(file.getPath());
f.delete();
} catch (Exception e) {
e.printStackTrace();
}
}
}
return response;
}


}