createWorkerTransport() — vite Function Reference

Architecture documentation for the createWorkerTransport() function in server-worker-runner.spec.ts from the vite codebase.

Function typescript ViteCore ConfigEngine

Entity Profile

ViteCore→ ConfigEngine→ createWorkerTransport() — vite Function Reference

Dependency Diagram

graph TD
  81e5b879_3bdb_6a12_d9b1_daea8e2beb81["createWorkerTransport()"]
  8cd46fdb_679a_7d1f_531c_774b779345ea["server-worker-runner.spec.ts"]
  81e5b879_3bdb_6a12_d9b1_daea8e2beb81 -->|defined in| 8cd46fdb_679a_7d1f_531c_774b779345ea
  style 81e5b879_3bdb_6a12_d9b1_daea8e2beb81 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

packages/vite/src/node/ssr/runtime/__tests__/server-worker-runner.spec.ts lines 7–58

const createWorkerTransport = (worker: Worker): HotChannel => {
  const handlerToWorkerListener = new WeakMap<
    HotChannelListener,
    (value: HotPayload) => void
  >()
  const client = {
    send(payload: HotPayload) {
      worker.postMessage(payload)
    },
  }

  return {
    send: (data) => worker.postMessage(data),
    on: (event: string, handler: HotChannelListener) => {
      // client is already connected
      if (event === 'vite:client:connect') return
      if (event === 'vite:client:disconnect') {
        const listener = () => {
          handler(undefined, client)
        }
        handlerToWorkerListener.set(handler, listener)
        worker.on('exit', listener)
        return
      }

      const listener = (value: HotPayload) => {
        if (value.type === 'custom' && value.event === event) {
          handler(value.data, client)
        }
      }
      handlerToWorkerListener.set(handler, listener)
      worker.on('message', listener)
    },
    off: (event, handler: HotChannelListener) => {
      if (event === 'vite:client:connect') return
      if (event === 'vite:client:disconnect') {
        const listener = handlerToWorkerListener.get(handler)
        if (listener) {
          worker.off('exit', listener)
          handlerToWorkerListener.delete(handler)
        }
        return
      }

      const listener = handlerToWorkerListener.get(handler)
      if (listener) {
        worker.off('message', listener)
        handlerToWorkerListener.delete(handler)
      }
    },
  }
}