L'API WebSocket côté serveur de Bun fournit une API pub-sub native. Les sockets peuvent être abonnés à un ensemble de canaux nommés en utilisant socket.subscribe(<name>) ; les messages peuvent être publiés sur un canal en utilisant socket.publish(<name>, <message>).
Cet extrait de code implémente un simple serveur de chat à canal unique.
ts
const server = Bun.serve({
fetch(req, server) {
const cookies = req.headers.get("cookie");
const username = getUsernameFromCookies(cookies);
const success = server.upgrade(req, { data: { username } });
if (success) return undefined;
return new Response("Hello world");
},
websocket: {
// TypeScript : spécifiez le type de ws.data comme ceci
data: {} as { username: string },
open(ws) {
const msg = `${ws.data.username} a rejoint le chat`;
ws.subscribe("the-group-chat");
server.publish("the-group-chat", msg);
},
message(ws, message) {
// le serveur rediffuse les messages entrants à tout le monde
server.publish("the-group-chat", `${ws.data.username} : ${message}`);
},
close(ws) {
const msg = `${ws.data.username} a quitté le chat`;
server.publish("the-group-chat", msg);
ws.unsubscribe("the-group-chat");
},
},
});
console.log(`Écoute sur ${server.hostname}:${server.port}`);