51 lines
1.7 KiB
TypeScript
51 lines
1.7 KiB
TypeScript
import { ElNotification } from "element-plus";
|
||
|
||
/**
|
||
* @description 接收数据流生成blob，创建链接，下载文件
|
||
* @param {Function} api 导出表格的api方法(必传)
|
||
* @param {String} tempName 导出的文件名(必传)
|
||
* @param {Object} params 导出的参数(默认为空对象)
|
||
* @param {Boolean} isNotify 是否有导出消息提示(默认为 true)
|
||
* @param {String} fileType 导出的文件格式(默认为.xlsx)
|
||
* @return void
|
||
* */
|
||
export const useDownload = async (
|
||
api: (param: any) => Promise<any>,
|
||
tempName: string,
|
||
params: any = {},
|
||
isNotify: boolean = true
|
||
// fileType: string = ".xlsx"
|
||
) => {
|
||
if (isNotify) {
|
||
ElNotification({
|
||
title: "温馨提示",
|
||
message: "如果数据庞大会导致下载缓慢哦，请您耐心等待！",
|
||
type: "info",
|
||
duration: 3000
|
||
});
|
||
}
|
||
try {
|
||
const res = await api(params);
|
||
// const blob = new Blob([res], {
|
||
// type: "application/vnd.ms-excel;charset=UTF-8"
|
||
// });
|
||
const blob = new Blob([res]);
|
||
// 兼容 edge 不支持 createObjectURL 方法
|
||
// if ("msSaveOrOpenBlob" in navigator) return window.navigator.msSaveOrOpenBlob(blob, tempName + fileType);
|
||
if ("msSaveOrOpenBlob" in navigator) return window.navigator.msSaveOrOpenBlob(blob, tempName);
|
||
const blobUrl = window.URL.createObjectURL(blob);
|
||
const exportFile = document.createElement("a");
|
||
exportFile.style.display = "none";
|
||
// exportFile.download = `${tempName}${fileType}`;
|
||
exportFile.download = `${tempName}`;
|
||
exportFile.href = blobUrl;
|
||
document.body.appendChild(exportFile);
|
||
exportFile.click();
|
||
// 去除下载对 url 的影响
|
||
document.body.removeChild(exportFile);
|
||
window.URL.revokeObjectURL(blobUrl);
|
||
} catch (error) {
|
||
console.log(error);
|
||
}
|
||
};
|