Fixed memory leak in PushStream caused by reusing context (#6854)

alexqyle · web-flow · commit cc202c782f73 · 2025-07-01T16:42:51.000-07:00
Signed-off-by: Alex Le &lt;leqiyue@amazon.com&gt;
diff --git a/pkg/ingester/ingester.go b/pkg/ingester/ingester.go
@@ -1591,8 +1591,8 @@ func (i *Ingester) PushStream(srv client.Ingester_PushStreamServer) error {
 		if err != nil {
 			return err
 		}
-		ctx = user.InjectOrgID(ctx, req.TenantID)
-		resp, err := i.Push(ctx, req.Request)
+		pushCtx := user.InjectOrgID(ctx, req.TenantID)
+		resp, err := i.Push(pushCtx, req.Request)
 		if resp == nil {
 			resp = &cortexpb.WriteResponse{}
 		}

Original file line number	Diff line number	Diff line change
`@@ -1591,8 +1591,8 @@ func (i *Ingester) PushStream(srv client.Ingester_PushStreamServer) error {`
`1591`	`1591`	`if err != nil {`
`1592`	`1592`	`return err`
`1593`	`1593`	`}`
`1594`		`- ctx = user.InjectOrgID(ctx, req.TenantID)`
`1595`		`- resp, err := i.Push(ctx, req.Request)`
	`1594`	`+ pushCtx := user.InjectOrgID(ctx, req.TenantID)`
	`1595`	`+ resp, err := i.Push(pushCtx, req.Request)`
`1596`	`1596`	`if resp == nil {`
`1597`	`1597`	`resp = &cortexpb.WriteResponse{}`
`1598`	`1598`	`}`