Added SSE function and add readAt for notification
This commit is contained in:
24
src/hooks/use-sse.ts
Normal file
24
src/hooks/use-sse.ts
Normal file
@@ -0,0 +1,24 @@
|
||||
import { useEffect } from 'react';
|
||||
|
||||
export function useSSE(onMessage: (data: any) => void) {
|
||||
useEffect(() => {
|
||||
const eventSource = new EventSource('/api/notify');
|
||||
|
||||
eventSource.onmessage = (event) => {
|
||||
try {
|
||||
const parsed = JSON.parse(event.data);
|
||||
onMessage(parsed);
|
||||
} catch (err) {
|
||||
console.error('Invalid SSE data', err);
|
||||
}
|
||||
};
|
||||
|
||||
eventSource.onerror = () => {
|
||||
eventSource.close();
|
||||
};
|
||||
|
||||
return () => {
|
||||
eventSource.close();
|
||||
};
|
||||
}, [onMessage]);
|
||||
}
|
||||
Reference in New Issue
Block a user