WebSocket - Nitro 中文文档

Nitro 原生支持使用 CrossWS 和 H3 WebSocket 的运行时无关的 WebSocket API。

选择实验性功能

WebSockets 支持目前处于实验阶段。请查看 unjs/nitro#2171 了解平台支持状态。

要启用 WebSocket 支持，您需要启用实验性的 websocket 功能标志。

export default defineNitroConfig({
  experimental: {
    websocket: true
  }
})

用法

在 routes/_ws.ts 中创建一个 websocket 处理程序（或者在 Nuxt 中使用 server/routes/_ws.ts）。

您可以使用任何路由，如 routes/chatroom.ts，在 /chatroom 上注册升级处理程序。

_ws.ts

export default defineWebSocketHandler({
  open(peer) {
    console.log("[ws] open", peer);
  },

  message(peer, message) {
    console.log("[ws] message", peer, message);
    if (message.text().includes("ping")) {
      peer.send("pong");
    }
  },

  close(peer, event) {
    console.log("[ws] close", peer, event);
  },

  error(peer, error) {
    console.log("[ws] error", peer, error);
  },
});

Nitro 允许您使用相同的事件处理路由定义多个 websocket 处理程序。

使用客户端连接到服务器。例如：(routes/websocket.ts 或 Nuxt 的 server/routes/websocket.ts)

index.ts

export default defineEventHandler(() => {
  return $fetch(
    "https://raw.githubusercontent.com/unjs/crossws/main/examples/h3/public/index.html"
  );
});

现在您可以在 /websocket 路由上进行尝试！

查看我们的聊天演示，使用 Nitro WebSocket API。

服务器发送事件 (SSE)

作为 WebSockets 的替代方案，您可以使用服务器发送事件

示例

在 routes/sse.ts 中创建一个 SSE 处理程序（或在 Nuxt 中使用 server/routes/sse.ts）。

sse.ts

export default defineEventHandler(async (event) => {
  const eventStream = createEventStream(event)

  const interval = setInterval(async () => {
    await eventStream.push(`Message @ ${new Date().toLocaleTimeString()}`)
  }, 1000)

  eventStream.onClosed(async () => {
    clearInterval(interval)
    await eventStream.close()
  })

  return eventStream.send()
})

然后从客户端连接到这个 SSE 端点

const eventSource = new EventSource('http://localhost:3000/sse')

eventSource.onmessage = (event) => {
  console.log(event.data)
}