opencode fixed my connection errors :)

This commit is contained in:
2026-06-29 14:46:41 +01:00
parent 07692fe0bd
commit 3be0033a32
4 changed files with 57 additions and 17 deletions

View File

@@ -1,34 +1,77 @@
export async function generateEndpoint(
startFunction?: (enqueue: (data: any) => void) => void | Promise<void | (() => void)>
startFunction?: (enqueue: (data: any) => void) => void | Promise<void | (() => void)>,
request?: Request
) {
let streamController: ReadableStreamDefaultController | null = null;
let cleanupFunction: (() => void) | void = undefined;
let heartbeatInterval: ReturnType<typeof setInterval> | null = null;
let cleanedUp = false;
const safeCleanup = () => {
if (cleanedUp) return;
cleanedUp = true;
if (heartbeatInterval) {
clearInterval(heartbeatInterval);
heartbeatInterval = null;
}
if (typeof cleanupFunction === 'function') {
cleanupFunction();
cleanupFunction = undefined;
}
};
const enqueue = (data: any) => {
if (cleanedUp) return;
let transferdata = JSON.stringify(data);
// stringify data and add to controller queue
if (streamController) {
streamController.enqueue(`data: ${transferdata}\n\n`);
} else {
console.log('no controller');
try {
streamController.enqueue(`data: ${transferdata}\n\n`);
} catch {
safeCleanup();
}
}
};
const stream = new ReadableStream({
async start(controller) {
streamController = controller;
heartbeatInterval = setInterval(() => {
try {
streamController!.enqueue(': keepalive\n\n');
} catch {
safeCleanup();
}
}, 30000);
heartbeatInterval.unref();
if (startFunction) {
const result = await startFunction(enqueue);
if (typeof result === 'function') {
cleanupFunction = result;
try {
const result = await startFunction(enqueue);
if (typeof result === 'function') {
cleanupFunction = result;
}
} catch (err) {
safeCleanup();
throw err;
}
}
},
async cancel() {
if (cleanupFunction) cleanupFunction();
safeCleanup();
}
});
if (request?.signal) {
if (request.signal.aborted) {
safeCleanup();
} else {
request.signal.addEventListener('abort', () => safeCleanup());
}
}
return {
response: new Response(stream, {
headers: {

View File

@@ -1,5 +1,4 @@
import { EventEmitter } from 'node:events';
// Main emitter for everything
export const globalEmitter = new EventEmitter();
globalEmitter.setMaxListeners(1000);