From 2c5b7da8165e3f8b90748e8ced1de0e3e8d52309 Mon Sep 17 00:00:00 2001 From: lazydancer Date: Sun, 22 Dec 2024 17:33:11 -0500 Subject: Initial --- frontend/src/useSSE.js | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 frontend/src/useSSE.js (limited to 'frontend/src/useSSE.js') diff --git a/frontend/src/useSSE.js b/frontend/src/useSSE.js new file mode 100644 index 0000000..e741895 --- /dev/null +++ b/frontend/src/useSSE.js @@ -0,0 +1,32 @@ +import { useState, useEffect } from 'react'; + +const useSSE = (url) => { + const [data, setData] = useState(null); + const [error, setError] = useState(''); + + useEffect(() => { + const eventSource = new EventSource(url); + + eventSource.onmessage = e => { + try { + const result = JSON.parse(e.data); + setData(result); + } catch (err) { + setError('Failed to parse data: ' + err.message); + } + }; + + eventSource.onerror = e => { + setError('EventSource failed'); + eventSource.close(); + }; + + return () => { + eventSource.close(); + }; + }, [url]); + + return { data, error }; +}; + +export default useSSE; -- cgit v1.3