Skip to main content
Bun 的服务器端 WebSocket API 提供了原生的发布-订阅（pub-sub）API。可以使用 socket.subscribe(<name>) 让套接字订阅一组命名频道；可以使用 socket.publish(<name>, <message>) 向某个频道发布消息。 以下代码片段实现了一个简单的单频道聊天室服务器。
https://mintcdn.com/ikxin/RzFFGbzo0-4huILA/icons/typescript.svg?fit=max&auto=format&n=RzFFGbzo0-4huILA&q=85&s=a3dffd2241f05776d3bd25171d0c5a79server.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: 像这样指定 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}`);