47 lines
1.3 KiB
TypeScript
47 lines
1.3 KiB
TypeScript
import { userEvents } from "~~/server/utils/events";
|
|
|
|
export default defineEventHandler(async (event) => {
|
|
await protectRoute(event);
|
|
|
|
const userId = event.context.user.id;
|
|
|
|
let controller: ReadableStreamDefaultController;
|
|
let pingInterval: NodeJS.Timeout;
|
|
|
|
const stream = new ReadableStream({
|
|
start(c) {
|
|
controller = c;
|
|
|
|
// Send initial connection message
|
|
try {
|
|
controller.enqueue(':connected\n\n');
|
|
} catch (e) {
|
|
console.error('Failed to send initial connection:', e);
|
|
return;
|
|
}
|
|
|
|
userEvents.subscribe(userId, controller);
|
|
|
|
pingInterval = setInterval(() => {
|
|
try {
|
|
controller.enqueue(':heartbeat\n\n');
|
|
} catch (e) {
|
|
clearInterval(pingInterval);
|
|
}
|
|
}, 15000);
|
|
},
|
|
cancel() {
|
|
console.log(`User ${userId} event stream cancelled`);
|
|
clearInterval(pingInterval);
|
|
userEvents.unsubscribe(userId, controller!);
|
|
}
|
|
});
|
|
|
|
setHeaders(event, {
|
|
'Content-Type': 'text/event-stream',
|
|
'Cache-Control': 'no-cache',
|
|
'Connection': 'keep-alive',
|
|
});
|
|
|
|
return sendStream(event, stream);
|
|
}); |