feat(DownloadAll): 并行下载

This commit is contained in:
awalol
2025-07-12 18:44:57 +08:00
committed by awalol
parent 2c461df5fc
commit 17200150dd

View File

@@ -6,6 +6,7 @@ import { toast } from 'react-toastify';
export function DownloadAll() {
const files = useAppSelector(selectFiles);
const onClickDownloadAll = async () => {
console.time('DownloadAll'); //开始计时
if (Object.keys(files).length === 0) {
toast.warning('未添加文件');
return;
@@ -20,11 +21,10 @@ export function DownloadAll() {
//过滤处理失败的文件
const completeFiles = Object.values(files).filter((file) => file.state === ProcessState.COMPLETE);
const filesLength = Object.keys(completeFiles).length;
const fileCount = Object.keys(files).length;
//开始下载
let dir: FileSystemDirectoryHandle | undefined;
let success = 0;
try {
dir = await window.showDirectoryPicker({ mode: 'readwrite' });
} catch (e) {
@@ -34,27 +34,35 @@ export function DownloadAll() {
}
}
toast.warning('开始下载,请稍候');
let success = 0;
const promises: Promise<void>[] = [];
for (const [_, file] of Object.entries(completeFiles)) {
if (file.state !== ProcessState.COMPLETE) {
return;
}
try {
if (dir) {
await DownloadNew(dir, file);
} else {
await DownloadOld(file);
}
success++;
} catch (e) {
console.error(`下载失败: ${file.fileName}`, e);
toast.error(`出现错误: ${e}`);
}
const promise = new Promise<void>((resolve, reject) => {
console.log(`开始下载: ${file.fileName}`);
const action = dir ? DownloadNew(dir, file) : DownloadOld(file);
action.then(
() => {
console.log(`成功下载: ${file.fileName}`);
success++;
resolve();
},
(e) => {
console.error(`下载失败: ${file.fileName}`, e);
toast.error(`出现错误: ${e}`);
reject(e);
},
);
});
promises.push(promise);
}
if (success === filesLength) {
toast.success(`成功下载: ${success}/${filesLength}`);
} else {
toast.warning(`成功下载: ${success}/${filesLength}`);
try {
await Promise.allSettled(promises);
toast.success(`成功下载: ${success}/${fileCount}`);
} catch {
toast.warning(`成功下载: ${success}/${fileCount}`);
}
console.timeEnd('DownloadAll'); //停止计时
};
return (