建置一個釋出-訂閱 WebSocket 伺服器

Bun 伺服器端的 WebSocket API 內建原生髮布-訂閱功能。使用 socket.subscribe(<name>) 讓套接字訂閱一組命名頻道；使用 server.publish(<name>, <message>) 向頻道釋出訊息。

這段程式碼片段實現了一個單頻道聊天伺服器。

server.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("你好，世界");
  },
  websocket: {
    // TypeScript：像這樣指定 ws.data 的型別
    data: {} as { username: string },

    open(ws) {
      const msg = `${ws.data.username} 已進入聊天室`;
      ws.subscribe("the-group-chat");
      server.publish("the-group-chat", msg);
    },
    message(ws, message) {
      // 伺服器將收到的訊息重新廣播給所有人
      server.publish("the-group-chat", `${ws.data.username}: ${message}`);
    },
    close(ws) {
      const msg = `${ws.data.username} 已離開聊天室`;
      server.publish("the-group-chat", msg);
      ws.unsubscribe("the-group-chat");
    },
  },
});

console.log(`監聽中，地址為 ${server.hostname}:${server.port}`);