fix first and subsequent message handling

marhaupe · Oct 12, 2024 · 791d30e · 791d30e
1 parent db81e66
commit 791d30e
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/packages/frontend/src/App.tsx b/packages/frontend/src/App.tsx
@@ -51,17 +51,13 @@ export function App() {
     };
 
     eventSource.onmessage = (event: MessageEvent) => {
-      try {
-        const incomingLogs = JSON.parse(event.data) as Log[];
-        if (incomingLogs.length !== 1) {
-          logListRef.current?.resetVirtualization();
-          setLogs(incomingLogs);
-        } else {
-          setLogs((prevLogs) => [...prevLogs, ...incomingLogs]);
-        }
-      } catch (error) {
-        console.error("Error parsing log", error);
-      }
+      const incomingLogs = safeParseLogs(event.data);
+      logListRef.current?.resetVirtualization();
+      setLogs(incomingLogs);
+      eventSource.onmessage = (event: MessageEvent) => {
+        const incomingLogs = safeParseLogs(event.data);
+        setLogs((prevLogs) => [...prevLogs, ...incomingLogs]);
+      };
     };
 
     return () => {
@@ -151,3 +147,12 @@ export function App() {
     </div>
   );
 }
+
+function safeParseLogs(data: string) {
+  try {
+    return JSON.parse(data) as Log[];
+  } catch (error) {
+    console.error("Error parsing log", error);
+    return [];
+  }
+}