Function connectWebSocket

packages/platform/deno/test/DenoHttpServer.test.ts:715–735

(url: string, onOpen: (socket: WebSocket) => void, messageCount: number)

  )

const connectWebSocket = (url: string, onOpen: (socket: WebSocket) => void, messageCount: number) =>
  Effect.acquireUseRelease(
    openWebSocket(url),
    (socket) =>
      Effect.gen(function*() {
        const messages: Array<string> = []
        const fiber = yield* Effect.callback<void, Error>((resume) => {
          socket.addEventListener("message", (event) => {
            messages.push(event.data)
            if (messages.length === messageCount) resume(Effect.void)
          })
          socket.addEventListener("error", () => resume(Effect.fail(new Error("WebSocket connection failed"))), {
            once: true
          })
          onOpen(socket)
        }).pipe(Effect.forkChild)
        yield* Fiber.join(fiber)
        return messages
      }),
    (socket) => Effect.sync(() => socket.close())
  )