RepoMirrors
/
prometheus
mirror of https://github.com/prometheus/prometheus
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remote client: apply custom headers before sigv4 transport 

Signed-off-by: Craig Peterson <192540+captncraig@users.noreply.github.com>
This commit is contained in:
Craig Peterson 2024-01-29 22:49:40 -05:00
parent 501bc6419e
commit 5b5230deb7
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
storage/remote/client.go
View File

@ -141,24 +141,24 @@ func NewWriteClient(name string, conf *ClientConfig) (WriteClient, error) {
} }
t := httpClient.Transport t := httpClient.Transport
if len(conf.Headers) > 0 {
t = newInjectHeadersRoundTripper(conf.Headers, t)
}
if conf.SigV4Config != nil { if conf.SigV4Config != nil {
t, err = sigv4.NewSigV4RoundTripper(conf.SigV4Config, httpClient.Transport) t, err = sigv4.NewSigV4RoundTripper(conf.SigV4Config, t)
if err != nil { if err != nil {
return nil, err return nil, err
} }
} }
if conf.AzureADConfig != nil { if conf.AzureADConfig != nil {
t, err = azuread.NewAzureADRoundTripper(conf.AzureADConfig, httpClient.Transport) t, err = azuread.NewAzureADRoundTripper(conf.AzureADConfig, t)
if err != nil { if err != nil {
return nil, err return nil, err
} }
} }
if len(conf.Headers) > 0 {
t = newInjectHeadersRoundTripper(conf.Headers, t)
}
httpClient.Transport = otelhttp.NewTransport(t) httpClient.Transport = otelhttp.NewTransport(t)
return &Client{ return &Client{