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.
93 lines
3.3 KiB
93 lines
3.3 KiB
4 months ago
|
/// <reference types="bun-types" />
|
||
|
|
import { GRAPHQL_TRANSPORT_WS_PROTOCOL, } from '../common.mjs';
|
||
|
|
import { makeServer } from '../server.mjs';
|
||
|
|
/**
|
||
|
|
* Convenience export for checking the WebSocket protocol on the request in `Bun.serve`.
|
||
|
|
*/
|
||
|
|
export { handleProtocols } from '../server.mjs';
|
||
|
|
/**
|
||
|
|
* Use the server with [Bun](https://bun.sh/).
|
||
|
|
* This is a basic starter, feel free to copy the code over and adjust it to your needs
|
||
|
|
*
|
||
|
|
* The WebSocket subprotocol is not available on the established socket and therefore
|
||
|
|
* needs to be checked during the request handling.
|
||
|
|
*
|
||
|
|
* Additionally, the keep-alive logic _seems_ to be handled by Bun seeing that
|
||
|
|
* they default [`sendPingsAutomatically` to `true`](https://github.com/oven-sh/bun/blob/6a163cf933542506354dc836bd92693bcae5939b/src/deps/uws.zig#L893).
|
||
|
|
*
|
||
|
|
* ```ts
|
||
|
|
* import { makeHandler, handleProtocols } from 'graphql-ws/lib/use/lib/bun';
|
||
|
|
* import { schema } from './my-schema/index.mjs';
|
||
|
|
*
|
||
|
|
* Bun.serve({
|
||
|
|
* fetch(req, server) {
|
||
|
|
* const [path, _search] = req.url.split('?');
|
||
|
|
* if (!path.endsWith('/graphql')) {
|
||
|
|
* return new Response('Not Found', { status: 404 });
|
||
|
|
* }
|
||
|
|
* if (req.headers.get('upgrade') != 'websocket') {
|
||
|
|
* return new Response('Upgrade Required', { status: 426 });
|
||
|
|
* }
|
||
|
|
* if (!handleProtocols(req.headers.get('sec-websocket-protocol') || '')) {
|
||
|
|
* return new Response('Bad Request', { status: 404 });
|
||
|
|
* }
|
||
|
|
* if (!server.upgrade(req)) {
|
||
|
|
* return new Response('Internal Server Error', { status: 500 });
|
||
|
|
* }
|
||
|
|
* return new Response();
|
||
|
|
* },
|
||
|
|
* websocket: makeHandler({ schema }),
|
||
|
|
* port: 4000,
|
||
|
|
* });
|
||
|
|
*
|
||
|
|
* console.log('Listening to port 4000');
|
||
|
|
* ```
|
||
|
|
*
|
||
|
|
* @category Server/bun
|
||
|
|
*/
|
||
|
|
export function makeHandler(options) {
|
||
|
|
const server = makeServer(options);
|
||
|
|
const clients = new WeakMap();
|
||
|
|
return {
|
||
|
|
open(ws) {
|
||
|
|
const client = {
|
||
|
|
handleMessage: () => {
|
||
|
|
throw new Error('Message received before handler was registered');
|
||
|
|
},
|
||
|
|
closed: () => {
|
||
|
|
throw new Error('Closed before handler was registered');
|
||
|
|
},
|
||
|
|
};
|
||
|
|
client.closed = server.opened({
|
||
|
|
// TODO: use protocol on socket once Bun exposes it
|
||
|
|
protocol: GRAPHQL_TRANSPORT_WS_PROTOCOL,
|
||
|
|
send: async (message) => {
|
||
|
|
// ws might have been destroyed in the meantime, send only if exists
|
||
|
|
if (clients.has(ws)) {
|
||
|
|
ws.sendText(message);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
close: (code, reason) => {
|
||
|
|
if (clients.has(ws)) {
|
||
|
|
ws.close(code, reason);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
onMessage: (cb) => (client.handleMessage = cb),
|
||
|
|
}, { socket: ws });
|
||
|
|
clients.set(ws, client);
|
||
|
|
},
|
||
|
|
message(ws, message) {
|
||
|
|
const client = clients.get(ws);
|
||
|
|
if (!client)
|
||
|
|
throw new Error('Message received for a missing client');
|
||
|
|
return client.handleMessage(String(message));
|
||
|
|
},
|
||
|
|
close(ws, code, message) {
|
||
|
|
const client = clients.get(ws);
|
||
|
|
if (!client)
|
||
|
|
throw new Error('Closing a missing client');
|
||
|
|
return client.closed(code, message);
|
||
|
|
},
|
||
|
|
};
|
||
|
|
}
|