You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
6.2 KiB
126 lines
6.2 KiB
"use strict"; |
|
Object.defineProperty(exports, "__esModule", { value: true }); |
|
exports.makeHandler = void 0; |
|
const server_1 = require("../../server"); |
|
const common_1 = require("../../common"); |
|
const utils_1 = require("../../utils"); |
|
/** |
|
* Make a handler to use on a [@fastify/websocket](https://github.com/fastify/fastify-websocket) route. |
|
* This is a basic starter, feel free to copy the code over and adjust it to your needs |
|
* |
|
* @category Server/@fastify/websocket |
|
*/ |
|
function makeHandler(options, |
|
/** |
|
* The timout between dispatched keep-alive messages. Internally uses the [ws Ping and Pongs]((https://developer.mozilla.org/en-US/docs/Web/API/wss_API/Writing_ws_servers#Pings_and_Pongs_The_Heartbeat_of_wss)) |
|
* to check that the link between the clients and the server is operating and to prevent the link |
|
* from being broken due to idling. |
|
* |
|
* @default 12_000 // 12 seconds |
|
*/ |
|
keepAlive = 12000) { |
|
const isProd = process.env.NODE_ENV === 'production'; |
|
const server = (0, server_1.makeServer)(options); |
|
// we dont have access to the fastify-websocket server instance yet, |
|
// register an error handler on first connection ONCE only |
|
let handlingServerEmittedErrors = false; |
|
return function handler(connection, request) { |
|
const { socket } = connection; |
|
// might be too late, but meh |
|
this.websocketServer.options.handleProtocols = server_1.handleProtocols; |
|
// handle server emitted errors only if not already handling |
|
if (!handlingServerEmittedErrors) { |
|
handlingServerEmittedErrors = true; |
|
this.websocketServer.once('error', (err) => { |
|
console.error('Internal error emitted on the WebSocket server. ' + |
|
'Please check your implementation.', err); |
|
// catch the first thrown error and re-throw it once all clients have been notified |
|
let firstErr = null; |
|
// report server errors by erroring out all clients with the same error |
|
for (const client of this.websocketServer.clients) { |
|
try { |
|
client.close(common_1.CloseCode.InternalServerError, isProd |
|
? 'Internal server error' |
|
: (0, utils_1.limitCloseReason)(err instanceof Error ? err.message : String(err), 'Internal server error')); |
|
} |
|
catch (err) { |
|
firstErr = firstErr !== null && firstErr !== void 0 ? firstErr : err; |
|
} |
|
} |
|
if (firstErr) |
|
throw firstErr; |
|
}); |
|
} |
|
// used as listener on two streams, prevent superfluous calls on close |
|
let emittedErrorHandled = false; |
|
function handleEmittedError(err) { |
|
if (emittedErrorHandled) |
|
return; |
|
emittedErrorHandled = true; |
|
console.error('Internal error emitted on a WebSocket socket. ' + |
|
'Please check your implementation.', err); |
|
socket.close(common_1.CloseCode.InternalServerError, isProd |
|
? 'Internal server error' |
|
: (0, utils_1.limitCloseReason)(err instanceof Error ? err.message : String(err), 'Internal server error')); |
|
} |
|
// fastify-websocket uses the WebSocket.createWebSocketStream, |
|
// therefore errors get emitted on both the connection and the socket |
|
connection.once('error', handleEmittedError); |
|
socket.once('error', handleEmittedError); |
|
// keep alive through ping-pong messages |
|
let pongWait = null; |
|
const pingInterval = keepAlive > 0 && isFinite(keepAlive) |
|
? setInterval(() => { |
|
// ping pong on open sockets only |
|
if (socket.readyState === socket.OPEN) { |
|
// terminate the connection after pong wait has passed because the client is idle |
|
pongWait = setTimeout(() => { |
|
socket.terminate(); |
|
}, keepAlive); |
|
// listen for client's pong and stop socket termination |
|
socket.once('pong', () => { |
|
if (pongWait) { |
|
clearTimeout(pongWait); |
|
pongWait = null; |
|
} |
|
}); |
|
socket.ping(); |
|
} |
|
}, keepAlive) |
|
: null; |
|
const closed = server.opened({ |
|
protocol: socket.protocol, |
|
send: (data) => new Promise((resolve, reject) => { |
|
if (socket.readyState !== socket.OPEN) |
|
return resolve(); |
|
socket.send(data, (err) => (err ? reject(err) : resolve())); |
|
}), |
|
close: (code, reason) => socket.close(code, reason), |
|
onMessage: (cb) => socket.on('message', async (event) => { |
|
try { |
|
await cb(String(event)); |
|
} |
|
catch (err) { |
|
console.error('Internal error occurred during message handling. ' + |
|
'Please check your implementation.', err); |
|
socket.close(common_1.CloseCode.InternalServerError, isProd |
|
? 'Internal server error' |
|
: (0, utils_1.limitCloseReason)(err instanceof Error ? err.message : String(err), 'Internal server error')); |
|
} |
|
}), |
|
}, { connection, request }); |
|
socket.once('close', (code, reason) => { |
|
if (pongWait) |
|
clearTimeout(pongWait); |
|
if (pingInterval) |
|
clearInterval(pingInterval); |
|
if (!isProd && |
|
code === common_1.CloseCode.SubprotocolNotAcceptable && |
|
socket.protocol === common_1.DEPRECATED_GRAPHQL_WS_PROTOCOL) |
|
console.warn(`Client provided the unsupported and deprecated subprotocol "${socket.protocol}" used by subscriptions-transport-ws.` + |
|
'Please see https://www.apollographql.com/docs/apollo-server/data/subscriptions/#switching-from-subscriptions-transport-ws.'); |
|
closed(code, String(reason)); |
|
}); |
|
}; |
|
} |
|
exports.makeHandler = makeHandler;
|
|
|