@rocicorp/zero 0.25.0 → 0.26.0-canary.0

package/out/zero-cache/src/workers/connection.js

@@ -1,259 +1,298 @@
-import { pipeline, Readable, Writable } from 'node:stream';
-import WebSocket, { createWebSocketStream } from 'ws';
+import { pipeline, Writable, Readable } from "node:stream";
+import WebSocket, { createWebSocketStream } from "ws";
 import { assert } from "../../../shared/src/asserts.js";
-import * as valita from "../../../shared/src/valita.js";
-import { ErrorKind } from "../../../zero-protocol/src/error-kind.js";
-import { MIN_SERVER_SUPPORTED_SYNC_PROTOCOL, PROTOCOL_VERSION, } from "../../../zero-protocol/src/protocol-version.js";
+import { parse } from "../../../shared/src/valita.js";
+import { ClientNotFound, TransformFailed, VersionNotSupported, InvalidMessage, Internal } from "../../../zero-protocol/src/error-kind-enum.js";
+import { PROTOCOL_VERSION, MIN_SERVER_SUPPORTED_SYNC_PROTOCOL } from "../../../zero-protocol/src/protocol-version.js";
 import { upstreamSchema } from "../../../zero-protocol/src/up.js";
-import { ProtocolErrorWithLevel, getLogLevel, wrapWithProtocolError, } from "../types/error-with-level.js";
-import { isProtocolError, } from "../../../zero-protocol/src/error.js";
-import { ErrorOrigin } from "../../../zero-protocol/src/error-origin.js";
-// Ensures that a downstream message is sent at least every interval, sending a
-// 'pong' if necessary. This is set to be slightly longer than the client-side
-// PING_INTERVAL of 5 seconds, so that in the common case, 'pong's are sent in
-// response to client-initiated 'ping's. However, if the inbound stream is
-// backed up because a command is taking a long time to process, the pings
-// will be stuck in the queue (i.e. back-pressured), in which case pongs will
-// be manually sent to notify the client of server liveness.
-//
-// This is equivalent to what is done for Postgres keepalives on the
-// replication stream (which can similarly be back-pressured):
-// https://github.com/rocicorp/mono/blob/f98cb369a2dbb15650328859c732db358f187ef0/packages/zero-cache/src/services/change-source/pg/logical-replication/stream.ts#L21
-const DOWNSTREAM_MSG_INTERVAL_MS = 6_000;
-/**
- * Represents a connection between the client and server.
- *
- * Handles incoming messages on the connection and dispatches
- * them to the correct service.
- *
- * Listens to the ViewSyncer and sends messages to the client.
- */
-export class Connection {
-    #ws;
-    #wsID;
-    #protocolVersion;
-    #lc;
-    #onClose;
-    #messageHandler;
-    #downstreamMsgTimer;
-    #viewSyncerOutboundStream;
-    #pusherOutboundStream;
-    #closed = false;
-    constructor(lc, connectParams, ws, messageHandler, onClose) {
-        const { clientGroupID, clientID, wsID, protocolVersion } = connectParams;
-        this.#messageHandler = messageHandler;
-        this.#ws = ws;
-        this.#wsID = wsID;
-        this.#protocolVersion = protocolVersion;
-        this.#lc = lc
-            .withContext('connection')
-            .withContext('clientID', clientID)
-            .withContext('clientGroupID', clientGroupID)
-            .withContext('wsID', wsID);
-        this.#lc.debug?.('new connection');
-        this.#onClose = onClose;
-        this.#ws.addEventListener('close', this.#handleClose);
-        this.#ws.addEventListener('error', this.#handleError);
-        this.#proxyInbound();
-        this.#downstreamMsgTimer = setInterval(this.#maybeSendPong, DOWNSTREAM_MSG_INTERVAL_MS / 2);
+import { ProtocolErrorWithLevel, getLogLevel, wrapWithProtocolError } from "../types/error-with-level.js";
+import { isProtocolError } from "../../../zero-protocol/src/error.js";
+import { ZeroCache } from "../../../zero-protocol/src/error-origin-enum.js";
+const DOWNSTREAM_MSG_INTERVAL_MS = 6e3;
+class Connection {
+  #ws;
+  #wsID;
+  #protocolVersion;
+  #lc;
+  #onClose;
+  #messageHandler;
+  #downstreamMsgTimer;
+  #viewSyncerOutboundStream;
+  #pusherOutboundStream;
+  #closed = false;
+  constructor(lc, connectParams, ws, messageHandler, onClose) {
+    const { clientGroupID, clientID, wsID, protocolVersion } = connectParams;
+    this.#messageHandler = messageHandler;
+    this.#ws = ws;
+    this.#wsID = wsID;
+    this.#protocolVersion = protocolVersion;
+    this.#lc = lc.withContext("connection").withContext("clientID", clientID).withContext("clientGroupID", clientGroupID).withContext("wsID", wsID);
+    this.#lc.debug?.("new connection");
+    this.#onClose = onClose;
+    this.#ws.addEventListener("close", this.#handleClose);
+    this.#ws.addEventListener("error", this.#handleError);
+    this.#proxyInbound();
+    this.#downstreamMsgTimer = setInterval(
+      this.#maybeSendPong,
+      DOWNSTREAM_MSG_INTERVAL_MS / 2
+    );
+  }
+  /**
+   * Checks the protocol version and errors for unsupported protocols,
+   * sending the initial `connected` response on success.
+   *
+   * This is early in the connection lifecycle because {@link #handleMessage}
+   * will only parse messages with schema(s) of supported protocol versions.
+   */
+  init() {
+    if (this.#protocolVersion > PROTOCOL_VERSION || this.#protocolVersion < MIN_SERVER_SUPPORTED_SYNC_PROTOCOL) {
+      this.#closeWithError({
+        kind: VersionNotSupported,
+        message: `server is at sync protocol v${PROTOCOL_VERSION} and does not support v${this.#protocolVersion}. The ${this.#protocolVersion > PROTOCOL_VERSION ? "server" : "client"} must be updated to a newer release.`,
+        origin: ZeroCache
+      });
+    } else {
+      const connectedMessage = [
+        "connected",
+        { wsid: this.#wsID, timestamp: Date.now() }
+      ];
+      this.send(connectedMessage, "ignore-backpressure");
+      return true;
     }
-    /**
-     * Checks the protocol version and errors for unsupported protocols,
-     * sending the initial `connected` response on success.
-     *
-     * This is early in the connection lifecycle because {@link #handleMessage}
-     * will only parse messages with schema(s) of supported protocol versions.
-     */
-    init() {
-        if (this.#protocolVersion > PROTOCOL_VERSION ||
-            this.#protocolVersion < MIN_SERVER_SUPPORTED_SYNC_PROTOCOL) {
-            this.#closeWithError({
-                kind: ErrorKind.VersionNotSupported,
-                message: `server is at sync protocol v${PROTOCOL_VERSION} and does not support v${this.#protocolVersion}. The ${this.#protocolVersion > PROTOCOL_VERSION ? 'server' : 'client'} must be updated to a newer release.`,
-                origin: ErrorOrigin.ZeroCache,
-            });
-        }
-        else {
-            const connectedMessage = [
-                'connected',
-                { wsid: this.#wsID, timestamp: Date.now() },
-            ];
-            this.send(connectedMessage, 'ignore-backpressure');
-            return true;
-        }
-        return false;
-    }
-    close(reason, ...args) {
-        if (this.#closed) {
-            return;
-        }
-        this.#closed = true;
-        this.#lc.info?.(`closing connection: ${reason}`, ...args);
-        this.#ws.removeEventListener('close', this.#handleClose);
-        this.#ws.removeEventListener('error', this.#handleError);
-        this.#viewSyncerOutboundStream?.cancel();
-        this.#viewSyncerOutboundStream = undefined;
-        this.#pusherOutboundStream?.cancel();
-        this.#pusherOutboundStream = undefined;
-        this.#onClose();
-        if (this.#ws.readyState !== this.#ws.CLOSED) {
-            this.#ws.close();
-        }
-        clearTimeout(this.#downstreamMsgTimer);
-        // spin down services if we have
-        // no more client connections for the client group?
-    }
-    handleInitConnection(initConnectionMsg) {
-        return this.#handleMessage({ data: initConnectionMsg });
-    }
-    #handleMessage = async (event) => {
-        const data = event.data.toString();
-        if (this.#closed) {
-            this.#lc.debug?.('Ignoring message received after closed', data);
-            return;
-        }
-        let msg;
-        try {
-            const value = JSON.parse(data);
-            msg = valita.parse(value, upstreamSchema);
-        }
-        catch (e) {
-            this.#lc.warn?.(`failed to parse message "${data}": ${String(e)}`);
-            this.#closeWithError({
-                kind: ErrorKind.InvalidMessage,
-                message: String(e),
-                origin: ErrorOrigin.ZeroCache,
-            }, e);
-            return;
-        }
-        try {
-            const msgType = msg[0];
-            if (msgType === 'ping') {
-                this.send(['pong', {}], 'ignore-backpressure');
-                return;
-            }
-            const result = await this.#messageHandler.handleMessage(msg);
-            for (const r of result) {
-                this.#handleMessageResult(r);
-            }
-        }
-        catch (e) {
-            this.#closeWithThrown(e);
-        }
-    };
-    #handleMessageResult(result) {
-        switch (result.type) {
-            case 'fatal':
-                this.#closeWithError(result.error);
-                break;
-            case 'ok':
-                break;
-            case 'stream': {
-                switch (result.source) {
-                    case 'viewSyncer':
-                        assert(this.#viewSyncerOutboundStream === undefined, 'Outbound stream already set for this connection!');
-                        this.#viewSyncerOutboundStream = result.stream;
-                        break;
-                    case 'pusher':
-                        assert(this.#pusherOutboundStream === undefined, 'Outbound stream already set for this connection!');
-                        this.#pusherOutboundStream = result.stream;
-                        break;
-                }
-                this.#proxyOutbound(result.stream);
-                break;
-            }
-            case 'transient': {
-                for (const error of result.errors) {
-                    void this.sendError(error);
-                }
-            }
-        }
+    return false;
+  }
+  close(reason, ...args) {
+    if (this.#closed) {
+      return;
     }
-    #handleClose = (e) => {
-        const { code, reason, wasClean } = e;
-        this.close('WebSocket close event', { code, reason, wasClean });
-    };
-    #handleError = (e) => {
-        this.#lc.error?.('WebSocket error event', e.message, e.error);
-    };
-    #proxyInbound() {
-        pipeline(createWebSocketStream(this.#ws), new Writable({
-            write: (data, _encoding, callback) => {
-                this.#handleMessage({ data }).then(() => callback(), callback);
-            },
-        }), 
-        // The done callback is not used, as #handleClose and #handleError,
-        // configured on the underlying WebSocket, provide more complete
-        // information.
-        () => { });
+    this.#closed = true;
+    this.#lc.info?.(`closing connection: ${reason}`, ...args);
+    this.#ws.removeEventListener("close", this.#handleClose);
+    this.#ws.removeEventListener("error", this.#handleError);
+    this.#viewSyncerOutboundStream?.cancel();
+    this.#viewSyncerOutboundStream = void 0;
+    this.#pusherOutboundStream?.cancel();
+    this.#pusherOutboundStream = void 0;
+    this.#onClose();
+    if (this.#ws.readyState !== this.#ws.CLOSED) {
+      this.#ws.close();
     }
-    #proxyOutbound(outboundStream) {
-        // Note: createWebSocketStream() is avoided here in order to control
-        //       exception handling with #closeWithThrown(). If the Writable
-        //       from createWebSocketStream() were instead used, exceptions
-        //       from the outboundStream result in the Writable closing the
-        //       the websocket before the error message can be sent.
-        pipeline(Readable.from(outboundStream), new Writable({
-            objectMode: true,
-            write: (downstream, _encoding, callback) => this.send(downstream, callback),
-        }), e => e
-            ? this.#closeWithThrown(e)
-            : this.close(`downstream closed by ViewSyncer`));
+    clearTimeout(this.#downstreamMsgTimer);
+  }
+  handleInitConnection(initConnectionMsg) {
+    return this.#handleMessage({ data: initConnectionMsg });
+  }
+  #handleMessage = async (event) => {
+    const data = event.data.toString();
+    if (this.#closed) {
+      this.#lc.debug?.("Ignoring message received after closed", data);
+      return;
     }
-    #closeWithThrown(e) {
-        const errorBody = findProtocolError(e)?.errorBody ?? wrapWithProtocolError(e).errorBody;
-        this.#closeWithError(errorBody, e);
+    let msg;
+    try {
+      const value = JSON.parse(data);
+      msg = parse(value, upstreamSchema);
+    } catch (e) {
+      this.#lc.warn?.(`failed to parse message "${data}": ${String(e)}`);
+      this.#closeWithError(
+        {
+          kind: InvalidMessage,
+          message: String(e),
+          origin: ZeroCache
+        },
+        e
+      );
+      return;
     }
-    #closeWithError(errorBody, thrown) {
-        this.sendError(errorBody, thrown);
-        this.close(`${errorBody.kind} (${errorBody.origin}): ${errorBody.message}`, errorBody);
+    try {
+      const msgType = msg[0];
+      if (msgType === "ping") {
+        this.send(["pong", {}], "ignore-backpressure");
+        return;
+      }
+      const result = await this.#messageHandler.handleMessage(msg);
+      for (const r of result) {
+        this.#handleMessageResult(r);
+      }
+    } catch (e) {
+      this.#closeWithThrown(e);
     }
-    #lastDownstreamMsgTime = Date.now();
-    #maybeSendPong = () => {
-        if (Date.now() - this.#lastDownstreamMsgTime > DOWNSTREAM_MSG_INTERVAL_MS) {
-            this.#lc.debug?.('manually sending pong');
-            this.send(['pong', {}], 'ignore-backpressure');
+  };
+  #handleMessageResult(result) {
+    switch (result.type) {
+      case "fatal":
+        this.#closeWithError(result.error);
+        break;
+      case "ok":
+        break;
+      case "stream": {
+        switch (result.source) {
+          case "viewSyncer":
+            assert(
+              this.#viewSyncerOutboundStream === void 0,
+              "Outbound stream already set for this connection!"
+            );
+            this.#viewSyncerOutboundStream = result.stream;
+            break;
+          case "pusher":
+            assert(
+              this.#pusherOutboundStream === void 0,
+              "Outbound stream already set for this connection!"
+            );
+            this.#pusherOutboundStream = result.stream;
+            break;
+        }
+        this.#proxyOutbound(result.stream);
+        break;
+      }
+      case "transient": {
+        for (const error of result.errors) {
+          void this.sendError(error);
         }
-    };
-    send(data, callback) {
-        this.#lastDownstreamMsgTime = Date.now();
-        return send(this.#lc, this.#ws, data, callback);
+      }
     }
-    sendError(errorBody, thrown) {
-        sendError(this.#lc, this.#ws, errorBody, thrown);
+  }
+  #handleClose = (e) => {
+    const { code, reason, wasClean } = e;
+    this.close("WebSocket close event", { code, reason, wasClean });
+  };
+  #handleError = (e) => {
+    this.#lc.error?.("WebSocket error event", e.message, e.error);
+  };
+  #proxyInbound() {
+    pipeline(
+      createWebSocketStream(this.#ws),
+      new Writable({
+        write: (data, _encoding, callback) => {
+          this.#handleMessage({ data }).then(() => callback(), callback);
+        }
+      }),
+      // The done callback is not used, as #handleClose and #handleError,
+      // configured on the underlying WebSocket, provide more complete
+      // information.
+      () => {
+      }
+    );
+  }
+  #proxyOutbound(outboundStream) {
+    pipeline(
+      Readable.from(outboundStream),
+      new Writable({
+        objectMode: true,
+        write: (downstream, _encoding, callback) => this.send(downstream, callback)
+      }),
+      (e) => e ? this.#closeWithThrown(e) : this.close(`downstream closed by ViewSyncer`)
+    );
+  }
+  #closeWithThrown(e) {
+    const errorBody = findProtocolError(e)?.errorBody ?? wrapWithProtocolError(e).errorBody;
+    this.#closeWithError(errorBody, e);
+  }
+  #closeWithError(errorBody, thrown) {
+    this.sendError(errorBody, thrown);
+    this.close(
+      `${errorBody.kind} (${errorBody.origin}): ${errorBody.message}`,
+      errorBody
+    );
+  }
+  #lastDownstreamMsgTime = Date.now();
+  #maybeSendPong = () => {
+    if (Date.now() - this.#lastDownstreamMsgTime > DOWNSTREAM_MSG_INTERVAL_MS) {
+      this.#lc.debug?.("manually sending pong");
+      this.send(["pong", {}], "ignore-backpressure");
     }
+  };
+  send(data, callback) {
+    this.#lastDownstreamMsgTime = Date.now();
+    return send(this.#lc, this.#ws, data, callback);
+  }
+  sendError(errorBody, thrown) {
+    sendError(this.#lc, this.#ws, errorBody, thrown);
+  }
 }
-// Exported for testing purposes.
-export function send(lc, ws, data, callback) {
-    if (ws.readyState === WebSocket.OPEN) {
-        ws.send(JSON.stringify(data), callback === 'ignore-backpressure' ? undefined : callback);
-    }
-    else {
-        lc.debug?.(`Dropping outbound message on ws (state: ${ws.readyState})`, {
-            dropped: data,
-        });
-        if (callback !== 'ignore-backpressure') {
-            callback(new ProtocolErrorWithLevel({
-                kind: ErrorKind.Internal,
-                message: 'WebSocket closed',
-                origin: ErrorOrigin.ZeroCache,
-            }, 'info'));
-        }
+function send(lc, ws, data, callback) {
+  if (ws.readyState === WebSocket.OPEN) {
+    ws.send(
+      JSON.stringify(data),
+      callback === "ignore-backpressure" ? void 0 : callback
+    );
+  } else {
+    lc.debug?.(`Dropping outbound message on ws (state: ${ws.readyState})`, {
+      dropped: data
+    });
+    if (callback !== "ignore-backpressure") {
+      callback(
+        new ProtocolErrorWithLevel(
+          {
+            kind: Internal,
+            message: "WebSocket closed",
+            origin: ZeroCache
+          },
+          "info"
+        )
+      );
     }
+  }
 }
-export function sendError(lc, ws, errorBody, thrown) {
-    lc = lc.withContext('errorKind', errorBody.kind);
-    const logLevel = thrown ? getLogLevel(thrown) : 'info';
-    lc[logLevel]?.('Sending error on WebSocket', errorBody, thrown ?? '');
-    send(lc, ws, ['error', errorBody], 'ignore-backpressure');
+function sendError(lc, ws, errorBody, thrown) {
+  lc = lc.withContext("errorKind", errorBody.kind);
+  let logLevel;
+  if (thrown instanceof ProtocolErrorWithLevel) {
+    logLevel = thrown.logLevel;
+  } else if (hasErrno(thrown) || hasTransientSocketCode(thrown) || isTransientSocketMessage(errorBody.message)) {
+    logLevel = "warn";
+  } else if (errorBody.kind === ClientNotFound || errorBody.kind === TransformFailed) {
+    logLevel = "warn";
+  } else {
+    logLevel = thrown ? getLogLevel(thrown) : "info";
+  }
+  lc[logLevel]?.("Sending error on WebSocket", errorBody, thrown ?? "");
+  send(lc, ws, ["error", errorBody], "ignore-backpressure");
 }
-export function findProtocolError(error) {
-    if (isProtocolError(error)) {
-        return error;
-    }
-    if (error instanceof Error && error.cause) {
-        return findProtocolError(error.cause);
-    }
-    return undefined;
+function findProtocolError(error) {
+  if (isProtocolError(error)) {
+    return error;
+  }
+  if (error instanceof Error && error.cause) {
+    return findProtocolError(error.cause);
+  }
+  return void 0;
+}
+function hasErrno(error) {
+  return Boolean(
+    error && typeof error === "object" && "errno" in error && typeof error.errno !== "undefined"
+  );
+}
+const TRANSIENT_SOCKET_ERROR_CODES = /* @__PURE__ */ new Set([
+  "EPIPE",
+  "ECONNRESET",
+  "ECANCELED"
+]);
+const TRANSIENT_SOCKET_MESSAGE_PATTERNS = [
+  "socket was closed while data was being compressed"
+];
+function hasTransientSocketCode(error) {
+  if (!error || typeof error !== "object") {
+    return false;
+  }
+  const maybeCode = "code" in error ? String(error.code) : void 0;
+  return Boolean(
+    maybeCode && TRANSIENT_SOCKET_ERROR_CODES.has(maybeCode.toUpperCase())
+  );
+}
+function isTransientSocketMessage(message) {
+  if (!message) {
+    return false;
+  }
+  const lower = message.toLowerCase();
+  return TRANSIENT_SOCKET_MESSAGE_PATTERNS.some(
+    (pattern) => lower.includes(pattern)
+  );
 }
-//# sourceMappingURL=connection.js.map
+export {
+  Connection,
+  findProtocolError,
+  send,
+  sendError
+};
+//# sourceMappingURL=connection.js.map

package/out/zero-cache/src/workers/connection.js.map

@@ -1 +1 @@
-{"version":3,"file":"connection.js","sourceRoot":"","sources":["../../../../../zero-cache/src/workers/connection.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAC,MAAM,aAAa,CAAC;AAEzD,OAAO,SAAS,EAAE,EAAC,qBAAqB,EAAC,MAAM,IAAI,CAAC;AACpD,OAAO,EAAC,MAAM,EAAC,MAAM,gCAAgC,CAAC;AACtD,OAAO,KAAK,MAAM,MAAM,+BAA+B,CAAC;AAGxD,OAAO,EAAC,SAAS,EAAC,MAAM,0CAA0C,CAAC;AAEnE,OAAO,EACL,kCAAkC,EAClC,gBAAgB,GACjB,MAAM,gDAAgD,CAAC;AACxD,OAAO,EAAC,cAAc,EAAgB,MAAM,kCAAkC,CAAC;AAC/E,OAAO,EACL,sBAAsB,EACtB,WAAW,EACX,qBAAqB,GACtB,MAAM,8BAA8B,CAAC;AAGtC,OAAO,EACL,eAAe,GAEhB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAC,WAAW,EAAC,MAAM,4CAA4C,CAAC;AA0BvE,+EAA+E;AAC/E,8EAA8E;AAC9E,8EAA8E;AAC9E,0EAA0E;AAC1E,0EAA0E;AAC1E,6EAA6E;AAC7E,4DAA4D;AAC5D,EAAE;AACF,oEAAoE;AACpE,8DAA8D;AAC9D,qKAAqK;AACrK,MAAM,0BAA0B,GAAG,KAAK,CAAC;AAEzC;;;;;;;GAOG;AACH,MAAM,OAAO,UAAU;IACZ,GAAG,CAAY;IACf,KAAK,CAAS;IACd,gBAAgB,CAAS;IACzB,GAAG,CAAa;IAChB,QAAQ,CAAa;IACrB,eAAe,CAAiB;IAChC,mBAAmB,CAA6B;IAEzD,yBAAyB,CAAiC;IAC1D,qBAAqB,CAAiC;IACtD,OAAO,GAAG,KAAK,CAAC;IAEhB,YACE,EAAc,EACd,aAA4B,EAC5B,EAAa,EACb,cAA8B,EAC9B,OAAmB;QAEnB,MAAM,EAAC,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,eAAe,EAAC,GAAG,aAAa,CAAC;QACvE,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;QAEtC,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC;QACd,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;QAExC,IAAI,CAAC,GAAG,GAAG,EAAE;aACV,WAAW,CAAC,YAAY,CAAC;aACzB,WAAW,CAAC,UAAU,EAAE,QAAQ,CAAC;aACjC,WAAW,CAAC,eAAe,EAAE,aAAa,CAAC;aAC3C,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,gBAAgB,CAAC,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QAExB,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QAEtD,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,mBAAmB,GAAG,WAAW,CACpC,IAAI,CAAC,cAAc,EACnB,0BAA0B,GAAG,CAAC,CAC/B,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,IAAI;QACF,IACE,IAAI,CAAC,gBAAgB,GAAG,gBAAgB;YACxC,IAAI,CAAC,gBAAgB,GAAG,kCAAkC,EAC1D,CAAC;YACD,IAAI,CAAC,eAAe,CAAC;gBACnB,IAAI,EAAE,SAAS,CAAC,mBAAmB;gBACnC,OAAO,EAAE,+BAA+B,gBAAgB,0BACtD,IAAI,CAAC,gBACP,SACE,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QACxD,sCAAsC;gBACtC,MAAM,EAAE,WAAW,CAAC,SAAS;aAC9B,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,gBAAgB,GAAqB;gBACzC,WAAW;gBACX,EAAC,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAC;aAC1C,CAAC;YACF,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,qBAAqB,CAAC,CAAC;YACnD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,MAAc,EAAE,GAAG,IAAe;QACtC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,uBAAuB,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACzD,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACzD,IAAI,CAAC,yBAAyB,EAAE,MAAM,EAAE,CAAC;QACzC,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;QAC3C,IAAI,CAAC,qBAAqB,EAAE,MAAM,EAAE,CAAC;QACrC,IAAI,CAAC,qBAAqB,GAAG,SAAS,CAAC;QACvC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChB,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,KAAK,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;YAC5C,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QACnB,CAAC;QACD,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAEvC,gCAAgC;QAChC,mDAAmD;IACrD,CAAC;IAED,oBAAoB,CAAC,iBAAyB;QAC5C,OAAO,IAAI,CAAC,cAAc,CAAC,EAAC,IAAI,EAAE,iBAAiB,EAAC,CAAC,CAAC;IACxD,CAAC;IAED,cAAc,GAAG,KAAK,EAAE,KAAmB,EAAE,EAAE;QAC7C,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,wCAAwC,EAAE,IAAI,CAAC,CAAC;YACjE,OAAO;QACT,CAAC;QAED,IAAI,GAAG,CAAC;QACR,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC/B,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,4BAA4B,IAAI,MAAM,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACnE,IAAI,CAAC,eAAe,CAClB;gBACE,IAAI,EAAE,SAAS,CAAC,cAAc;gBAC9B,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;gBAClB,MAAM,EAAE,WAAW,CAAC,SAAS;aAC9B,EACD,CAAC,CACF,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;YACvB,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,qBAAqB,CAAC,CAAC;gBAC/C,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;YAC7D,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;gBACvB,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC,CAAC;IAEF,oBAAoB,CAAC,MAAqB;QACxC,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;YACpB,KAAK,OAAO;gBACV,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACnC,MAAM;YACR,KAAK,IAAI;gBACP,MAAM;YACR,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;oBACtB,KAAK,YAAY;wBACf,MAAM,CACJ,IAAI,CAAC,yBAAyB,KAAK,SAAS,EAC5C,kDAAkD,CACnD,CAAC;wBACF,IAAI,CAAC,yBAAyB,GAAG,MAAM,CAAC,MAAM,CAAC;wBAC/C,MAAM;oBACR,KAAK,QAAQ;wBACX,MAAM,CACJ,IAAI,CAAC,qBAAqB,KAAK,SAAS,EACxC,kDAAkD,CACnD,CAAC;wBACF,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,MAAM,CAAC;wBAC3C,MAAM;gBACV,CAAC;gBACD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACnC,MAAM;YACR,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClC,KAAK,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,YAAY,GAAG,CAAC,CAAa,EAAE,EAAE;QAC/B,MAAM,EAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAC,GAAG,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,CAAC,uBAAuB,EAAE,EAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAC,CAAC,CAAC;IAChE,CAAC,CAAC;IAEF,YAAY,GAAG,CAAC,CAAa,EAAE,EAAE;QAC/B,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;IAChE,CAAC,CAAC;IAEF,aAAa;QACX,QAAQ,CACN,qBAAqB,CAAC,IAAI,CAAC,GAAG,CAAC,EAC/B,IAAI,QAAQ,CAAC;YACX,KAAK,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE;gBACnC,IAAI,CAAC,cAAc,CAAC,EAAC,IAAI,EAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,EAAE,QAAQ,CAAC,CAAC;YAC/D,CAAC;SACF,CAAC;QACF,mEAAmE;QACnE,gEAAgE;QAChE,eAAe;QACf,GAAG,EAAE,GAAE,CAAC,CACT,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,cAAkC;QAC/C,oEAAoE;QACpE,oEAAoE;QACpE,mEAAmE;QACnE,mEAAmE;QACnE,4DAA4D;QAC5D,QAAQ,CACN,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,EAC7B,IAAI,QAAQ,CAAC;YACX,UAAU,EAAE,IAAI;YAChB,KAAK,EAAE,CAAC,UAAsB,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,CACrD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC;SAClC,CAAC,EACF,CAAC,CAAC,EAAE,CACF,CAAC;YACC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAC1B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,iCAAiC,CAAC,CACpD,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,CAAU;QACzB,MAAM,SAAS,GACb,iBAAiB,CAAC,CAAC,CAAC,EAAE,SAAS,IAAI,qBAAqB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAExE,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC;IAED,eAAe,CAAC,SAAoB,EAAE,MAAgB;QACpD,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,CACR,GAAG,SAAS,CAAC,IAAI,KAAK,SAAS,CAAC,MAAM,MAAM,SAAS,CAAC,OAAO,EAAE,EAC/D,SAAS,CACV,CAAC;IACJ,CAAC;IAED,sBAAsB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAEpC,cAAc,GAAG,GAAG,EAAE;QACpB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,sBAAsB,GAAG,0BAA0B,EAAE,CAAC;YAC1E,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,uBAAuB,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,qBAAqB,CAAC,CAAC;QACjD,CAAC;IACH,CAAC,CAAC;IAEF,IAAI,CACF,IAAgB,EAChB,QAAgE;QAEhE,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzC,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IAClD,CAAC;IAED,SAAS,CAAC,SAAoB,EAAE,MAAgB;QAC9C,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACnD,CAAC;CACF;AAMD,iCAAiC;AACjC,MAAM,UAAU,IAAI,CAClB,EAAc,EACd,EAAiB,EACjB,IAAgB,EAChB,QAAgE;IAEhE,IAAI,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;QACrC,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EACpB,QAAQ,KAAK,qBAAqB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAC1D,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,EAAE,CAAC,KAAK,EAAE,CAAC,2CAA2C,EAAE,CAAC,UAAU,GAAG,EAAE;YACtE,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QACH,IAAI,QAAQ,KAAK,qBAAqB,EAAE,CAAC;YACvC,QAAQ,CACN,IAAI,sBAAsB,CACxB;gBACE,IAAI,EAAE,SAAS,CAAC,QAAQ;gBACxB,OAAO,EAAE,kBAAkB;gBAC3B,MAAM,EAAE,WAAW,CAAC,SAAS;aAC9B,EACD,MAAM,CACP,CACF,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,SAAS,CACvB,EAAc,EACd,EAAa,EACb,SAAoB,EACpB,MAAgB;IAEhB,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,WAAW,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IACvD,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,4BAA4B,EAAE,SAAS,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC;IACtE,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE,qBAAqB,CAAC,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAAc;IAC9C,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAC1C,OAAO,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}
+{"version":3,"file":"connection.js","sources":["../../../../../zero-cache/src/workers/connection.ts"],"sourcesContent":["import type {LogContext, LogLevel} from '@rocicorp/logger';\nimport {pipeline, Readable, Writable} from 'node:stream';\nimport type {CloseEvent, Data, ErrorEvent} from 'ws';\nimport WebSocket, {createWebSocketStream} from 'ws';\nimport {assert} from '../../../shared/src/asserts.ts';\nimport * as valita from '../../../shared/src/valita.ts';\nimport type {ConnectedMessage} from '../../../zero-protocol/src/connect.ts';\nimport type {Downstream} from '../../../zero-protocol/src/down.ts';\nimport {ErrorKind} from '../../../zero-protocol/src/error-kind.ts';\nimport type {ErrorBody} from '../../../zero-protocol/src/error.ts';\nimport {\n  MIN_SERVER_SUPPORTED_SYNC_PROTOCOL,\n  PROTOCOL_VERSION,\n} from '../../../zero-protocol/src/protocol-version.ts';\nimport {upstreamSchema, type Upstream} from '../../../zero-protocol/src/up.ts';\nimport {\n  ProtocolErrorWithLevel,\n  getLogLevel,\n  wrapWithProtocolError,\n} from '../types/error-with-level.ts';\nimport type {Source} from '../types/streams.ts';\nimport type {ConnectParams} from './connect-params.ts';\nimport {\n  isProtocolError,\n  type ProtocolError,\n} from '../../../zero-protocol/src/error.ts';\nimport {ErrorOrigin} from '../../../zero-protocol/src/error-origin.ts';\n\nexport type HandlerResult =\n  | {\n      type: 'ok';\n    }\n  | {\n      type: 'fatal';\n      error: ErrorBody;\n    }\n  | {\n      type: 'transient';\n      errors: ErrorBody[];\n    }\n  | StreamResult;\n\nexport type StreamResult = {\n  type: 'stream';\n  source: 'viewSyncer' | 'pusher';\n  stream: Source<Downstream>;\n};\n\nexport interface MessageHandler {\n  handleMessage(msg: Upstream): Promise<HandlerResult[]>;\n}\n\n// Ensures that a downstream message is sent at least every interval, sending a\n// 'pong' if necessary. This is set to be slightly longer than the client-side\n// PING_INTERVAL of 5 seconds, so that in the common case, 'pong's are sent in\n// response to client-initiated 'ping's. However, if the inbound stream is\n// backed up because a command is taking a long time to process, the pings\n// will be stuck in the queue (i.e. back-pressured), in which case pongs will\n// be manually sent to notify the client of server liveness.\n//\n// This is equivalent to what is done for Postgres keepalives on the\n// replication stream (which can similarly be back-pressured):\n// https://github.com/rocicorp/mono/blob/f98cb369a2dbb15650328859c732db358f187ef0/packages/zero-cache/src/services/change-source/pg/logical-replication/stream.ts#L21\nconst DOWNSTREAM_MSG_INTERVAL_MS = 6_000;\n\n/**\n * Represents a connection between the client and server.\n *\n * Handles incoming messages on the connection and dispatches\n * them to the correct service.\n *\n * Listens to the ViewSyncer and sends messages to the client.\n */\nexport class Connection {\n  readonly #ws: WebSocket;\n  readonly #wsID: string;\n  readonly #protocolVersion: number;\n  readonly #lc: LogContext;\n  readonly #onClose: () => void;\n  readonly #messageHandler: MessageHandler;\n  readonly #downstreamMsgTimer: NodeJS.Timeout | undefined;\n\n  #viewSyncerOutboundStream: Source<Downstream> | undefined;\n  #pusherOutboundStream: Source<Downstream> | undefined;\n  #closed = false;\n\n  constructor(\n    lc: LogContext,\n    connectParams: ConnectParams,\n    ws: WebSocket,\n    messageHandler: MessageHandler,\n    onClose: () => void,\n  ) {\n    const {clientGroupID, clientID, wsID, protocolVersion} = connectParams;\n    this.#messageHandler = messageHandler;\n\n    this.#ws = ws;\n    this.#wsID = wsID;\n    this.#protocolVersion = protocolVersion;\n\n    this.#lc = lc\n      .withContext('connection')\n      .withContext('clientID', clientID)\n      .withContext('clientGroupID', clientGroupID)\n      .withContext('wsID', wsID);\n    this.#lc.debug?.('new connection');\n    this.#onClose = onClose;\n\n    this.#ws.addEventListener('close', this.#handleClose);\n    this.#ws.addEventListener('error', this.#handleError);\n\n    this.#proxyInbound();\n    this.#downstreamMsgTimer = setInterval(\n      this.#maybeSendPong,\n      DOWNSTREAM_MSG_INTERVAL_MS / 2,\n    );\n  }\n\n  /**\n   * Checks the protocol version and errors for unsupported protocols,\n   * sending the initial `connected` response on success.\n   *\n   * This is early in the connection lifecycle because {@link #handleMessage}\n   * will only parse messages with schema(s) of supported protocol versions.\n   */\n  init(): boolean {\n    if (\n      this.#protocolVersion > PROTOCOL_VERSION ||\n      this.#protocolVersion < MIN_SERVER_SUPPORTED_SYNC_PROTOCOL\n    ) {\n      this.#closeWithError({\n        kind: ErrorKind.VersionNotSupported,\n        message: `server is at sync protocol v${PROTOCOL_VERSION} and does not support v${\n          this.#protocolVersion\n        }. The ${\n          this.#protocolVersion > PROTOCOL_VERSION ? 'server' : 'client'\n        } must be updated to a newer release.`,\n        origin: ErrorOrigin.ZeroCache,\n      });\n    } else {\n      const connectedMessage: ConnectedMessage = [\n        'connected',\n        {wsid: this.#wsID, timestamp: Date.now()},\n      ];\n      this.send(connectedMessage, 'ignore-backpressure');\n      return true;\n    }\n    return false;\n  }\n\n  close(reason: string, ...args: unknown[]) {\n    if (this.#closed) {\n      return;\n    }\n    this.#closed = true;\n    this.#lc.info?.(`closing connection: ${reason}`, ...args);\n    this.#ws.removeEventListener('close', this.#handleClose);\n    this.#ws.removeEventListener('error', this.#handleError);\n    this.#viewSyncerOutboundStream?.cancel();\n    this.#viewSyncerOutboundStream = undefined;\n    this.#pusherOutboundStream?.cancel();\n    this.#pusherOutboundStream = undefined;\n    this.#onClose();\n    if (this.#ws.readyState !== this.#ws.CLOSED) {\n      this.#ws.close();\n    }\n    clearTimeout(this.#downstreamMsgTimer);\n\n    // spin down services if we have\n    // no more client connections for the client group?\n  }\n\n  handleInitConnection(initConnectionMsg: string) {\n    return this.#handleMessage({data: initConnectionMsg});\n  }\n\n  #handleMessage = async (event: {data: Data}) => {\n    const data = event.data.toString();\n    if (this.#closed) {\n      this.#lc.debug?.('Ignoring message received after closed', data);\n      return;\n    }\n\n    let msg;\n    try {\n      const value = JSON.parse(data);\n      msg = valita.parse(value, upstreamSchema);\n    } catch (e) {\n      this.#lc.warn?.(`failed to parse message \"${data}\": ${String(e)}`);\n      this.#closeWithError(\n        {\n          kind: ErrorKind.InvalidMessage,\n          message: String(e),\n          origin: ErrorOrigin.ZeroCache,\n        },\n        e,\n      );\n      return;\n    }\n\n    try {\n      const msgType = msg[0];\n      if (msgType === 'ping') {\n        this.send(['pong', {}], 'ignore-backpressure');\n        return;\n      }\n\n      const result = await this.#messageHandler.handleMessage(msg);\n      for (const r of result) {\n        this.#handleMessageResult(r);\n      }\n    } catch (e) {\n      this.#closeWithThrown(e);\n    }\n  };\n\n  #handleMessageResult(result: HandlerResult): void {\n    switch (result.type) {\n      case 'fatal':\n        this.#closeWithError(result.error);\n        break;\n      case 'ok':\n        break;\n      case 'stream': {\n        switch (result.source) {\n          case 'viewSyncer':\n            assert(\n              this.#viewSyncerOutboundStream === undefined,\n              'Outbound stream already set for this connection!',\n            );\n            this.#viewSyncerOutboundStream = result.stream;\n            break;\n          case 'pusher':\n            assert(\n              this.#pusherOutboundStream === undefined,\n              'Outbound stream already set for this connection!',\n            );\n            this.#pusherOutboundStream = result.stream;\n            break;\n        }\n        this.#proxyOutbound(result.stream);\n        break;\n      }\n      case 'transient': {\n        for (const error of result.errors) {\n          void this.sendError(error);\n        }\n      }\n    }\n  }\n\n  #handleClose = (e: CloseEvent) => {\n    const {code, reason, wasClean} = e;\n    this.close('WebSocket close event', {code, reason, wasClean});\n  };\n\n  #handleError = (e: ErrorEvent) => {\n    this.#lc.error?.('WebSocket error event', e.message, e.error);\n  };\n\n  #proxyInbound() {\n    pipeline(\n      createWebSocketStream(this.#ws),\n      new Writable({\n        write: (data, _encoding, callback) => {\n          this.#handleMessage({data}).then(() => callback(), callback);\n        },\n      }),\n      // The done callback is not used, as #handleClose and #handleError,\n      // configured on the underlying WebSocket, provide more complete\n      // information.\n      () => {},\n    );\n  }\n\n  #proxyOutbound(outboundStream: Source<Downstream>) {\n    // Note: createWebSocketStream() is avoided here in order to control\n    //       exception handling with #closeWithThrown(). If the Writable\n    //       from createWebSocketStream() were instead used, exceptions\n    //       from the outboundStream result in the Writable closing the\n    //       the websocket before the error message can be sent.\n    pipeline(\n      Readable.from(outboundStream),\n      new Writable({\n        objectMode: true,\n        write: (downstream: Downstream, _encoding, callback) =>\n          this.send(downstream, callback),\n      }),\n      e =>\n        e\n          ? this.#closeWithThrown(e)\n          : this.close(`downstream closed by ViewSyncer`),\n    );\n  }\n\n  #closeWithThrown(e: unknown) {\n    const errorBody =\n      findProtocolError(e)?.errorBody ?? wrapWithProtocolError(e).errorBody;\n\n    this.#closeWithError(errorBody, e);\n  }\n\n  #closeWithError(errorBody: ErrorBody, thrown?: unknown) {\n    this.sendError(errorBody, thrown);\n    this.close(\n      `${errorBody.kind} (${errorBody.origin}): ${errorBody.message}`,\n      errorBody,\n    );\n  }\n\n  #lastDownstreamMsgTime = Date.now();\n\n  #maybeSendPong = () => {\n    if (Date.now() - this.#lastDownstreamMsgTime > DOWNSTREAM_MSG_INTERVAL_MS) {\n      this.#lc.debug?.('manually sending pong');\n      this.send(['pong', {}], 'ignore-backpressure');\n    }\n  };\n\n  send(\n    data: Downstream,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n  ) {\n    this.#lastDownstreamMsgTime = Date.now();\n    return send(this.#lc, this.#ws, data, callback);\n  }\n\n  sendError(errorBody: ErrorBody, thrown?: unknown) {\n    sendError(this.#lc, this.#ws, errorBody, thrown);\n  }\n}\n\nexport type WebSocketLike = Pick<WebSocket, 'readyState'> & {\n  send(data: string, cb?: (err?: Error) => void): void;\n};\n\n// Exported for testing purposes.\nexport function send(\n  lc: LogContext,\n  ws: WebSocketLike,\n  data: Downstream,\n  callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n) {\n  if (ws.readyState === WebSocket.OPEN) {\n    ws.send(\n      JSON.stringify(data),\n      callback === 'ignore-backpressure' ? undefined : callback,\n    );\n  } else {\n    lc.debug?.(`Dropping outbound message on ws (state: ${ws.readyState})`, {\n      dropped: data,\n    });\n    if (callback !== 'ignore-backpressure') {\n      callback(\n        new ProtocolErrorWithLevel(\n          {\n            kind: ErrorKind.Internal,\n            message: 'WebSocket closed',\n            origin: ErrorOrigin.ZeroCache,\n          },\n          'info',\n        ),\n      );\n    }\n  }\n}\n\nexport function sendError(\n  lc: LogContext,\n  ws: WebSocket,\n  errorBody: ErrorBody,\n  thrown?: unknown,\n) {\n  lc = lc.withContext('errorKind', errorBody.kind);\n\n  let logLevel: LogLevel;\n\n  // If the thrown error is a ProtocolErrorWithLevel, its explicit logLevel takes precedence\n  if (thrown instanceof ProtocolErrorWithLevel) {\n    logLevel = thrown.logLevel;\n  }\n  // Errors with errno or transient socket codes are low-level, transient I/O issues\n  // (e.g., EPIPE, ECONNRESET) and should be warnings, not errors\n  else if (\n    hasErrno(thrown) ||\n    hasTransientSocketCode(thrown) ||\n    isTransientSocketMessage(errorBody.message)\n  ) {\n    logLevel = 'warn';\n  }\n  // Fallback: check errorBody.kind for errors that weren't thrown as ProtocolErrorWithLevel\n  else if (\n    errorBody.kind === ErrorKind.ClientNotFound ||\n    errorBody.kind === ErrorKind.TransformFailed\n  ) {\n    logLevel = 'warn';\n  } else {\n    logLevel = thrown ? getLogLevel(thrown) : 'info';\n  }\n\n  lc[logLevel]?.('Sending error on WebSocket', errorBody, thrown ?? '');\n  send(lc, ws, ['error', errorBody], 'ignore-backpressure');\n}\n\nexport function findProtocolError(error: unknown): ProtocolError | undefined {\n  if (isProtocolError(error)) {\n    return error;\n  }\n  if (error instanceof Error && error.cause) {\n    return findProtocolError(error.cause);\n  }\n  return undefined;\n}\n\nfunction hasErrno(error: unknown): boolean {\n  return Boolean(\n    error &&\n      typeof error === 'object' &&\n      'errno' in error &&\n      typeof (error as {errno: unknown}).errno !== 'undefined',\n  );\n}\n\n// System error codes that indicate transient socket conditions.\n// These are checked via the `code` property on errors.\nconst TRANSIENT_SOCKET_ERROR_CODES = new Set([\n  'EPIPE',\n  'ECONNRESET',\n  'ECANCELED',\n]);\n\n// Error messages that indicate transient socket conditions but don't have\n// standard error codes (e.g., WebSocket library errors).\nconst TRANSIENT_SOCKET_MESSAGE_PATTERNS = [\n  'socket was closed while data was being compressed',\n];\n\nfunction hasTransientSocketCode(error: unknown): boolean {\n  if (!error || typeof error !== 'object') {\n    return false;\n  }\n  const maybeCode =\n    'code' in error ? String((error as {code?: unknown}).code) : undefined;\n  return Boolean(\n    maybeCode && TRANSIENT_SOCKET_ERROR_CODES.has(maybeCode.toUpperCase()),\n  );\n}\n\nfunction isTransientSocketMessage(message: string | undefined): boolean {\n  if (!message) {\n    return false;\n  }\n  const lower = message.toLowerCase();\n  return TRANSIENT_SOCKET_MESSAGE_PATTERNS.some(pattern =>\n    lower.includes(pattern),\n  );\n}\n"],"names":["ErrorKind.VersionNotSupported","ErrorOrigin.ZeroCache","valita.parse","ErrorKind.InvalidMessage","ErrorKind.Internal","ErrorKind.ClientNotFound","ErrorKind.TransformFailed"],"mappings":";;;;;;;;;;AA+DA,MAAM,6BAA6B;AAU5B,MAAM,WAAW;AAAA,EACb;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAET;AAAA,EACA;AAAA,EACA,UAAU;AAAA,EAEV,YACE,IACA,eACA,IACA,gBACA,SACA;AACA,UAAM,EAAC,eAAe,UAAU,MAAM,oBAAmB;AACzD,SAAK,kBAAkB;AAEvB,SAAK,MAAM;AACX,SAAK,QAAQ;AACb,SAAK,mBAAmB;AAExB,SAAK,MAAM,GACR,YAAY,YAAY,EACxB,YAAY,YAAY,QAAQ,EAChC,YAAY,iBAAiB,aAAa,EAC1C,YAAY,QAAQ,IAAI;AAC3B,SAAK,IAAI,QAAQ,gBAAgB;AACjC,SAAK,WAAW;AAEhB,SAAK,IAAI,iBAAiB,SAAS,KAAK,YAAY;AACpD,SAAK,IAAI,iBAAiB,SAAS,KAAK,YAAY;AAEpD,SAAK,cAAA;AACL,SAAK,sBAAsB;AAAA,MACzB,KAAK;AAAA,MACL,6BAA6B;AAAA,IAAA;AAAA,EAEjC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,OAAgB;AACd,QACE,KAAK,mBAAmB,oBACxB,KAAK,mBAAmB,oCACxB;AACA,WAAK,gBAAgB;AAAA,QACnB,MAAMA;AAAAA,QACN,SAAS,+BAA+B,gBAAgB,0BACtD,KAAK,gBACP,SACE,KAAK,mBAAmB,mBAAmB,WAAW,QACxD;AAAA,QACA,QAAQC;AAAAA,MAAY,CACrB;AAAA,IACH,OAAO;AACL,YAAM,mBAAqC;AAAA,QACzC;AAAA,QACA,EAAC,MAAM,KAAK,OAAO,WAAW,KAAK,MAAI;AAAA,MAAC;AAE1C,WAAK,KAAK,kBAAkB,qBAAqB;AACjD,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,WAAmB,MAAiB;AACxC,QAAI,KAAK,SAAS;AAChB;AAAA,IACF;AACA,SAAK,UAAU;AACf,SAAK,IAAI,OAAO,uBAAuB,MAAM,IAAI,GAAG,IAAI;AACxD,SAAK,IAAI,oBAAoB,SAAS,KAAK,YAAY;AACvD,SAAK,IAAI,oBAAoB,SAAS,KAAK,YAAY;AACvD,SAAK,2BAA2B,OAAA;AAChC,SAAK,4BAA4B;AACjC,SAAK,uBAAuB,OAAA;AAC5B,SAAK,wBAAwB;AAC7B,SAAK,SAAA;AACL,QAAI,KAAK,IAAI,eAAe,KAAK,IAAI,QAAQ;AAC3C,WAAK,IAAI,MAAA;AAAA,IACX;AACA,iBAAa,KAAK,mBAAmB;AAAA,EAIvC;AAAA,EAEA,qBAAqB,mBAA2B;AAC9C,WAAO,KAAK,eAAe,EAAC,MAAM,mBAAkB;AAAA,EACtD;AAAA,EAEA,iBAAiB,OAAO,UAAwB;AAC9C,UAAM,OAAO,MAAM,KAAK,SAAA;AACxB,QAAI,KAAK,SAAS;AAChB,WAAK,IAAI,QAAQ,0CAA0C,IAAI;AAC/D;AAAA,IACF;AAEA,QAAI;AACJ,QAAI;AACF,YAAM,QAAQ,KAAK,MAAM,IAAI;AAC7B,YAAMC,MAAa,OAAO,cAAc;AAAA,IAC1C,SAAS,GAAG;AACV,WAAK,IAAI,OAAO,4BAA4B,IAAI,MAAM,OAAO,CAAC,CAAC,EAAE;AACjE,WAAK;AAAA,QACH;AAAA,UACE,MAAMC;AAAAA,UACN,SAAS,OAAO,CAAC;AAAA,UACjB,QAAQF;AAAAA,QAAY;AAAA,QAEtB;AAAA,MAAA;AAEF;AAAA,IACF;AAEA,QAAI;AACF,YAAM,UAAU,IAAI,CAAC;AACrB,UAAI,YAAY,QAAQ;AACtB,aAAK,KAAK,CAAC,QAAQ,CAAA,CAAE,GAAG,qBAAqB;AAC7C;AAAA,MACF;AAEA,YAAM,SAAS,MAAM,KAAK,gBAAgB,cAAc,GAAG;AAC3D,iBAAW,KAAK,QAAQ;AACtB,aAAK,qBAAqB,CAAC;AAAA,MAC7B;AAAA,IACF,SAAS,GAAG;AACV,WAAK,iBAAiB,CAAC;AAAA,IACzB;AAAA,EACF;AAAA,EAEA,qBAAqB,QAA6B;AAChD,YAAQ,OAAO,MAAA;AAAA,MACb,KAAK;AACH,aAAK,gBAAgB,OAAO,KAAK;AACjC;AAAA,MACF,KAAK;AACH;AAAA,MACF,KAAK,UAAU;AACb,gBAAQ,OAAO,QAAA;AAAA,UACb,KAAK;AACH;AAAA,cACE,KAAK,8BAA8B;AAAA,cACnC;AAAA,YAAA;AAEF,iBAAK,4BAA4B,OAAO;AACxC;AAAA,UACF,KAAK;AACH;AAAA,cACE,KAAK,0BAA0B;AAAA,cAC/B;AAAA,YAAA;AAEF,iBAAK,wBAAwB,OAAO;AACpC;AAAA,QAAA;AAEJ,aAAK,eAAe,OAAO,MAAM;AACjC;AAAA,MACF;AAAA,MACA,KAAK,aAAa;AAChB,mBAAW,SAAS,OAAO,QAAQ;AACjC,eAAK,KAAK,UAAU,KAAK;AAAA,QAC3B;AAAA,MACF;AAAA,IAAA;AAAA,EAEJ;AAAA,EAEA,eAAe,CAAC,MAAkB;AAChC,UAAM,EAAC,MAAM,QAAQ,SAAA,IAAY;AACjC,SAAK,MAAM,yBAAyB,EAAC,MAAM,QAAQ,UAAS;AAAA,EAC9D;AAAA,EAEA,eAAe,CAAC,MAAkB;AAChC,SAAK,IAAI,QAAQ,yBAAyB,EAAE,SAAS,EAAE,KAAK;AAAA,EAC9D;AAAA,EAEA,gBAAgB;AACd;AAAA,MACE,sBAAsB,KAAK,GAAG;AAAA,MAC9B,IAAI,SAAS;AAAA,QACX,OAAO,CAAC,MAAM,WAAW,aAAa;AACpC,eAAK,eAAe,EAAC,KAAA,CAAK,EAAE,KAAK,MAAM,SAAA,GAAY,QAAQ;AAAA,QAC7D;AAAA,MAAA,CACD;AAAA;AAAA;AAAA;AAAA,MAID,MAAM;AAAA,MAAC;AAAA,IAAA;AAAA,EAEX;AAAA,EAEA,eAAe,gBAAoC;AAMjD;AAAA,MACE,SAAS,KAAK,cAAc;AAAA,MAC5B,IAAI,SAAS;AAAA,QACX,YAAY;AAAA,QACZ,OAAO,CAAC,YAAwB,WAAW,aACzC,KAAK,KAAK,YAAY,QAAQ;AAAA,MAAA,CACjC;AAAA,MACD,CAAA,MACE,IACI,KAAK,iBAAiB,CAAC,IACvB,KAAK,MAAM,iCAAiC;AAAA,IAAA;AAAA,EAEtD;AAAA,EAEA,iBAAiB,GAAY;AAC3B,UAAM,YACJ,kBAAkB,CAAC,GAAG,aAAa,sBAAsB,CAAC,EAAE;AAE9D,SAAK,gBAAgB,WAAW,CAAC;AAAA,EACnC;AAAA,EAEA,gBAAgB,WAAsB,QAAkB;AACtD,SAAK,UAAU,WAAW,MAAM;AAChC,SAAK;AAAA,MACH,GAAG,UAAU,IAAI,KAAK,UAAU,MAAM,MAAM,UAAU,OAAO;AAAA,MAC7D;AAAA,IAAA;AAAA,EAEJ;AAAA,EAEA,yBAAyB,KAAK,IAAA;AAAA,EAE9B,iBAAiB,MAAM;AACrB,QAAI,KAAK,IAAA,IAAQ,KAAK,yBAAyB,4BAA4B;AACzE,WAAK,IAAI,QAAQ,uBAAuB;AACxC,WAAK,KAAK,CAAC,QAAQ,CAAA,CAAE,GAAG,qBAAqB;AAAA,IAC/C;AAAA,EACF;AAAA,EAEA,KACE,MACA,UACA;AACA,SAAK,yBAAyB,KAAK,IAAA;AACnC,WAAO,KAAK,KAAK,KAAK,KAAK,KAAK,MAAM,QAAQ;AAAA,EAChD;AAAA,EAEA,UAAU,WAAsB,QAAkB;AAChD,cAAU,KAAK,KAAK,KAAK,KAAK,WAAW,MAAM;AAAA,EACjD;AACF;AAOO,SAAS,KACd,IACA,IACA,MACA,UACA;AACA,MAAI,GAAG,eAAe,UAAU,MAAM;AACpC,OAAG;AAAA,MACD,KAAK,UAAU,IAAI;AAAA,MACnB,aAAa,wBAAwB,SAAY;AAAA,IAAA;AAAA,EAErD,OAAO;AACL,OAAG,QAAQ,2CAA2C,GAAG,UAAU,KAAK;AAAA,MACtE,SAAS;AAAA,IAAA,CACV;AACD,QAAI,aAAa,uBAAuB;AACtC;AAAA,QACE,IAAI;AAAA,UACF;AAAA,YACE,MAAMG;AAAAA,YACN,SAAS;AAAA,YACT,QAAQH;AAAAA,UAAY;AAAA,UAEtB;AAAA,QAAA;AAAA,MACF;AAAA,IAEJ;AAAA,EACF;AACF;AAEO,SAAS,UACd,IACA,IACA,WACA,QACA;AACA,OAAK,GAAG,YAAY,aAAa,UAAU,IAAI;AAE/C,MAAI;AAGJ,MAAI,kBAAkB,wBAAwB;AAC5C,eAAW,OAAO;AAAA,EACpB,WAIE,SAAS,MAAM,KACf,uBAAuB,MAAM,KAC7B,yBAAyB,UAAU,OAAO,GAC1C;AACA,eAAW;AAAA,EACb,WAGE,UAAU,SAASI,kBACnB,UAAU,SAASC,iBACnB;AACA,eAAW;AAAA,EACb,OAAO;AACL,eAAW,SAAS,YAAY,MAAM,IAAI;AAAA,EAC5C;AAEA,KAAG,QAAQ,IAAI,8BAA8B,WAAW,UAAU,EAAE;AACpE,OAAK,IAAI,IAAI,CAAC,SAAS,SAAS,GAAG,qBAAqB;AAC1D;AAEO,SAAS,kBAAkB,OAA2C;AAC3E,MAAI,gBAAgB,KAAK,GAAG;AAC1B,WAAO;AAAA,EACT;AACA,MAAI,iBAAiB,SAAS,MAAM,OAAO;AACzC,WAAO,kBAAkB,MAAM,KAAK;AAAA,EACtC;AACA,SAAO;AACT;AAEA,SAAS,SAAS,OAAyB;AACzC,SAAO;AAAA,IACL,SACE,OAAO,UAAU,YACjB,WAAW,SACX,OAAQ,MAA2B,UAAU;AAAA,EAAA;AAEnD;AAIA,MAAM,mDAAmC,IAAI;AAAA,EAC3C;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAID,MAAM,oCAAoC;AAAA,EACxC;AACF;AAEA,SAAS,uBAAuB,OAAyB;AACvD,MAAI,CAAC,SAAS,OAAO,UAAU,UAAU;AACvC,WAAO;AAAA,EACT;AACA,QAAM,YACJ,UAAU,QAAQ,OAAQ,MAA2B,IAAI,IAAI;AAC/D,SAAO;AAAA,IACL,aAAa,6BAA6B,IAAI,UAAU,aAAa;AAAA,EAAA;AAEzE;AAEA,SAAS,yBAAyB,SAAsC;AACtE,MAAI,CAAC,SAAS;AACZ,WAAO;AAAA,EACT;AACA,QAAM,QAAQ,QAAQ,YAAA;AACtB,SAAO,kCAAkC;AAAA,IAAK,CAAA,YAC5C,MAAM,SAAS,OAAO;AAAA,EAAA;AAE1B;"}

package/out/zero-cache/src/workers/mutator.js

@@ -0,0 +1,24 @@
+import { resolver } from "@rocicorp/resolver";
+import { pid } from "node:process";
+class Mutator {
+  id = `mutator-${pid}`;
+  #stopped;
+  constructor() {
+    this.#stopped = resolver();
+  }
+  run() {
+    return this.#stopped.promise;
+  }
+  stop() {
+    this.#stopped.resolve();
+    return this.#stopped.promise;
+  }
+  drain() {
+    this.#stopped.resolve();
+    return this.#stopped.promise;
+  }
+}
+export {
+  Mutator
+};
+//# sourceMappingURL=mutator.js.map

package/out/zero-cache/src/workers/mutator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mutator.js","sources":["../../../../../zero-cache/src/workers/mutator.ts"],"sourcesContent":["import {resolver} from '@rocicorp/resolver';\nimport type {SingletonService} from '../services/service.ts';\nimport {pid} from 'node:process';\n\n// TODO:\n// - install websocket receiver\n// - spin up pusher services for each unique client group that connects\nexport class Mutator implements SingletonService {\n  readonly id = `mutator-${pid}`;\n  readonly #stopped;\n\n  constructor() {\n    this.#stopped = resolver();\n  }\n\n  run(): Promise<void> {\n    return this.#stopped.promise;\n  }\n\n  stop(): Promise<void> {\n    this.#stopped.resolve();\n    return this.#stopped.promise;\n  }\n\n  drain(): Promise<void> {\n    this.#stopped.resolve();\n    return this.#stopped.promise;\n  }\n}\n"],"names":[],"mappings":";;AAOO,MAAM,QAAoC;AAAA,EACtC,KAAK,WAAW,GAAG;AAAA,EACnB;AAAA,EAET,cAAc;AACZ,SAAK,WAAW,SAAA;AAAA,EAClB;AAAA,EAEA,MAAqB;AACnB,WAAO,KAAK,SAAS;AAAA,EACvB;AAAA,EAEA,OAAsB;AACpB,SAAK,SAAS,QAAA;AACd,WAAO,KAAK,SAAS;AAAA,EACvB;AAAA,EAEA,QAAuB;AACrB,SAAK,SAAS,QAAA;AACd,WAAO,KAAK,SAAS;AAAA,EACvB;AACF;"}

package/out/zero-cache/src/workers/replicator.d.ts

@@ -1,4 +1,4 @@
-import { LogContext } from '@rocicorp/logger';
+import type { LogContext } from '@rocicorp/logger';
 import * as v from '../../../shared/src/valita.ts';
 import { Database } from '../../../zqlite/src/db.ts';
 import type { ReplicaOptions } from '../config/zero-config.ts';