Method handleWS

framework/framework/server.ts:620–731

(ctx: KoaContext, HandlerClass, checker, conn?, layer?, savedContext?: CordisContext)

    }

    private async handleWS(ctx: KoaContext, HandlerClass, checker, conn?, layer?, savedContext?: CordisContext) {
        const { args } = ctx.HydroContext;
        const sub = await forkContextWithScope(savedContext);
        const h = new HandlerClass(ctx, sub.ctx);
        let stream: PassThrough;
        if (!conn) {
            // By HTTP
            stream = new PassThrough();
            ctx.request.socket.setTimeout(0);
            ctx.req.socket.setNoDelay(true);
            ctx.req.socket.setKeepAlive(true);
            ctx.set({
                'X-Accel-Buffering': 'no',
                'Content-Type': 'text/event-stream',
                'Cache-Control': 'no-cache',
                Connection: 'keep-alive',
            });
            ctx.HydroContext.request.websocket = true;
            ctx.compress = false;
            conn = {
                close() {
                    stream.end();
                },
                send(data: any) {
                    stream.write(`${args.sse ? 'data: ' : ''}${data}\n${args.sse ? '\n' : ''}`);
                },
            };
        }
        ctx.handler = h;
        h.conn = conn;
        let closed = false;
        this.activeHandlers.set(h, { start: Date.now(), name: HandlerClass.name });

        const clean = async (err?: Error) => {
            if (closed) return;
            closed = true;
            try {
                try {
                    if (err) await h.onerror(err);
                    // FIXME: should pass type check
                    else (this.ctx.emit as any)('connection/close', h);
                } finally {
                    h.active = false;
                    if (layer) layer.clients.delete(conn);
                    if (err && !layer) ctx.status = 500;
                    await h.cleanup?.(args);
                }
            } finally {
                await sub.dispose();
                this.activeHandlers.delete(h);
            }
        };

        try {
            // FIXME: should pass type check
            await (this.ctx.parallel as any)('handler/create', h, 'ws');
            await (this.ctx.parallel as any)('handler/create/ws', h);
            checker.call(h);