re-add conn telemetry span

replit · Oct 18, 2024 · 9c6a536 · 9c6a536
1 parent 7065b8b
commit 9c6a536
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/transport/sessionStateMachine/transitions.ts b/transport/sessionStateMachine/transitions.ts
@@ -14,7 +14,7 @@ import {
   IdentifiedSessionWithGracePeriodProps,
   SessionOptions,
 } from './common';
-import { PropagationContext, createSessionTelemetryInfo } from '../../tracing';
+import { PropagationContext, createConnectionTelemetryInfo, createSessionTelemetryInfo } from '../../tracing';
 import {
   SessionWaitingForHandshake,
   SessionWaitingForHandshakeListeners,
@@ -187,6 +187,7 @@ export const SessionStateGraph = {
         ...carriedState,
       });
 
+      conn.telemetry = createConnectionTelemetryInfo(conn, session.telemetry);
       session.log?.info(
         `session ${session.id} transition from Connecting to Handshaking`,
         {
@@ -262,6 +263,8 @@ export const SessionStateGraph = {
         listeners,
         ...carriedState,
       });
+
+      conn.telemetry = createConnectionTelemetryInfo(conn, session.telemetry);
       session.log?.info(
         `session ${session.id} transition from WaitingForHandshake to Connected`,
         {