32 lines
1002 B
TypeScript
32 lines
1002 B
TypeScript
import type { SqlFileProgress } from "@/lib/backend/tauri";
|
|
import { apiUrl } from "@/lib/common/webPath";
|
|
|
|
export function listenSqlFileProgressById(executionId: string, handler: (progress: SqlFileProgress) => void, onError?: (error: Error) => void): Promise<() => void> {
|
|
return new Promise((resolve, reject) => {
|
|
const es = new EventSource(apiUrl(`/api/sql-file/progress/${executionId}`));
|
|
let opened = false;
|
|
const close = () => es.close();
|
|
|
|
es.onopen = () => {
|
|
opened = true;
|
|
resolve(close);
|
|
};
|
|
es.onmessage = (e) => {
|
|
const progress: SqlFileProgress = JSON.parse(e.data);
|
|
handler(progress);
|
|
if (progress.status === "done" || progress.status === "error" || progress.status === "cancelled") {
|
|
close();
|
|
}
|
|
};
|
|
es.onerror = () => {
|
|
close();
|
|
const error = new Error("SQL file progress connection failed");
|
|
if (opened) {
|
|
onError?.(error);
|
|
} else {
|
|
reject(error);
|
|
}
|
|
};
|
|
});
|
|
}
|