fix: 自由创作完成后同步余额通知

This commit is contained in:
hh
2026-07-13 17:11:37 +08:00
parent c5983abe35
commit 0e93d52db1
2 changed files with 24 additions and 5 deletions
+16 -4
View File
@@ -47,15 +47,20 @@ function pollDelay(count: number): number {
let refKeySeq = 0;
const nextRefKey = () => `ref-${Date.now()}-${refKeySeq++}`;
export function FreeCreatePage({ modelConfigs, onNotify }: {
export function FreeCreatePage({ modelConfigs, onNotify, onTaskSettled }: {
modelConfigs: ModelConfig[];
onNotify: (type: "success" | "error" | "info", text: string) => void;
onTaskSettled: () => void;
}) {
// App 每次渲染会重建 onNotify 箭头函数;用 ref 稳定身份,否则依赖它的 effect(首屏拉取/轮询)
// 会随 App 任意 setState(如 30s 未读轮询)反复重跑 → 任务流无谓全量刷新(实测踩坑)。
const onNotifyRef = useRef(onNotify);
onNotifyRef.current = onNotify;
const notify = useCallback((type: "success" | "error" | "info", text: string) => onNotifyRef.current(type, text), []);
// 同 onNotify:App 外壳重渲染时回调身份会变化,不能让任务列表/轮询 effect 因此重启。
const onTaskSettledRef = useRef(onTaskSettled);
onTaskSettledRef.current = onTaskSettled;
const refreshGlobalShell = useCallback(() => onTaskSettledRef.current(), []);
const videoConfigs = useMemo(
() => modelConfigs.filter((c) => c.capability === "video" && FC_MODELS.some((m) => m.name === c.name)),
[modelConfigs]
@@ -128,6 +133,8 @@ export function FreeCreatePage({ modelConfigs, onNotify }: {
} else {
stopPolling(id);
setProgress((prev) => { const next = { ...prev }; delete next[id]; sessionStorage.setItem(PROGRESS_KEY, JSON.stringify(next)); return next; });
// 任务结算会改变余额,并产生已有的生成/计费消息;同步 App 外壳即可,页面任务流仍由 patchTask 管理。
refreshGlobalShell();
if (data.task.status === "succeeded") notify("success", "视频生成完成");
else if (data.task.status === "failed") notify("error", data.task.error_message || "视频生成失败");
}
@@ -137,7 +144,7 @@ export function FreeCreatePage({ modelConfigs, onNotify }: {
}
};
pollTimersRef.current.set(id, window.setTimeout(() => void tick(), pollDelay(pollCountsRef.current.get(id) || 0)));
}, [patchTask, stopPolling, notify]);
}, [patchTask, stopPolling, notify, refreshGlobalShell]);
// 平滑进度动画:每 2s 给在途任务 +0.6~1.6%,封顶 95,sessionStorage 持久(刷新可续)
useEffect(() => {
@@ -332,7 +339,12 @@ export function FreeCreatePage({ modelConfigs, onNotify }: {
return next;
});
if (isInFlight(data.task.status)) schedulePoll(data.task.id);
else if (data.task.status === "failed") notify("error", data.task.error_message || "创建任务失败");
else {
// 极快失败/成功时不会进入 schedulePoll,也要同步一次全局余额与消息徽标。
refreshGlobalShell();
if (data.task.status === "succeeded") notify("success", "视频生成完成");
else if (data.task.status === "failed") notify("error", data.task.error_message || "创建任务失败");
}
return true;
} catch (error) {
setTasks((prev) => prev.filter((t) => t.id !== localId));
@@ -341,7 +353,7 @@ export function FreeCreatePage({ modelConfigs, onNotify }: {
} finally {
setSubmitting(false);
}
}, [schedulePoll, notify]);
}, [schedulePoll, notify, refreshGlobalShell]);
const handleSend = useCallback(async () => {
const prompt = promptRef.current?.getText() || "";