fix: embedding background persistence, polling timeout, Error section hidden

- Embed task_id persisted in Zustand store (survives page navigation)
- Resume polling on page mount if active task exists
- Polling timeout increased to 30min (was 5min, too short for large sections)
- Embed button hidden for Error and Other sections
- Shared file-based status for multi-worker visibility
This commit is contained in:
nogueiraguh
2026-03-21 12:04:29 -03:00
parent ab2d0a3e9b
commit dcc202ff9f
2 changed files with 47 additions and 25 deletions

View File

@@ -581,7 +581,9 @@ export default function ReportsPage() {
const setEmbedAdb = store.setRptEmbedAdb;
const embedTable = store.rptEmbedTable;
const setEmbedTable = store.setRptEmbedTable;
const [embedLoading, setEmbedLoading] = useState(''); // '' = idle, 'all' = full embed, section name = section embed
const embedTaskId = store.rptEmbedTaskId;
const setEmbedTaskId = store.setRptEmbedTaskId;
const [embedLoading, setEmbedLoading] = useState('');
const [embedMsg, setEmbedMsg] = useState<{ type: 's' | 'e'; text: string } | null>(null);
const [embedProgress, setEmbedProgress] = useState<{ inserted: number; skipped: number; processed: number; total: number } | null>(null);
@@ -782,37 +784,53 @@ export default function ReportsPage() {
}
};
const pollEmbeddingStatus = useCallback(async (taskId: string) => {
setEmbedMsg({ type: 's', text: 'Embedding iniciado...' });
setEmbedProgress(null);
const pollEmbeddingStatus = useCallback((taskId: string) => {
setEmbedMsg({ type: 's', text: 'Embedding em andamento...' });
setEmbedTaskId(taskId);
const poll = setInterval(async () => {
try {
const s = await reportsApi.embeddingStatus(taskId);
if (s.total) {
setEmbedProgress({
inserted: s.inserted || 0,
skipped: (s as unknown as Record<string, number>).skipped || 0,
processed: (s as unknown as Record<string, number>).processed || s.inserted || 0,
total: s.total,
});
const data = s as unknown as Record<string, unknown>;
const total = (data.total as number) || 0;
const inserted = (data.inserted as number) || 0;
const skipped = (data.skipped as number) || 0;
const processed = (data.processed as number) || inserted;
if (total > 0) {
setEmbedProgress({ inserted, skipped, processed, total });
}
if (s.status === 'done') {
clearInterval(poll);
setEmbedProgress({ inserted, skipped, processed: total, total });
setEmbedMsg({ type: 's', text: s.message });
setTimeout(() => { setEmbedProgress(null); setEmbedLoading(''); setEmbedTaskId(''); }, 3000);
return;
}
if (s.status === 'error') {
clearInterval(poll);
setEmbedMsg({ type: 'e', text: s.message });
setTimeout(() => { setEmbedProgress(null); setEmbedLoading(''); setEmbedTaskId(''); }, 3000);
return;
}
if (s.status === 'running') {
setEmbedMsg({ type: 's', text: s.message });
} else if (s.status === 'done') {
clearInterval(poll);
setEmbedMsg({ type: 's', text: s.message });
setEmbedProgress(null);
setEmbedLoading('');
} else if (s.status === 'error') {
clearInterval(poll);
setEmbedMsg({ type: 'e', text: s.message });
setEmbedProgress(null);
setEmbedLoading('');
}
} catch { /* keep polling */ }
} catch {
// Network error — keep polling
}
}, 2000);
setTimeout(() => { clearInterval(poll); setEmbedLoading(''); setEmbedProgress(null); }, 300000);
}, []);
// Cleanup on unmount
return () => clearInterval(poll);
}, [setEmbedTaskId]);
// Resume polling if returning to page with active embed task
useEffect(() => {
if (!embedTaskId) return;
setEmbedLoading('all');
const cleanup = pollEmbeddingStatus(embedTaskId);
return cleanup;
}, []); // eslint-disable-line react-hooks/exhaustive-deps
const handleEmbed = async (rid: string) => {
const adbId = embedAdb || (adbCfg.length > 0 ? adbCfg[0].id : '');
@@ -1530,7 +1548,7 @@ export default function ReportsPage() {
<Icon size={13} style={{ color: 'var(--t3)', opacity: 0.7 }} />
<span className="text-[.72rem] font-bold" style={{ color: 'var(--t1)' }}>{sec}</span>
<span className="text-[.6rem]" style={{ color: 'var(--t4)' }}>({secFiles.length})</span>
{adbCfg.length > 0 && secFiles.some((f) => f.file_name.endsWith('.csv')) && (
{adbCfg.length > 0 && sec !== 'Error' && sec !== 'Other' && secFiles.some((f) => f.file_name.endsWith('.csv')) && (
<button
onClick={() => handleEmbedSection(sec, secFiles)}
disabled={!!embedLoading}

View File

@@ -222,6 +222,8 @@ interface AppState {
rptShowCompliance: boolean;
rptEmbedAdb: string;
rptEmbedTable: string;
rptEmbedTaskId: string;
setRptEmbedTaskId: (v: string) => void;
setRptOciVal: (v: string) => void;
setRptLevel: (v: 1 | 2) => void;
setRptSelectedRegions: (v: Updater<string[]>) => void;
@@ -353,6 +355,8 @@ export const useAppStore = create<AppState>((set) => ({
rptShowCompliance: false,
rptEmbedAdb: '',
rptEmbedTable: '',
rptEmbedTaskId: '',
setRptEmbedTaskId: (v) => set({ rptEmbedTaskId: v }),
setRptOciVal: (v) => set({ rptOciVal: v }),
setRptLevel: (v) => set({ rptLevel: v }),
setRptSelectedRegions: (v) => set((s) => ({ rptSelectedRegions: typeof v === 'function' ? v(s.rptSelectedRegions) : v })),