Function attachSocket

projects/runtime-node/src/server.ts:68–127

(runtime: RuntimeServer, socket: WebSocket, request: http.IncomingMessage, options: RuntimeNodeHttpServerOptions)

}

function attachSocket(runtime: RuntimeServer, socket: WebSocket, request: http.IncomingMessage, options: RuntimeNodeHttpServerOptions): void {
    const connectionId = `node:${randomUUID()}`
    const maxBufferedBytes = options.maxBufferedBytes ?? 16 * 1024 * 1024
    const heartbeatMs = options.heartbeatMs ?? 30_000
    let alive = true
    let disposed = false
    let heartbeat: ReturnType<typeof setInterval>

    const connection: RuntimeConnection = {
        id: connectionId,
        permissions: options.permissions,
        metadata: {
            clientRequestPrincipal: clientRequestPrincipalFromRequest(request),
        },
        send(message) {
            if (socket.readyState !== WebSocket.OPEN) return
            if (socket.bufferedAmount > maxBufferedBytes) {
                socket.close(1013, "client is too far behind")
                runtime.notify("connection/lagged", { connectionId, bufferedAmount: socket.bufferedAmount })
                return
            }
            socket.send(JSON.stringify(message))
        },
        close() {
            socket.close()
        },
    }

    const disposeRuntimeConnection = runtime.connect(connection)
    const disposeOnce = () => {
        if (disposed) return
        disposed = true
        clearInterval(heartbeat)
        disposeRuntimeConnection()
    }

    heartbeat = setInterval(() => {
        if (socket.readyState !== WebSocket.OPEN) {
            disposeOnce()
            return
        }
        if (!alive) {
            socket.terminate()
            disposeOnce()
            return
        }
        alive = false
        socket.ping()
    }, heartbeatMs)
    ;(heartbeat as { unref?: () => void }).unref?.()

    socket.on("message", (data) => {
        void runtime.handleMessage(connection, rawText(data))
    })
    socket.on("pong", () => {
        alive = true
    })
    socket.on("close", disposeOnce)