コンテンツへスキップ

WebSocket ヘルパー

WebSocket ヘルパーは、Hono アプリケーションにおけるサーバーサイド WebSocket のヘルパーです。現在は、Cloudflare Workers / Pages、Deno、および Bun アダプターが使用できます。

インポート

ts
import { Hono } from 'hono'
import { upgradeWebSocket } from 'hono/cloudflare-workers'
ts
import { Hono } from 'hono'
import { upgradeWebSocket } from 'hono/deno'
ts
import { Hono } from 'hono'
import { createBunWebSocket } from 'hono/bun'
import type { ServerWebSocket } from 'bun'

const { upgradeWebSocket, websocket } =
  createBunWebSocket<ServerWebSocket>()

// ...

export default {
  fetch: app.fetch,
  websocket,
}

Node.js を使用している場合は、@hono/node-ws を使用できます。

upgradeWebSocket()

upgradeWebSocket() は、WebSocket を処理するためのハンドラーを返します。

ts
const app = new Hono()

app.get(
  '/ws',
  upgradeWebSocket((c) => {
    return {
      onMessage(event, ws) {
        console.log(`Message from client: ${event.data}`)
        ws.send('Hello from server!')
      },
      onClose: () => {
        console.log('Connection closed')
      },
    }
  })
)

使用可能なイベント

  • onOpen - 現在、Cloudflare Workers はこれをサポートしていません。
  • onMessage
  • onClose
  • onError

警告

WebSocket ヘルパーを使用するルートでヘッダーを変更する(CORS を適用するなど)ミドルウェアを使用している場合は、「immutable のヘッダーは変更できません」というエラーが発生する場合があります。これは、upgradeWebSocket() も内部的にヘッダーを変更するためです。

したがって、WebSocket ヘルパーとミドルウェアを同時に使用している場合は注意してください。

RPC モード

WebSocket ヘルパーで定義されたハンドラーは、RPC モードをサポートします。

ts
// server.ts
const wsApp = app.get(
  '/ws',
  upgradeWebSocket((c) => {
    //...
  })
)

export type WebSocketApp = typeof wsApp

// client.ts
const client = hc<WebSocketApp>('http://localhost:8787')
const socket = client.ws.$ws() // A WebSocket object for a client

WebSocket ヘルパーを使用した例をご覧ください。

サーバーとクライアント

ts
// server.ts
import { Hono } from 'hono'
import { upgradeWebSocket } from 'hono/cloudflare-workers'

const app = new Hono().get(
  '/ws',
  upgradeWebSocket(() => {
    return {
      onMessage: (event) => {
        console.log(event.data)
      },
    }
  })
)

export default app
ts
// client.ts
import { hc } from 'hono/client'
import type app from './server'

const client = hc<typeof app>('http://localhost:8787')
const ws = client.ws.$ws(0)

ws.addEventListener('open', () => {
  setInterval(() => {
    ws.send(new Date().toString())
  }, 1000)
})

JSX を使ったバン

tsx
import { Hono } from 'hono'
import { createBunWebSocket } from 'hono/bun'

const { upgradeWebSocket, websocket } = createBunWebSocket()

const app = new Hono()

app.get('/', (c) => {
  return c.html(
    <html>
      <head>
        <meta charset='UTF-8' />
      </head>
      <body>
        <div id='now-time'></div>
        <script
          dangerouslySetInnerHTML={{
            __html: `
        const ws = new WebSocket('ws://localhost:3000/ws')
        const $nowTime = document.getElementById('now-time')
        ws.onmessage = (event) => {
          $nowTime.textContent = event.data
        }
        `,
          }}
        ></script>
      </body>
    </html>
  )
})

const ws = app.get(
  '/ws',
  upgradeWebSocket((c) => {
    let intervalId
    return {
      onOpen(_event, ws) {
        intervalId = setInterval(() => {
          ws.send(new Date().toString())
        }, 200)
      },
      onClose() {
        clearInterval(intervalId)
      },
    }
  })
)

export default {
  fetch: app.fetch,
  websocket,
}

MIT ライセンス下で公開されています。