From c9b7f579512befd9a1832ae7e4a384faefe5a344 Mon Sep 17 00:00:00 2001
From: Abhishek Ranjan <aranjans@google.com>
Date: Thu, 4 Apr 2024 22:57:48 +0530
Subject: [PATCH] Update clientStreamAttempt context with transport context

---
 stream.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/stream.go b/stream.go
index d939ffc63489..b54563e81cdb 100644
--- a/stream.go
+++ b/stream.go
@@ -516,6 +516,7 @@ func (a *csAttempt) newStream() error {
 		return toRPCErr(nse.Err)
 	}
 	a.s = s
+	a.ctx = s.Context()
 	a.p = &parser{r: s, recvBufferPool: a.cs.cc.dopts.recvBufferPool}
 	return nil
 }