Make remote client Store use passed context (#6673)

* Remote store client's `Store` API currently doesn't use passed
context, but instead just constructs a new `context.Background()`

Signed-off-by: Anand Singh Kunwar <anandkunwar95@gmail.com>
This commit is contained in:
Anand Singh Kunwar 2020-01-27 20:13:20 +05:30 committed by Chris Marchbanks
parent 56ebd5afde
commit aa61e392b2

View file

@ -85,9 +85,8 @@ func (c *Client) Store(ctx context.Context, req []byte) error {
httpReq.Header.Set("Content-Type", "application/x-protobuf") httpReq.Header.Set("Content-Type", "application/x-protobuf")
httpReq.Header.Set("User-Agent", userAgent) httpReq.Header.Set("User-Agent", userAgent)
httpReq.Header.Set("X-Prometheus-Remote-Write-Version", "0.1.0") httpReq.Header.Set("X-Prometheus-Remote-Write-Version", "0.1.0")
httpReq = httpReq.WithContext(ctx)
ctx, cancel := context.WithTimeout(context.Background(), c.timeout) ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel() defer cancel()
httpResp, err := c.client.Do(httpReq.WithContext(ctx)) httpResp, err := c.client.Do(httpReq.WithContext(ctx))