95 lines
3.3 KiB
TypeScript
95 lines
3.3 KiB
TypeScript
'use client';
|
|
import { create } from 'zustand';
|
|
import { loadServerData, saveServerData } from '@/lib/server-data';
|
|
import {
|
|
mergeDailySnapshotCacheForSave,
|
|
mergeInsightCacheForSave,
|
|
upsertDailySnapshot,
|
|
upsertInsight,
|
|
type XiaobaoRiskInsightCacheItem,
|
|
} from '@/lib/xiaobao-risk-cache';
|
|
import type { XiaobaoRiskSnapshot } from '@/lib/xiaobao-risk-trend';
|
|
|
|
interface XiaobaoRiskState {
|
|
snapshots: XiaobaoRiskSnapshot[];
|
|
insights: XiaobaoRiskInsightCacheItem[];
|
|
riskDataLoaded: boolean;
|
|
error?: string;
|
|
fetchRiskData: () => Promise<void>;
|
|
saveSnapshot: (item: XiaobaoRiskSnapshot) => Promise<void>;
|
|
saveInsight: (item: XiaobaoRiskInsightCacheItem) => Promise<void>;
|
|
}
|
|
|
|
async function loadSnapshots(): Promise<XiaobaoRiskSnapshot[] | null> {
|
|
try {
|
|
const rows = await loadServerData<XiaobaoRiskSnapshot[]>('xiaobao-risk-snapshots');
|
|
return Array.isArray(rows) ? rows : [];
|
|
} catch {}
|
|
return null;
|
|
}
|
|
|
|
async function loadInsights(): Promise<XiaobaoRiskInsightCacheItem[] | null> {
|
|
try {
|
|
const rows = await loadServerData<XiaobaoRiskInsightCacheItem[]>('xiaobao-risk-insights');
|
|
return Array.isArray(rows) ? rows : [];
|
|
} catch {}
|
|
return null;
|
|
}
|
|
|
|
let snapshotSaveQueue: Promise<void> = Promise.resolve();
|
|
let insightSaveQueue: Promise<void> = Promise.resolve();
|
|
|
|
export const useXiaobaoRiskStore = create<XiaobaoRiskState>((set, get) => ({
|
|
snapshots: [],
|
|
insights: [],
|
|
riskDataLoaded: false,
|
|
error: undefined,
|
|
|
|
fetchRiskData: async () => {
|
|
set({ riskDataLoaded: false });
|
|
const [snapshots, insights] = await Promise.all([loadSnapshots(), loadInsights()]);
|
|
set({
|
|
...(snapshots ? { snapshots } : {}),
|
|
...(insights ? { insights } : {}),
|
|
riskDataLoaded: snapshots !== null && insights !== null,
|
|
error: snapshots === null || insights === null ? '小宝预警缓存加载失败' : undefined,
|
|
});
|
|
},
|
|
|
|
saveSnapshot: async (item) => {
|
|
const optimistic = upsertDailySnapshot(get().snapshots, item);
|
|
set({ snapshots: optimistic, error: undefined });
|
|
const task = snapshotSaveQueue.then(async () => {
|
|
const remote = await loadServerData<XiaobaoRiskSnapshot[]>('xiaobao-risk-snapshots');
|
|
const snapshots = mergeDailySnapshotCacheForSave(get().snapshots, Array.isArray(remote) ? remote : [], item);
|
|
set({ snapshots, error: undefined });
|
|
await saveServerData('xiaobao-risk-snapshots', snapshots);
|
|
});
|
|
snapshotSaveQueue = task.catch(() => undefined);
|
|
try {
|
|
await task;
|
|
} catch (error) {
|
|
set({ error: '小宝预警快照保存失败' });
|
|
throw error;
|
|
}
|
|
},
|
|
|
|
saveInsight: async (item) => {
|
|
const optimistic = upsertInsight(get().insights, item);
|
|
set({ insights: optimistic, error: undefined });
|
|
const task = insightSaveQueue.then(async () => {
|
|
const remote = await loadServerData<XiaobaoRiskInsightCacheItem[]>('xiaobao-risk-insights');
|
|
const insights = mergeInsightCacheForSave(get().insights, Array.isArray(remote) ? remote : [], item);
|
|
set({ insights, error: undefined });
|
|
await saveServerData('xiaobao-risk-insights', insights);
|
|
});
|
|
insightSaveQueue = task.catch(() => undefined);
|
|
try {
|
|
await task;
|
|
} catch (error) {
|
|
set({ error: '小宝预警解读保存失败' });
|
|
throw error;
|
|
}
|
|
},
|
|
}));
|