Fix missing TraceInfo when download callback is set(#200)

imroc · Jan 5, 2023 · 9d5943a · 9d5943a
1 parent fe5fafc
commit 9d5943a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -1296,7 +1296,10 @@ func (c *Client) roundTrip(r *Request) (resp *Response, err error) {
 				interval: r.downloadCallbackInterval,
 			}
 		}
-		ctx = context.WithValue(r.Context(), wrapResponseBodyKey, wrap)
+		if ctx == nil {
+			ctx = context.Background()
+		}
+		ctx = context.WithValue(ctx, wrapResponseBodyKey, wrap)
 	}
 	if ctx != nil {
 		req = req.WithContext(ctx)