58 lines
1.6 KiB
JavaScript
58 lines
1.6 KiB
JavaScript
|
import { useState, useEffect } from 'react';
|
|||
|
|
|
|||
|
|
// 存储文件处理状态的共享对象
|
|||
|
|
const fileProcessingSubscribers = {
|
|||
|
|
value: false,
|
|||
|
|
listeners: new Set()
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 存储文件任务信息的共享对象
|
|||
|
|
const fileTaskSubscribers = {
|
|||
|
|
value: null,
|
|||
|
|
listeners: new Set()
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 自定义hook，用于在组件间共享文件处理任务的状态
|
|||
|
|
*/
|
|||
|
|
export default function useFileProcessingStatus() {
|
|||
|
|
const [taskFileProcessing, setTaskFileProcessing] = useState(fileProcessingSubscribers.value);
|
|||
|
|
const [task, setTask] = useState(fileTaskSubscribers.value);
|
|||
|
|
|
|||
|
|
useEffect(() => {
|
|||
|
|
// 添加当前组件为订阅者
|
|||
|
|
const updateProcessingState = newValue => setTaskFileProcessing(newValue);
|
|||
|
|
const updateTaskState = newTask => setTask(newTask);
|
|||
|
|
|
|||
|
|
fileProcessingSubscribers.listeners.add(updateProcessingState);
|
|||
|
|
fileTaskSubscribers.listeners.add(updateTaskState);
|
|||
|
|
|
|||
|
|
// 组件卸载时清理
|
|||
|
|
return () => {
|
|||
|
|
fileProcessingSubscribers.listeners.delete(updateProcessingState);
|
|||
|
|
fileTaskSubscribers.listeners.delete(updateTaskState);
|
|||
|
|
};
|
|||
|
|
}, []);
|
|||
|
|
|
|||
|
|
// 共享的setState函数
|
|||
|
|
const setSharedFileProcessing = newValue => {
|
|||
|
|
fileProcessingSubscribers.value = newValue;
|
|||
|
|
// 通知所有订阅者
|
|||
|
|
fileProcessingSubscribers.listeners.forEach(listener => listener(newValue));
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 共享的setTask函数
|
|||
|
|
const setSharedTask = newTask => {
|
|||
|
|
fileTaskSubscribers.value = newTask;
|
|||
|
|
// 通知所有订阅者
|
|||
|
|
fileTaskSubscribers.listeners.forEach(listener => listener(newTask));
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
taskFileProcessing,
|
|||
|
|
task,
|
|||
|
|
setTaskFileProcessing: setSharedFileProcessing,
|
|||
|
|
setTask: setSharedTask
|
|||
|
|
};
|
|||
|
|
}
|