From 9c6a5368c68dc416af0a59233d83633b065727ec Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Fri, 18 Oct 2024 02:19:35 -0700
Subject: [PATCH] re-add conn telemetry span

---
 transport/sessionStateMachine/transitions.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/transport/sessionStateMachine/transitions.ts b/transport/sessionStateMachine/transitions.ts
index 925f347c..3331ddeb 100644
--- 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`,
         {