@replit/river 0.203.0 → 0.203.1

package/dist/router/index.d.cts

@@ -56,6 +56,6 @@ declare function createServer<Services extends AnyServiceSchemaMap>(transport: S
     maxCancelledStreamTombstonesPerSession?: number;
 }>): Server<Services>;
 
-var version = "0.203.0";
+var version = "0.203.1";
 
 export { ParsedMetadata, version as RIVER_VERSION, Server, ServiceContext, createServer };

package/dist/router/index.d.ts

@@ -56,6 +56,6 @@ declare function createServer<Services extends AnyServiceSchemaMap>(transport: S
     maxCancelledStreamTombstonesPerSession?: number;
 }>): Server<Services>;
 
-var version = "0.203.0";
+var version = "0.203.1";
 
 export { ParsedMetadata, version as RIVER_VERSION, Server, ServiceContext, createServer };

package/dist/router/index.js

@@ -12,7 +12,7 @@ import {
   isStreamClose,
   isStreamOpen,
   version
-} from "../chunk-SZHNMLKC.js";
+} from "../chunk-AK7NTFAM.js";
 
 // router/services.ts
 import { Type as Type2, Kind as Kind2 } from "@sinclair/typebox";
@@ -725,7 +725,7 @@ function handleProc(procType, transport, serverId, init, serviceName, procedureN
   const procClosesWithInit = procType === "rpc" || procType === "subscription";
   const streamId = generateId();
   const { span, ctx } = createProcTelemetryInfo(
-    transport,
+    session,
     procType,
     serviceName,
     procedureName,
@@ -1000,10 +1000,17 @@ var RiverServer = class {
       if (!newStreamProps) {
         return;
       }
-      this.createNewProcStream({
-        ...newStreamProps,
-        ...message
-      });
+      createHandlerSpan(
+        newStreamProps.initialSession,
+        newStreamProps.procedure.type,
+        newStreamProps.serviceName,
+        newStreamProps.procedureName,
+        newStreamProps.streamId,
+        newStreamProps.tracingCtx,
+        (span) => {
+          this.createNewProcStream(span, newStreamProps);
+        }
+      );
     };
     const handleSessionStatus = (evt) => {
       if (evt.status !== "disconnect")
@@ -1037,7 +1044,7 @@ var RiverServer = class {
     this.transport.addEventListener("sessionStatus", handleSessionStatus);
     this.transport.addEventListener("transportStatus", handleTransportStatus);
   }
-  createNewProcStream(props) {
+  createNewProcStream(span, props) {
     const {
       streamId,
       initialSession,
@@ -1047,7 +1054,6 @@ var RiverServer = class {
       sessionMetadata,
       serviceContext,
       initPayload,
-      tracingCtx,
       procClosesWithInit,
       passInitAsDataForBackwardsCompat
     } = props;
@@ -1057,6 +1063,10 @@ var RiverServer = class {
       protocolVersion,
       id: sessionId
     } = initialSession;
+    loggingMetadata.telemetry = {
+      traceId: span.spanContext().traceId,
+      spanId: span.spanContext().spanId
+    };
     let cleanClose = true;
     const onMessage = (msg) => {
       if (msg.from !== from) {
@@ -1229,10 +1239,10 @@ var RiverServer = class {
         }
       }
     });
-    const onHandlerError = (err, span) => {
+    const onHandlerError = (err, span2) => {
       const errorMsg = coerceErrorString(err);
-      span.recordException(err instanceof Error ? err : new Error(errorMsg));
-      span.setStatus({ code: SpanStatusCode.ERROR });
+      span2.recordException(err instanceof Error ? err : new Error(errorMsg));
+      span2.setStatus({ code: SpanStatusCode.ERROR });
       this.log?.error(
         `${serviceName}.${procedureName} handler threw an uncaught error`,
         {
@@ -1256,12 +1266,12 @@ var RiverServer = class {
     if (procClosesWithInit) {
       closeReadable();
     }
-    const handlerContextWithSpan = (span) => ({
+    const handlerContextWithSpan = (span2) => ({
       ...serviceContext,
       from,
       sessionId,
       metadata: sessionMetadata,
-      span,
+      span: span2,
       cancel: () => {
         onServerCancel({
           code: CANCEL_CODE,
@@ -1272,100 +1282,72 @@ var RiverServer = class {
     });
     switch (procedure.type) {
       case "rpc":
-        void createHandlerSpan(
-          procedure.type,
-          serviceName,
-          procedureName,
-          streamId,
-          tracingCtx,
-          async (span) => {
-            try {
-              const responsePayload = await procedure.handler({
-                ctx: handlerContextWithSpan(span),
-                reqInit: initPayload
-              });
-              if (resWritable.isClosed()) {
-                return;
-              }
-              resWritable.write(responsePayload);
-            } catch (err) {
-              onHandlerError(err, span);
-            } finally {
-              span.end();
+        void (async () => {
+          try {
+            const responsePayload = await procedure.handler({
+              ctx: handlerContextWithSpan(span),
+              reqInit: initPayload
+            });
+            if (resWritable.isClosed()) {
+              return;
             }
+            resWritable.write(responsePayload);
+          } catch (err) {
+            onHandlerError(err, span);
+          } finally {
+            span.end();
           }
-        );
+        })();
         break;
       case "stream":
-        void createHandlerSpan(
-          procedure.type,
-          serviceName,
-          procedureName,
-          streamId,
-          tracingCtx,
-          async (span) => {
-            try {
-              await procedure.handler({
-                ctx: handlerContextWithSpan(span),
-                reqInit: initPayload,
-                reqReadable,
-                resWritable
-              });
-            } catch (err) {
-              onHandlerError(err, span);
-            } finally {
-              span.end();
-            }
+        void (async () => {
+          try {
+            await procedure.handler({
+              ctx: handlerContextWithSpan(span),
+              reqInit: initPayload,
+              reqReadable,
+              resWritable
+            });
+          } catch (err) {
+            onHandlerError(err, span);
+          } finally {
+            span.end();
           }
-        );
+        })();
         break;
       case "subscription":
-        void createHandlerSpan(
-          procedure.type,
-          serviceName,
-          procedureName,
-          streamId,
-          tracingCtx,
-          async (span) => {
-            try {
-              await procedure.handler({
-                ctx: handlerContextWithSpan(span),
-                reqInit: initPayload,
-                resWritable
-              });
-            } catch (err) {
-              onHandlerError(err, span);
-            } finally {
-              span.end();
-            }
+        void (async () => {
+          try {
+            await procedure.handler({
+              ctx: handlerContextWithSpan(span),
+              reqInit: initPayload,
+              resWritable
+            });
+          } catch (err) {
+            onHandlerError(err, span);
+          } finally {
+            span.end();
           }
-        );
+        })();
         break;
       case "upload":
-        void createHandlerSpan(
-          procedure.type,
-          serviceName,
-          procedureName,
-          streamId,
-          tracingCtx,
-          async (span) => {
-            try {
-              const responsePayload = await procedure.handler({
-                ctx: handlerContextWithSpan(span),
-                reqInit: initPayload,
-                reqReadable
-              });
-              if (resWritable.isClosed()) {
-                return;
-              }
-              resWritable.write(responsePayload);
-            } catch (err) {
-              onHandlerError(err, span);
-            } finally {
-              span.end();
+        void (async () => {
+          try {
+            const responsePayload = await procedure.handler({
+              ctx: handlerContextWithSpan(span),
+              reqInit: initPayload,
+              reqReadable
+            });
+            if (resWritable.isClosed()) {
+              return;
             }
+            resWritable.write(responsePayload);
+          } catch (err) {
+            onHandlerError(err, span);
+          } finally {
+            span.end();
           }
-        );
+        })();
         break;
     }
     if (!finishedController.signal.aborted) {