Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c220bb3b7c | |||
| 36b6bfd20d |
@@ -114,25 +114,4 @@ describe('useGenerator', () => {
|
|||||||
expect(result.current.isGenerating).toBe(false);
|
expect(result.current.isGenerating).toBe(false);
|
||||||
expect(worker.postedMessages).toEqual(expect.arrayContaining([{ type: 'cancel' }]));
|
expect(worker.postedMessages).toEqual(expect.arrayContaining([{ type: 'cancel' }]));
|
||||||
});
|
});
|
||||||
|
|
||||||
it('正在生成时不应重复发送 start 消息', () => {
|
|
||||||
const { result } = renderHook(() => useGenerator());
|
|
||||||
|
|
||||||
act(() => {
|
|
||||||
result.current.generate([mockField], 10);
|
|
||||||
result.current.generate([mockField], 20);
|
|
||||||
});
|
|
||||||
|
|
||||||
const worker = MockWorker.instances[0];
|
|
||||||
const startMessages = worker.postedMessages.filter(
|
|
||||||
(message): message is { type: 'start'; payload: { count: number } } =>
|
|
||||||
typeof message === 'object' &&
|
|
||||||
message !== null &&
|
|
||||||
'type' in message &&
|
|
||||||
message.type === 'start',
|
|
||||||
);
|
|
||||||
|
|
||||||
expect(startMessages).toHaveLength(1);
|
|
||||||
expect(startMessages[0]?.payload.count).toBe(10);
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -37,12 +37,6 @@ export function useGenerator(): UseGeneratorReturn {
|
|||||||
|
|
||||||
const workerRef = useRef<Worker | null>(null);
|
const workerRef = useRef<Worker | null>(null);
|
||||||
const generationIdRef = useRef(0);
|
const generationIdRef = useRef(0);
|
||||||
const isGeneratingRef = useRef(false);
|
|
||||||
|
|
||||||
const finishGenerating = useCallback(() => {
|
|
||||||
isGeneratingRef.current = false;
|
|
||||||
setIsGenerating(false);
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
// 清理 Worker
|
// 清理 Worker
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -79,7 +73,7 @@ export function useGenerator(): UseGeneratorReturn {
|
|||||||
setProgress(data.payload);
|
setProgress(data.payload);
|
||||||
break;
|
break;
|
||||||
case 'complete':
|
case 'complete':
|
||||||
finishGenerating();
|
setIsGenerating(false);
|
||||||
if (data.payload.success) {
|
if (data.payload.success) {
|
||||||
setResult(data.payload);
|
setResult(data.payload);
|
||||||
} else if (data.payload.error && data.payload.error !== '生成已取消') {
|
} else if (data.payload.error && data.payload.error !== '生成已取消') {
|
||||||
@@ -88,7 +82,7 @@ export function useGenerator(): UseGeneratorReturn {
|
|||||||
setProgress(null);
|
setProgress(null);
|
||||||
break;
|
break;
|
||||||
case 'error':
|
case 'error':
|
||||||
finishGenerating();
|
setIsGenerating(false);
|
||||||
setError(data.payload.error);
|
setError(data.payload.error);
|
||||||
setProgress(null);
|
setProgress(null);
|
||||||
break;
|
break;
|
||||||
@@ -97,7 +91,7 @@ export function useGenerator(): UseGeneratorReturn {
|
|||||||
|
|
||||||
worker.onerror = (err) => {
|
worker.onerror = (err) => {
|
||||||
console.error('[useGenerator] Worker 错误:', err);
|
console.error('[useGenerator] Worker 错误:', err);
|
||||||
finishGenerating();
|
setIsGenerating(false);
|
||||||
setError(err.message || 'Worker 运行错误');
|
setError(err.message || 'Worker 运行错误');
|
||||||
setProgress(null);
|
setProgress(null);
|
||||||
// Worker 出错后销毁,下次重新创建
|
// Worker 出错后销毁,下次重新创建
|
||||||
@@ -107,17 +101,16 @@ export function useGenerator(): UseGeneratorReturn {
|
|||||||
|
|
||||||
workerRef.current = worker;
|
workerRef.current = worker;
|
||||||
return worker;
|
return worker;
|
||||||
}, [finishGenerating]);
|
}, []);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 开始生成
|
* 开始生成
|
||||||
*/
|
*/
|
||||||
const generate = useCallback(
|
const generate = useCallback(
|
||||||
(fields: FieldConfig[], count: number, csvMode = false) => {
|
(fields: FieldConfig[], count: number, csvMode = false) => {
|
||||||
if (isGeneratingRef.current) return;
|
if (isGenerating) return;
|
||||||
|
|
||||||
const generationId = ++generationIdRef.current;
|
const generationId = ++generationIdRef.current;
|
||||||
isGeneratingRef.current = true;
|
|
||||||
|
|
||||||
setIsGenerating(true);
|
setIsGenerating(true);
|
||||||
setProgress(null);
|
setProgress(null);
|
||||||
@@ -131,22 +124,21 @@ export function useGenerator(): UseGeneratorReturn {
|
|||||||
};
|
};
|
||||||
worker.postMessage(message);
|
worker.postMessage(message);
|
||||||
},
|
},
|
||||||
[getWorker],
|
[isGenerating, getWorker],
|
||||||
);
|
);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 取消生成
|
* 取消生成
|
||||||
*/
|
*/
|
||||||
const cancel = useCallback(() => {
|
const cancel = useCallback(() => {
|
||||||
if (workerRef.current && isGeneratingRef.current) {
|
if (workerRef.current && isGenerating) {
|
||||||
++generationIdRef.current;
|
++generationIdRef.current;
|
||||||
const message: WorkerRequestMessage = { type: 'cancel' };
|
const message: WorkerRequestMessage = { type: 'cancel' };
|
||||||
workerRef.current.postMessage(message);
|
workerRef.current.postMessage(message);
|
||||||
isGeneratingRef.current = false;
|
|
||||||
setIsGenerating(false);
|
setIsGenerating(false);
|
||||||
setProgress(null);
|
setProgress(null);
|
||||||
}
|
}
|
||||||
}, []);
|
}, [isGenerating]);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 清除结果
|
* 清除结果
|
||||||
|
|||||||
@@ -14,14 +14,10 @@ import type {
|
|||||||
/** 每生成 N 行让出一次事件循环,以便处理 cancel 消息 */
|
/** 每生成 N 行让出一次事件循环,以便处理 cancel 消息 */
|
||||||
const YIELD_EVERY = 100;
|
const YIELD_EVERY = 100;
|
||||||
|
|
||||||
/** 当前活跃生成任务 ID;新 start 会 supersede 旧任务 */
|
// 生成结果缓存
|
||||||
let activeGenerationId: number | null = null;
|
let generatedData: Record<string, unknown>[] = [];
|
||||||
let isCancelled = false;
|
let isCancelled = false;
|
||||||
|
|
||||||
function shouldAbort(generationId: number): boolean {
|
|
||||||
return isCancelled || generationId !== activeGenerationId;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Worker 消息处理器
|
* Worker 消息处理器
|
||||||
*/
|
*/
|
||||||
@@ -31,17 +27,12 @@ self.onmessage = async (e: MessageEvent<WorkerRequestMessage>) => {
|
|||||||
|
|
||||||
switch (type) {
|
switch (type) {
|
||||||
case 'start':
|
case 'start':
|
||||||
activeGenerationId = data.payload.generationId;
|
|
||||||
isCancelled = false;
|
isCancelled = false;
|
||||||
await handleStart(data.payload);
|
await handleStart(data.payload);
|
||||||
break;
|
break;
|
||||||
case 'cancel':
|
case 'cancel':
|
||||||
isCancelled = true;
|
isCancelled = true;
|
||||||
break;
|
break;
|
||||||
default: {
|
|
||||||
const _exhaustive: never = type;
|
|
||||||
return _exhaustive;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -55,7 +46,7 @@ async function handleStart(payload: {
|
|||||||
csvMode: boolean;
|
csvMode: boolean;
|
||||||
}): Promise<void> {
|
}): Promise<void> {
|
||||||
const { generationId, fields, count } = payload;
|
const { generationId, fields, count } = payload;
|
||||||
const generatedData: Record<string, unknown>[] = [];
|
generatedData = [];
|
||||||
|
|
||||||
try {
|
try {
|
||||||
// 验证所有生成器是否存在
|
// 验证所有生成器是否存在
|
||||||
@@ -76,17 +67,15 @@ async function handleStart(payload: {
|
|||||||
|
|
||||||
// 生成数据
|
// 生成数据
|
||||||
for (let i = 0; i < count; i++) {
|
for (let i = 0; i < count; i++) {
|
||||||
if (shouldAbort(generationId)) {
|
if (isCancelled) {
|
||||||
if (generationId === activeGenerationId) {
|
self.postMessage({
|
||||||
self.postMessage({
|
type: 'complete',
|
||||||
type: 'complete',
|
generationId,
|
||||||
generationId,
|
payload: {
|
||||||
payload: {
|
success: false,
|
||||||
success: false,
|
error: '生成已取消',
|
||||||
error: '生成已取消',
|
},
|
||||||
},
|
});
|
||||||
});
|
|
||||||
}
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -159,26 +148,20 @@ async function handleStart(payload: {
|
|||||||
// 定期让出事件循环,使 cancel 消息能被处理
|
// 定期让出事件循环,使 cancel 消息能被处理
|
||||||
if ((i + 1) % YIELD_EVERY === 0) {
|
if ((i + 1) % YIELD_EVERY === 0) {
|
||||||
await new Promise<void>((resolve) => setTimeout(resolve, 0));
|
await new Promise<void>((resolve) => setTimeout(resolve, 0));
|
||||||
if (shouldAbort(generationId)) {
|
if (isCancelled) {
|
||||||
if (generationId === activeGenerationId) {
|
self.postMessage({
|
||||||
self.postMessage({
|
type: 'complete',
|
||||||
type: 'complete',
|
generationId,
|
||||||
generationId,
|
payload: {
|
||||||
payload: {
|
success: false,
|
||||||
success: false,
|
error: '生成已取消',
|
||||||
error: '生成已取消',
|
},
|
||||||
},
|
});
|
||||||
});
|
|
||||||
}
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (shouldAbort(generationId)) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const duration = Date.now() - startTime;
|
const duration = Date.now() - startTime;
|
||||||
const successCount = generatedData.filter((item) => Object.keys(item).length > 0).length;
|
const successCount = generatedData.filter((item) => Object.keys(item).length > 0).length;
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user