@replit/river 0.23.6 → 0.23.8

package/dist/transport/impls/ws/client.cjs

@@ -106,7 +106,57 @@ function isAck(controlFlag) {
 }
 
 // logging/log.ts
-var log = void 0;
+var LoggingLevels = {
+  debug: -1,
+  info: 0,
+  warn: 1,
+  error: 2
+};
+var cleanedLogFn = (log) => {
+  return (msg, metadata) => {
+    if (!metadata?.transportMessage) {
+      log(msg, metadata);
+      return;
+    }
+    const { payload, ...rest } = metadata.transportMessage;
+    metadata.transportMessage = rest;
+    log(msg, metadata);
+  };
+};
+var BaseLogger = class {
+  minLevel;
+  output;
+  constructor(output, minLevel = "info") {
+    this.minLevel = minLevel;
+    this.output = output;
+  }
+  debug(msg, metadata) {
+    if (LoggingLevels[this.minLevel] <= LoggingLevels.debug) {
+      this.output(msg, metadata ?? {}, "debug");
+    }
+  }
+  info(msg, metadata) {
+    if (LoggingLevels[this.minLevel] <= LoggingLevels.info) {
+      this.output(msg, metadata ?? {}, "info");
+    }
+  }
+  warn(msg, metadata) {
+    if (LoggingLevels[this.minLevel] <= LoggingLevels.warn) {
+      this.output(msg, metadata ?? {}, "warn");
+    }
+  }
+  error(msg, metadata) {
+    if (LoggingLevels[this.minLevel] <= LoggingLevels.error) {
+      this.output(msg, metadata ?? {}, "error");
+    }
+  }
+};
+var createLogProxy = (log) => ({
+  debug: cleanedLogFn(log.debug.bind(log)),
+  info: cleanedLogFn(log.info.bind(log)),
+  warn: cleanedLogFn(log.warn.bind(log)),
+  error: cleanedLogFn(log.error.bind(log))
+});
 
 // transport/events.ts
 var ProtocolError = {
@@ -149,7 +199,7 @@ var import_nanoid2 = require("nanoid");
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.23.6";
+var version = "0.23.8";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -161,7 +211,7 @@ function getPropagationContext(ctx) {
   return tracing;
 }
 function createSessionTelemetryInfo(session, propagationCtx) {
-  const ctx = propagationCtx ? import_api.propagation.extract(import_api.context.active(), propagationCtx) : import_api.context.active();
+  const parentCtx = propagationCtx ? import_api.propagation.extract(import_api.context.active(), propagationCtx) : import_api.context.active();
   const span = tracer.startSpan(
     `session ${session.id}`,
     {
@@ -172,12 +222,12 @@ function createSessionTelemetryInfo(session, propagationCtx) {
         "river.session.from": session.from
       }
     },
-    ctx
+    parentCtx
   );
+  const ctx = import_api.trace.setSpan(parentCtx, span);
   return { span, ctx };
 }
-function createConnectionTelemetryInfo(connection, sessionSpan) {
-  const ctx = import_api.trace.setSpan(import_api.context.active(), sessionSpan);
+function createConnectionTelemetryInfo(connection, info) {
   const span = tracer.startSpan(
     `connection ${connection.id}`,
     {
@@ -185,10 +235,11 @@ function createConnectionTelemetryInfo(connection, sessionSpan) {
         component: "river",
         "river.connection.id": connection.id
       },
-      links: [{ context: sessionSpan.spanContext() }]
+      links: [{ context: info.span.spanContext() }]
     },
-    ctx
+    info.ctx
   );
+  const ctx = import_api.trace.setSpan(info.ctx, span);
   return { span, ctx };
 }
 var tracer = import_api.trace.getTracer("river", version);
@@ -204,6 +255,17 @@ var Connection = class {
   constructor() {
     this.id = `conn-${nanoid2(12)}`;
   }
+  get loggingMetadata() {
+    const metadata = { connId: this.id };
+    const spanContext = this.telemetry?.span.spanContext();
+    if (spanContext) {
+      metadata.telemetry = {
+        traceId: spanContext.traceId,
+        spanId: spanContext.spanId
+      };
+    }
+    return metadata;
+  }
 };
 var Session = class {
   codec;
@@ -249,6 +311,7 @@ var Session = class {
    * The interval for sending heartbeats.
    */
   heartbeat;
+  log;
   constructor(conn, from, to, options, propagationCtx) {
     this.id = `session-${nanoid2(12)}`;
     this.options = options;
@@ -263,12 +326,20 @@ var Session = class {
     );
     this.telemetry = createSessionTelemetryInfo(this, propagationCtx);
   }
+  bindLogger(log) {
+    this.log = log;
+  }
   get loggingMetadata() {
+    const spanContext = this.telemetry.span.spanContext();
     return {
       clientId: this.from,
       connectedTo: this.to,
       sessionId: this.id,
-      connId: this.connection?.id
+      connId: this.connection?.id,
+      telemetry: {
+        traceId: spanContext.traceId,
+        spanId: spanContext.spanId
+      }
     };
   }
   /**
@@ -281,7 +352,7 @@ var Session = class {
    */
   send(msg) {
     const fullMsg = this.constructMsg(msg);
-    log?.debug(`sending msg`, {
+    this.log?.debug(`sending msg`, {
       ...this.loggingMetadata,
       transportMessage: fullMsg
     });
@@ -289,7 +360,7 @@ var Session = class {
       const ok = this.connection.send(this.codec.toBuffer(fullMsg));
       if (ok)
         return fullMsg.id;
-      log?.info(
+      this.log?.info(
         `failed to send msg to ${fullMsg.to}, connection is probably dead`,
         {
           ...this.loggingMetadata,
@@ -297,8 +368,8 @@ var Session = class {
         }
       );
     } else {
-      log?.info(
-        `failed to send msg to ${fullMsg.to}, connection not ready yet`,
+      this.log?.debug(
+        `buffering msg to ${fullMsg.to}, connection not ready yet`,
         { ...this.loggingMetadata, transportMessage: fullMsg }
       );
     }
@@ -309,7 +380,7 @@ var Session = class {
     const missDuration = misses * this.options.heartbeatIntervalMs;
     if (misses > this.options.heartbeatsUntilDead) {
       if (this.connection) {
-        log?.info(
+        this.log?.info(
           `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,
           this.loggingMetadata
         );
@@ -333,12 +404,12 @@ var Session = class {
     this.ack = 0;
   }
   sendBufferedMessages(conn) {
-    log?.info(`resending ${this.sendBuffer.length} buffered messages`, {
+    this.log?.info(`resending ${this.sendBuffer.length} buffered messages`, {
       ...this.loggingMetadata,
       connId: conn.id
     });
     for (const msg of this.sendBuffer) {
-      log?.debug(`resending msg`, {
+      this.log?.debug(`resending msg`, {
         ...this.loggingMetadata,
         transportMessage: msg,
         connId: conn.id
@@ -350,7 +421,7 @@ var Session = class {
           code: import_api2.SpanStatusCode.ERROR,
           message: errMsg
         });
-        log?.error(errMsg, {
+        this.log?.error(errMsg, {
           ...this.loggingMetadata,
           transportMessage: msg,
           connId: conn.id,
@@ -363,7 +434,7 @@ var Session = class {
   }
   updateBookkeeping(ack, seq) {
     if (seq + 1 < this.ack) {
-      log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {
+      this.log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {
         ...this.loggingMetadata,
         tags: ["invariant-violation"]
       });
@@ -375,7 +446,7 @@ var Session = class {
   closeStaleConnection(conn) {
     if (this.connection === void 0 || this.connection === conn)
       return;
-    log?.info(
+    this.log?.info(
       `closing old inner connection from session to ${this.to}`,
       this.loggingMetadata
     );
@@ -389,7 +460,7 @@ var Session = class {
     this.connection = newConn;
   }
   beginGrace(cb) {
-    log?.info(
+    this.log?.info(
       `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,
       this.loggingMetadata
     );
@@ -623,6 +694,7 @@ var Transport = class {
    * The options for this transport.
    */
   options;
+  log;
   /**
    * Creates a new Transport instance.
    * This should also set up {@link onConnect}, and {@link onDisconnect} listeners.
@@ -637,6 +709,13 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
+  bindLogger(fn, level) {
+    if (typeof fn === "function") {
+      this.log = createLogProxy(new BaseLogger(fn, level));
+      return;
+    }
+    this.log = createLogProxy(fn);
+  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -647,13 +726,15 @@ var Transport = class {
       status: "connect",
       conn
     });
-    conn.telemetry = createConnectionTelemetryInfo(
-      conn,
-      session.telemetry.span
-    );
+    conn.telemetry = createConnectionTelemetryInfo(conn, session.telemetry);
     if (isReconnect) {
       session.replaceWithNewConnection(conn);
-      log?.info(`reconnected to ${connectedTo}`, session.loggingMetadata);
+      this.log?.info(`reconnected to ${connectedTo}`, {
+        ...conn.loggingMetadata,
+        ...session.loggingMetadata,
+        clientId: this.clientId,
+        connectedTo
+      });
     }
   }
   createSession(to, conn, propagationCtx) {
@@ -664,6 +745,9 @@ var Transport = class {
       this.options,
       propagationCtx
     );
+    if (this.log) {
+      session.bindLogger(this.log);
+    }
     this.sessions.set(session.to, session);
     this.eventDispatcher.dispatchEvent("sessionStatus", {
       status: "connect",
@@ -675,7 +759,7 @@ var Transport = class {
     let session = this.sessions.get(to);
     let isReconnect = session !== void 0;
     if (session?.advertisedSessionId !== void 0 && sessionId !== void 0 && session.advertisedSessionId !== sessionId) {
-      log?.info(
+      this.log?.info(
         `session for ${to} already exists but has a different session id (expected: ${session.advertisedSessionId}, got: ${sessionId}), creating a new one`,
         session.loggingMetadata
       );
@@ -685,7 +769,7 @@ var Transport = class {
     }
     if (!session) {
       session = this.createSession(to, conn, propagationCtx);
-      log?.info(
+      this.log?.info(
         `no session for ${to}, created a new one`,
         session.loggingMetadata
       );
@@ -699,7 +783,7 @@ var Transport = class {
     session.close();
     session.telemetry.span.end();
     this.sessions.delete(session.to);
-    log?.info(
+    this.log?.info(
       `session ${session.id} disconnect from ${session.to}`,
       session.loggingMetadata
     );
@@ -730,18 +814,23 @@ var Transport = class {
    * @param msg The message to parse.
    * @returns The parsed message, or null if the message is malformed or invalid.
    */
-  parseMsg(msg) {
+  parseMsg(msg, conn) {
     const parsedMsg = this.codec.fromBuffer(msg);
     if (parsedMsg === null) {
       const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));
-      log?.error(`received malformed msg, killing conn: ${decodedBuffer}`, {
-        clientId: this.clientId
-      });
+      this.log?.error(
+        `received malformed msg, killing conn: ${decodedBuffer}`,
+        {
+          clientId: this.clientId,
+          ...conn.loggingMetadata
+        }
+      );
       return null;
     }
     if (!import_value.Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {
-      log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {
+      this.log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {
         clientId: this.clientId,
+        ...conn.loggingMetadata,
         validationErrors: [
           ...import_value.Value.Errors(OpaqueTransportMessageSchema, parsedMsg)
         ]
@@ -755,34 +844,41 @@ var Transport = class {
    * You generally shouldn't need to override this in downstream transport implementations.
    * @param msg The received message.
    */
-  handleMsg(msg) {
+  handleMsg(msg, conn) {
     if (this.state !== "open")
       return;
     const session = this.sessions.get(msg.from);
     if (!session) {
-      log?.error(`no existing session for ${msg.from}`, {
+      this.log?.error(`received message for unknown session from ${msg.from}`, {
         clientId: this.clientId,
         transportMessage: msg,
+        ...conn.loggingMetadata,
         tags: ["invariant-violation"]
       });
       return;
     }
     session.cancelGrace();
-    log?.debug(`received msg`, {
+    this.log?.debug(`received msg`, {
       clientId: this.clientId,
-      transportMessage: msg
+      transportMessage: msg,
+      ...conn.loggingMetadata
     });
     if (msg.seq !== session.nextExpectedSeq) {
       if (msg.seq < session.nextExpectedSeq) {
-        log?.debug(
+        this.log?.debug(
           `received duplicate msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq}), discarding`,
-          { clientId: this.clientId, transportMessage: msg }
+          {
+            clientId: this.clientId,
+            transportMessage: msg,
+            ...conn.loggingMetadata
+          }
         );
       } else {
         const errMsg = `received out-of-order msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq})`;
-        log?.error(`${errMsg}, marking connection as dead`, {
+        this.log?.error(`${errMsg}, marking connection as dead`, {
           clientId: this.clientId,
           transportMessage: msg,
+          ...conn.loggingMetadata,
           tags: ["invariant-violation"]
         });
         this.protocolError(ProtocolError.MessageOrderingViolated, errMsg);
@@ -798,9 +894,10 @@ var Transport = class {
     if (!isAck(msg.controlFlags)) {
       this.eventDispatcher.dispatchEvent("message", msg);
     } else {
-      log?.debug(`discarding msg (ack bit set)`, {
+      this.log?.debug(`discarding msg (ack bit set)`, {
         clientId: this.clientId,
-        transportMessage: msg
+        transportMessage: msg,
+        ...conn.loggingMetadata
       });
     }
   }
@@ -829,7 +926,7 @@ var Transport = class {
   send(to, msg) {
     if (this.state === "destroyed") {
       const err = "transport is destroyed, cant send";
-      log?.error(err, {
+      this.log?.error(err, {
         clientId: this.clientId,
         transportMessage: msg,
         tags: ["invariant-violation"]
@@ -837,7 +934,7 @@ var Transport = class {
       this.protocolError(ProtocolError.UseAfterDestroy, err);
       return void 0;
     } else if (this.state === "closed") {
-      log?.info(`transport closed when sending, discarding`, {
+      this.log?.info(`transport closed when sending, discarding`, {
         clientId: this.clientId,
         transportMessage: msg
       });
@@ -868,7 +965,7 @@ var Transport = class {
     for (const session of this.sessions.values()) {
       this.deleteSession(session);
     }
-    log?.info(`manually closed transport`, { clientId: this.clientId });
+    this.log?.info(`manually closed transport`, { clientId: this.clientId });
   }
   /**
    * Default destroy implementation for transports. You should override this in the downstream
@@ -880,7 +977,7 @@ var Transport = class {
     for (const session of this.sessions.values()) {
       this.deleteSession(session);
     }
-    log?.info(`manually destroyed transport`, { clientId: this.clientId });
+    this.log?.info(`manually destroyed transport`, { clientId: this.clientId });
   }
 };
 var ClientTransport = class extends Transport {
@@ -922,9 +1019,9 @@ var ClientTransport = class extends Transport {
     let session = void 0;
     const handshakeTimeout = setTimeout(() => {
       if (!session) {
-        log?.warn(
+        this.log?.warn(
           `connection to ${to} timed out waiting for handshake, closing`,
-          { clientId: this.clientId, connectedTo: to, connId: conn.id }
+          { ...conn.loggingMetadata, clientId: this.clientId, connectedTo: to }
         );
         conn.close();
       }
@@ -940,7 +1037,7 @@ var ClientTransport = class extends Transport {
       }
       conn.removeDataListener(handshakeHandler);
       conn.addDataListener((data2) => {
-        const parsed = this.parseMsg(data2);
+        const parsed = this.parseMsg(data2, conn);
         if (!parsed) {
           conn.telemetry?.span.setStatus({
             code: import_api3.SpanStatusCode.ERROR,
@@ -949,7 +1046,7 @@ var ClientTransport = class extends Transport {
           conn.close();
           return;
         }
-        this.handleMsg(parsed);
+        this.handleMsg(parsed, conn);
       });
     };
     conn.addDataListener(handshakeHandler);
@@ -957,7 +1054,8 @@ var ClientTransport = class extends Transport {
       if (session) {
         this.onDisconnect(conn, session);
       }
-      log?.info(`connection to ${to} disconnected`, {
+      this.log?.info(`connection to ${to} disconnected`, {
+        ...conn.loggingMetadata,
         ...session?.loggingMetadata,
         clientId: this.clientId,
         connectedTo: to
@@ -972,15 +1070,19 @@ var ClientTransport = class extends Transport {
         code: import_api3.SpanStatusCode.ERROR,
         message: "connection error"
       });
-      log?.warn(`error in connection to ${to}: ${coerceErrorString(err)}`, {
-        ...session?.loggingMetadata,
-        clientId: this.clientId,
-        connectedTo: to
-      });
+      this.log?.warn(
+        `error in connection to ${to}: ${coerceErrorString(err)}`,
+        {
+          ...conn.loggingMetadata,
+          ...session?.loggingMetadata,
+          clientId: this.clientId,
+          connectedTo: to
+        }
+      );
     });
   }
   receiveHandshakeResponseMessage(data, conn) {
-    const parsed = this.parseMsg(data);
+    const parsed = this.parseMsg(data, conn);
     if (!parsed) {
       conn.telemetry?.span.setStatus({
         code: import_api3.SpanStatusCode.ERROR,
@@ -997,7 +1099,8 @@ var ClientTransport = class extends Transport {
         code: import_api3.SpanStatusCode.ERROR,
         message: "invalid handshake response"
       });
-      log?.warn(`received invalid handshake resp`, {
+      this.log?.warn(`received invalid handshake resp`, {
+        ...conn.loggingMetadata,
         clientId: this.clientId,
         connectedTo: parsed.from,
         transportMessage: parsed,
@@ -1019,7 +1122,8 @@ var ClientTransport = class extends Transport {
         code: import_api3.SpanStatusCode.ERROR,
         message: "handshake rejected"
       });
-      log?.warn(`received handshake rejection`, {
+      this.log?.warn(`received handshake rejection`, {
+        ...conn.loggingMetadata,
         clientId: this.clientId,
         connectedTo: parsed.from,
         transportMessage: parsed
@@ -1030,7 +1134,8 @@ var ClientTransport = class extends Transport {
       );
       return false;
     }
-    log?.debug(`handshake from ${parsed.from} ok`, {
+    this.log?.debug(`handshake from ${parsed.from} ok`, {
+      ...conn.loggingMetadata,
       clientId: this.clientId,
       connectedTo: parsed.from,
       transportMessage: parsed
@@ -1051,7 +1156,7 @@ var ClientTransport = class extends Transport {
   async connect(to) {
     const canProceedWithConnection = () => this.state === "open";
     if (!canProceedWithConnection()) {
-      log?.info(
+      this.log?.info(
         `transport state is no longer open, cancelling attempt to connect to ${to}`,
         { clientId: this.clientId, connectedTo: to }
       );
@@ -1062,7 +1167,7 @@ var ClientTransport = class extends Transport {
       const budgetConsumed = this.retryBudget.getBudgetConsumed(to);
       if (!this.retryBudget.hasBudget(to)) {
         const errMsg = `tried to connect to ${to} but retry budget exceeded (more than ${budgetConsumed} attempts in the last ${this.retryBudget.totalBudgetRestoreTime}ms)`;
-        log?.warn(errMsg, { clientId: this.clientId, connectedTo: to });
+        this.log?.error(errMsg, { clientId: this.clientId, connectedTo: to });
         this.protocolError(ProtocolError.RetriesExceeded, errMsg);
         return;
       }
@@ -1071,10 +1176,13 @@ var ClientTransport = class extends Transport {
       if (backoffMs > 0) {
         sleep = new Promise((resolve) => setTimeout(resolve, backoffMs));
       }
-      log?.info(`attempting connection to ${to} (${backoffMs}ms backoff)`, {
-        clientId: this.clientId,
-        connectedTo: to
-      });
+      this.log?.info(
+        `attempting connection to ${to} (${backoffMs}ms backoff)`,
+        {
+          clientId: this.clientId,
+          connectedTo: to
+        }
+      );
       this.retryBudget.consumeBudget(to);
       reconnectPromise = tracing_default.startActiveSpan("connect", async (span) => {
         try {
@@ -1086,12 +1194,12 @@ var ClientTransport = class extends Transport {
           span.addEvent("connecting");
           const conn = await this.createNewOutgoingConnection(to);
           if (!canProceedWithConnection()) {
-            log?.info(
+            this.log?.info(
               `transport state is no longer open, closing pre-handshake connection to ${to}`,
               {
+                ...conn.loggingMetadata,
                 clientId: this.clientId,
-                connectedTo: to,
-                connId: conn.id
+                connectedTo: to
               }
             );
             conn.close();
@@ -1115,10 +1223,13 @@ var ClientTransport = class extends Transport {
       });
       this.inflightConnectionPromises.set(to, reconnectPromise);
     } else {
-      log?.info(`attempting connection to ${to} (reusing previous attempt)`, {
-        clientId: this.clientId,
-        connectedTo: to
-      });
+      this.log?.info(
+        `attempting connection to ${to} (reusing previous attempt)`,
+        {
+          clientId: this.clientId,
+          connectedTo: to
+        }
+      );
     }
     try {
       await reconnectPromise;
@@ -1126,12 +1237,12 @@ var ClientTransport = class extends Transport {
       this.inflightConnectionPromises.delete(to);
       const errStr = coerceErrorString(error);
       if (!this.reconnectOnConnectionDrop || !canProceedWithConnection()) {
-        log?.warn(`connection to ${to} failed (${errStr})`, {
+        this.log?.warn(`connection to ${to} failed (${errStr})`, {
           clientId: this.clientId,
           connectedTo: to
         });
       } else {
-        log?.warn(`connection to ${to} failed (${errStr}), retrying`, {
+        this.log?.warn(`connection to ${to} failed (${errStr}), retrying`, {
           clientId: this.clientId,
           connectedTo: to
         });
@@ -1148,7 +1259,8 @@ var ClientTransport = class extends Transport {
     if (this.handshakeExtensions) {
       metadata = await this.handshakeExtensions.construct();
       if (!import_value.Value.Check(this.handshakeExtensions.schema, metadata)) {
-        log?.error(`constructed handshake metadata did not match schema`, {
+        this.log?.error(`constructed handshake metadata did not match schema`, {
+          ...conn.loggingMetadata,
           clientId: this.clientId,
           connectedTo: to,
           validationErrors: [
@@ -1175,9 +1287,11 @@ var ClientTransport = class extends Transport {
       metadata,
       getPropagationContext(session.telemetry.ctx)
     );
-    log?.debug(`sending handshake request to ${to}`, {
+    this.log?.debug(`sending handshake request to ${to}`, {
+      ...conn.loggingMetadata,
       clientId: this.clientId,
-      connectedTo: to
+      connectedTo: to,
+      transportMessage: requestMsg
     });
     conn.send(this.codec.toBuffer(requestMsg));
     return true;
@@ -1208,7 +1322,6 @@ var WebSocketConnection = class extends Connection {
             `websocket closed with code and reason: ${code} - ${reason}`
           )
         );
-        return;
       }
       if (this.closeCb) {
         this.closeCb();
@@ -1258,7 +1371,7 @@ var WebSocketClientTransport = class extends ClientTransport {
     this.wsGetter = wsGetter;
   }
   async createNewOutgoingConnection(to) {
-    log?.info(`establishing a new websocket to ${to}`, {
+    this.log?.info(`establishing a new websocket to ${to}`, {
       clientId: this.clientId,
       connectedTo: to
     });
@@ -1280,7 +1393,7 @@ var WebSocketClientTransport = class extends ClientTransport {
       };
     });
     const conn = new WebSocketConnection(ws);
-    log?.info(`raw websocket to ${to} ok, starting handshake`, {
+    this.log?.info(`raw websocket to ${to} ok, starting handshake`, {
       clientId: this.clientId,
       connectedTo: to
     });