@replit/river 0.203.0 → 0.203.1

package/dist/testUtil/index.cjs

@@ -396,13 +396,13 @@ var IdentifiedSession = class extends CommonSession {
     this.protocolVersion = protocolVersion;
   }
   get loggingMetadata() {
-    const spanContext = this.telemetry.span.spanContext();
     const metadata = {
       clientId: this.from,
       connectedTo: this.to,
       sessionId: this.id
     };
     if (this.telemetry.span.isRecording()) {
+      const spanContext = this.telemetry.span.spanContext();
       metadata.telemetry = {
         traceId: spanContext.traceId,
         spanId: spanContext.spanId
@@ -532,7 +532,7 @@ var SessionNoConnection = class extends IdentifiedSessionWithGracePeriod {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.203.0";
+var version = "0.203.1";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -560,6 +560,21 @@ function createSessionTelemetryInfo(sessionId, to, from, propagationCtx) {
   const ctx = import_api.trace.setSpan(parentCtx, span);
   return { span, ctx };
 }
+function createConnectionTelemetryInfo(connection, info) {
+  const span = tracer.startSpan(
+    `connection ${connection.id}`,
+    {
+      attributes: {
+        component: "river",
+        "river.connection.id": connection.id
+      },
+      links: [{ context: info.span.spanContext() }]
+    },
+    info.ctx
+  );
+  const ctx = import_api.trace.setSpan(info.ctx, span);
+  return { span, ctx };
+}
 var tracer = import_api.trace.getTracer("river", version);
 var tracing_default = tracer;
 
@@ -943,6 +958,7 @@ var SessionStateGraph = {
         listeners,
         ...carriedState
       });
+      conn.telemetry = createConnectionTelemetryInfo(conn, session.telemetry);
       session.log?.info(
         `session ${session.id} transition from Connecting to Handshaking`,
         {
@@ -1003,6 +1019,7 @@ var SessionStateGraph = {
         listeners,
         ...carriedState
       });
+      conn.telemetry = createConnectionTelemetryInfo(conn, session.telemetry);
       session.log?.info(
         `session ${session.id} transition from WaitingForHandshake to Connected`,
         {
@@ -1129,7 +1146,7 @@ var ServerSessionStateGraph = {
 };
 
 // transport/client.ts
-var import_api3 = require("@opentelemetry/api");
+var import_api4 = require("@opentelemetry/api");
 
 // transport/rateLimit.ts
 var LeakyBucketRateLimit = class {
@@ -1199,6 +1216,7 @@ var LeakyBucketRateLimit = class {
 };
 
 // logging/log.ts
+var import_api3 = require("@opentelemetry/api");
 var LoggingLevels = {
   debug: -1,
   info: 0,
@@ -1207,6 +1225,15 @@ var LoggingLevels = {
 };
 var cleanedLogFn = (log) => {
   return (msg, metadata) => {
+    if (metadata && !metadata.telemetry) {
+      const span = import_api3.trace.getSpan(import_api3.context.active());
+      if (span) {
+        metadata.telemetry = {
+          traceId: span.spanContext().traceId,
+          spanId: span.spanContext().spanId
+        };
+      }
+    }
     if (!metadata?.transportMessage) {
       log(msg, metadata);
       return;
@@ -1634,7 +1661,7 @@ var ClientTransport = class extends Transport {
   }
   rejectHandshakeResponse(session, reason, metadata) {
     session.conn.telemetry?.span.setStatus({
-      code: import_api3.SpanStatusCode.ERROR,
+      code: import_api4.SpanStatusCode.ERROR,
       message: reason
     });
     this.log?.warn(reason, metadata);
@@ -1781,7 +1808,7 @@ var ClientTransport = class extends Transport {
       } catch (err) {
         const errStr = coerceErrorString(err);
         span.recordException(errStr);
-        span.setStatus({ code: import_api3.SpanStatusCode.ERROR });
+        span.setStatus({ code: import_api4.SpanStatusCode.ERROR });
         throw err;
       } finally {
         span.end();
@@ -1863,8 +1890,8 @@ var Connection = class {
   }
   get loggingMetadata() {
     const metadata = { connId: this.id };
-    const spanContext = this.telemetry?.span.spanContext();
-    if (this.telemetry?.span.isRecording() && spanContext) {
+    if (this.telemetry?.span.isRecording()) {
+      const spanContext = this.telemetry.span.spanContext();
       metadata.telemetry = {
         traceId: spanContext.traceId,
         spanId: spanContext.spanId
@@ -1926,7 +1953,7 @@ var Connection = class {
 };
 
 // transport/server.ts
-var import_api4 = require("@opentelemetry/api");
+var import_api5 = require("@opentelemetry/api");
 var import_value3 = require("@sinclair/typebox/value");
 var ServerTransport = class extends Transport {
   /**
@@ -2036,7 +2063,7 @@ var ServerTransport = class extends Transport {
   }
   rejectHandshakeRequest(session, to, reason, code, metadata) {
     session.conn.telemetry?.span.setStatus({
-      code: import_api4.SpanStatusCode.ERROR,
+      code: import_api5.SpanStatusCode.ERROR,
       message: reason
     });
     this.log?.warn(reason, metadata);