Bun 의 서버 측 WebSocket API 는 네이티브 pub-sub API 를 제공합니다. 소켓은 socket.subscribe(<name>) 을 사용하여 이름이 지정된 채널 세트를 구독할 수 있으며 socket.publish(<name>, <message>) 를 사용하여 채널에 메시지를 게시할 수 있습니다.

이 코드 스니펫은 간단한 단일 채널 채팅 서버를 구현합니다.

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} 에서 리스닝 중`);