132 lines
3.0 KiB
TypeScript
132 lines
3.0 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any, no-console */
|
|
|
|
/**
|
|
* 后台扫描任务管理
|
|
*/
|
|
|
|
export interface ScanTask {
|
|
id: string;
|
|
status: 'running' | 'completed' | 'failed';
|
|
progress: {
|
|
current: number;
|
|
total: number;
|
|
currentFolder?: string;
|
|
};
|
|
result?: {
|
|
total: number;
|
|
new: number;
|
|
existing: number;
|
|
errors: number;
|
|
};
|
|
error?: string;
|
|
startTime: number;
|
|
endTime?: number;
|
|
}
|
|
|
|
const tasks = new Map<string, ScanTask>();
|
|
|
|
export function createScanTask(): string {
|
|
const id = `scan_${Date.now()}_${Math.random().toString(36).substring(7)}`;
|
|
const task: ScanTask = {
|
|
id,
|
|
status: 'running',
|
|
progress: {
|
|
current: 0,
|
|
total: 0,
|
|
},
|
|
startTime: Date.now(),
|
|
};
|
|
tasks.set(id, task);
|
|
return id;
|
|
}
|
|
|
|
export function getScanTask(id: string): ScanTask | null {
|
|
return tasks.get(id) || null;
|
|
}
|
|
|
|
export function updateScanTaskProgress(
|
|
id: string,
|
|
current: number,
|
|
total: number,
|
|
currentFolder?: string
|
|
): void {
|
|
let task = tasks.get(id);
|
|
if (!task) {
|
|
// 如果任务不存在（可能因为模块重新加载），重新创建任务
|
|
console.warn(`[ScanTask] 任务 ${id} 不存在，重新创建`);
|
|
task = {
|
|
id,
|
|
status: 'running',
|
|
progress: {
|
|
current: 0,
|
|
total: 0,
|
|
},
|
|
startTime: Date.now(),
|
|
};
|
|
tasks.set(id, task);
|
|
}
|
|
task.progress = { current, total, currentFolder };
|
|
}
|
|
|
|
export function completeScanTask(
|
|
id: string,
|
|
result: ScanTask['result']
|
|
): void {
|
|
let task = tasks.get(id);
|
|
if (!task) {
|
|
// 如果任务不存在（可能因为模块重新加载），重新创建任务
|
|
console.warn(`[ScanTask] 任务 ${id} 不存在，重新创建并标记为完成`);
|
|
task = {
|
|
id,
|
|
status: 'completed',
|
|
progress: {
|
|
current: result?.total || 0,
|
|
total: result?.total || 0,
|
|
},
|
|
startTime: Date.now() - 60000, // 假设任务运行了1分钟
|
|
endTime: Date.now(),
|
|
result,
|
|
};
|
|
tasks.set(id, task);
|
|
return;
|
|
}
|
|
task.status = 'completed';
|
|
task.result = result;
|
|
task.endTime = Date.now();
|
|
}
|
|
|
|
export function failScanTask(id: string, error: string): void {
|
|
let task = tasks.get(id);
|
|
if (!task) {
|
|
// 如果任务不存在（可能因为模块重新加载），重新创建任务
|
|
console.warn(`[ScanTask] 任务 ${id} 不存在，重新创建并标记为失败`);
|
|
task = {
|
|
id,
|
|
status: 'failed',
|
|
progress: {
|
|
current: 0,
|
|
total: 0,
|
|
},
|
|
startTime: Date.now() - 60000, // 假设任务运行了1分钟
|
|
endTime: Date.now(),
|
|
error,
|
|
};
|
|
tasks.set(id, task);
|
|
return;
|
|
}
|
|
task.status = 'failed';
|
|
task.error = error;
|
|
task.endTime = Date.now();
|
|
}
|
|
|
|
export function cleanupOldTasks(): void {
|
|
const now = Date.now();
|
|
const maxAge = 60 * 60 * 1000; // 1小时
|
|
|
|
for (const [id, task] of Array.from(tasks.entries())) {
|
|
if (task.endTime && now - task.endTime > maxAge) {
|
|
tasks.delete(id);
|
|
}
|
|
}
|
|
}
|