Method constructor

script/computerd-stub-soak.mjs:53–99

(webSocket)

class WebSocketTransport {
  constructor(webSocket) {
    this._ws = webSocket;
    this._sendQueue = [];
    this._receiveQueue = [];
    this._receiveResolver = undefined;
    this._receiveRejecter = undefined;
    this._error = undefined;
    this._opened = webSocket.readyState === WebSocket.OPEN;

    if (!this._opened) {
      webSocket.addEventListener("open", () => {
        try {
          for (const m of this._sendQueue) webSocket.send(m);
        } catch (err) {
          this._receivedError(err);
        }
        this._sendQueue = undefined;
        this._opened = true;
      });
    } else {
      this._sendQueue = undefined;
    }

    webSocket.addEventListener("message", (event) => {
      if (this._error) return;
      const data = typeof event.data === "string" ? event.data : event.data?.toString("utf8");
      if (typeof data !== "string") {
        this._receivedError(new TypeError("non-string ws message"));
        return;
      }
      if (this._receiveResolver) {
        const r = this._receiveResolver;
        this._receiveResolver = undefined;
        this._receiveRejecter = undefined;
        r(data);
      } else {
        this._receiveQueue.push(data);
      }
    });

    webSocket.addEventListener("close", (event) => {
      this._receivedError(new Error(`Peer closed WebSocket: ${event.code} ${event.reason}`));
    });
    webSocket.addEventListener("error", () => {
      this._receivedError(new Error("WebSocket connection failed."));
    });
  }

  async send(message) {
    if (this._sendQueue !== undefined) this._sendQueue.push(message);