misty.cafe/mistykey
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

fix null safety in StreamingApiServerService.ts

Browse source
This commit is contained in:
Hazelnoot 2025-09-18 20:37:02 -04:00
parent 6b281af9fa
commit e1639f70df
1 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
packages/backend/src/server/api/StreamingApiServerService.ts
View file

@ -35,7 +35,7 @@ const MAX_CONNECTIONS_PER_CLIENT = 32;
@Injectable()
export class StreamingApiServerService implements OnApplicationShutdown {
#wss: WebSocket.WebSocketServer;
#wss?: WebSocket.WebSocketServer;
#connections = new Map<WebSocket.WebSocket, number>();
#connectionsByClient = new Map<string, Set<WebSocket.WebSocket>>(); // key: IP / user ID -> value: connection
#cleanConnectionsIntervalId: NodeJS.Timeout | null = null;
@ -100,7 +100,7 @@ export class StreamingApiServerService implements OnApplicationShutdown {
@bindThis
public attach(server: http.Server): void {
this.#wss = new WebSocket.WebSocketServer({
const wss = this.#wss = new WebSocket.WebSocketServer({
noServer: true,
perMessageDeflate: this.config.websocketCompression,
});
@ -212,7 +212,7 @@ export class StreamingApiServerService implements OnApplicationShutdown {
await stream.init();
this.#wss.handleUpgrade(request, socket, head, (ws) => {
wss.handleUpgrade(request, socket, head, (ws) => {
connectionsForClient.add(ws);
// Call before emit() in case it throws an error.
@ -241,7 +241,7 @@ export class StreamingApiServerService implements OnApplicationShutdown {
};
ws.on('error', onWsInitError);
this.#wss.emit('connection', ws, request, {
wss.emit('connection', ws, request, {
stream, user, app,
});
ws.off('error', onWsInitError);
@ -321,14 +321,18 @@ export class StreamingApiServerService implements OnApplicationShutdown {
this.#connectionsByClient.clear();
await new Promise<void>((resolve, reject) => {
this.#wss.close(err => {
if (err) reject(err);
else resolve();
});
if (this.#wss) {
this.#wss.close(err => {
if (err) reject(err);
else resolve();
});
} else {
resolve();
}
});
// Don't disconnect this until *after* close returns
this.#wss.off('error', this.onWsError);
this.#wss?.off('error', this.onWsError);
}
@bindThis