修正ts类型
This commit is contained in:
117
src/utils/exportZip.ts
Normal file
117
src/utils/exportZip.ts
Normal file
@@ -0,0 +1,117 @@
|
||||
import JSZip from 'jszip';
|
||||
import { saveAs } from 'file-saver';
|
||||
// 假设使用的是 Element Plus,请根据实际 UI 库调整导入
|
||||
import { ElMessage } from 'element-plus';
|
||||
|
||||
/**
|
||||
* 图片数据接口
|
||||
*/
|
||||
interface ImageItem {
|
||||
name: string;
|
||||
url: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量导出图片并生成 ZIP
|
||||
* @param imageList 图片数组
|
||||
* @param zipName 压缩包名称
|
||||
* @param concurrency 并发请求数,默认 5,避免浏览器请求过多被拦截
|
||||
*/
|
||||
export async function exportImagesToZip(imageList: ImageItem[], zipName: string = 'images', concurrency: number = 5) {
|
||||
if (!imageList || imageList.length === 0) {
|
||||
ElMessage.warning('没有可导出的图片');
|
||||
return;
|
||||
}
|
||||
|
||||
const zip = new JSZip();
|
||||
|
||||
try {
|
||||
// 使用并发控制处理图片获取
|
||||
const tasks = imageList.map((item) => () => processImageItem(item, zip));
|
||||
|
||||
// 执行并发任务
|
||||
await runWithConcurrency(tasks, concurrency);
|
||||
|
||||
ElMessage.success('导出完成');
|
||||
|
||||
// 生成 ZIP 并下载
|
||||
const content = await zip.generateAsync({
|
||||
type: 'blob',
|
||||
compression: 'DEFLATE',
|
||||
compressionOptions: { level: 6 }
|
||||
});
|
||||
|
||||
saveAs(content, `${zipName}.zip`);
|
||||
} catch (err) {
|
||||
console.error('ZIP 导出失败:', err);
|
||||
ElMessage.error('图片导出失败,请重试');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理单个图片项:获取 Blob 并添加到 Zip
|
||||
*/
|
||||
async function processImageItem(item: ImageItem, zip: JSZip): Promise<void> {
|
||||
try {
|
||||
// 清理文件名,防止非法字符或扩展名缺失
|
||||
const safeName = sanitizeFileName(item.name);
|
||||
const blob = await getImageBlobViaCanvas(item.url);
|
||||
zip.file(`${safeName}.png`, blob, { binary: true });
|
||||
} catch (error) {
|
||||
console.warn(`图片加载失败: ${item.url}`, error);
|
||||
throw error; // 抛出错误以便外层统计或处理
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 通过 Canvas 获取图片 Blob (解决 OSS 跨域 fetch 问题)
|
||||
* 注意:这要求 OSS _bucket_ 配置了 CORS 允许 '*' 或你的域名,且允许 GET 方法。
|
||||
* 如果 OSS 完全禁止跨域,此方法也会失败,但通常公共读 OSS 是允许 img 标签跨域显示的。
|
||||
*/
|
||||
async function getImageBlobViaCanvas(url: string): Promise<Blob> {
|
||||
const response = await fetch(url);
|
||||
if (!response.ok) {
|
||||
throw new Error(`HTTP error! status: ${response.status}`);
|
||||
}
|
||||
return response.blob();
|
||||
}
|
||||
|
||||
/**
|
||||
* 简单的并发控制执行器
|
||||
*/
|
||||
async function runWithConcurrency(tasks: (() => Promise<void>)[], limit: number): Promise<void> {
|
||||
const results: Promise<void>[] = [];
|
||||
let index = 0;
|
||||
|
||||
async function worker() {
|
||||
while (index < tasks.length) {
|
||||
const currentIndex = index++;
|
||||
const task = tasks[currentIndex];
|
||||
try {
|
||||
await task();
|
||||
} catch (e) {
|
||||
// 这里可以选择忽略单个失败,或者累计错误
|
||||
console.error(`Task ${currentIndex} failed`, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 创建 limit 个工人
|
||||
const workers = Array(Math.min(limit, tasks.length)).fill(null).map(worker);
|
||||
await Promise.all(workers);
|
||||
}
|
||||
|
||||
/**
|
||||
* 清理文件名,确保合法且唯一(简单版)
|
||||
*/
|
||||
function sanitizeFileName(name: string): string {
|
||||
// 移除非法字符,保留字母、数字、中文、下划线、点、横杠
|
||||
let cleanName = name.replace(/[<>:"/\\|?*]/g, '_');
|
||||
|
||||
// 如果名字为空或只有扩展名,给一个默认名
|
||||
if (!cleanName || cleanName.startsWith('.')) {
|
||||
cleanName = `image_${Date.now()}_${Math.random().toString(36).substr(2, 9)}${cleanName}`;
|
||||
}
|
||||
|
||||
return cleanName;
|
||||
}
|
||||
Reference in New Issue
Block a user