设计管理优化
This commit is contained in:
63
src/utils/useFileDownload.js
Normal file
63
src/utils/useFileDownload.js
Normal file
@ -0,0 +1,63 @@
|
||||
import { ElMessage } from 'element-plus';
|
||||
|
||||
/**
|
||||
* 下载文件下载工具函数
|
||||
* @param {string} fileUrl - 文件地址
|
||||
* @param {string} [fileName] - 可选,指定文件名
|
||||
* @param {Object} [headers] - 可选,请求头(如携带token)
|
||||
*/
|
||||
export const downloadFile = async (fileUrl, fileName, headers = {}) => {
|
||||
try {
|
||||
// 发起请求获取文件数据
|
||||
const response = await fetch(fileUrl, {
|
||||
method: 'GET',
|
||||
headers: {
|
||||
'Content-Type': 'application/octet-stream',
|
||||
...headers
|
||||
},
|
||||
credentials: 'include' // 允许携带cookie
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`下载失败: ${response.statusText}`);
|
||||
}
|
||||
|
||||
// 将响应转换为blob对象
|
||||
const blob = await response.blob();
|
||||
|
||||
// 创建临时URL
|
||||
const url = URL.createObjectURL(blob);
|
||||
|
||||
// 处理文件名
|
||||
let downloadName = fileName;
|
||||
if (!downloadName) {
|
||||
// 从响应头获取文件名
|
||||
const contentDisposition = response.headers.get('content-disposition');
|
||||
if (contentDisposition) {
|
||||
const match = contentDisposition.match(/filename[^;=\n]*=((['"]).*?\2|[^;\n]*)/);
|
||||
if (match && match[1]) {
|
||||
downloadName = decodeURIComponent(match[1].replace(/['"]/g, ''));
|
||||
}
|
||||
} else {
|
||||
// 从URL提取文件名
|
||||
downloadName = fileUrl.split('/').pop()?.split('?')[0] || 'download_file';
|
||||
}
|
||||
}
|
||||
|
||||
// 创建a标签并触发下载
|
||||
const link = document.createElement('a');
|
||||
link.href = url;
|
||||
link.download = downloadName;
|
||||
link.style.display = 'none'; // 隐藏a标签
|
||||
document.body.appendChild(link);
|
||||
link.click();
|
||||
|
||||
// 清理资源
|
||||
setTimeout(() => {
|
||||
document.body.removeChild(link);
|
||||
URL.revokeObjectURL(url); // 释放临时URL
|
||||
}, 100);
|
||||
} catch (error) {
|
||||
ElMessage.error(`文件下载失败: ${error.message}`);
|
||||
}
|
||||
};
|
Reference in New Issue
Block a user