diff --git a/go.mod b/go.mod index c6c22577..0079fa8a 100644 --- a/go.mod +++ b/go.mod @@ -10,6 +10,7 @@ require ( github.com/aws/aws-sdk-go-v2/service/sts v1.28.4 github.com/blang/semver/v4 v4.0.0 github.com/go-kit/log v0.2.1 + github.com/google/uuid v1.6.0 github.com/imdario/mergo v0.3.13 github.com/prometheus/client_golang v1.18.0 github.com/prometheus/common v0.46.0 diff --git a/go.sum b/go.sum index 49adcf43..f6625a8d 100644 --- a/go.sum +++ b/go.sum @@ -50,6 +50,8 @@ github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY= github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38= +github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= +github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= github.com/imdario/mergo v0.3.13 h1:lFzP57bqS/wsqKssCGmtLAb8A0wKjLGrve2q3PPVcBk= github.com/imdario/mergo v0.3.13/go.mod h1:4lJ1jqUDcsbIECGy0RUJAXNIhg+6ocWgb1ALK2O4oXg= github.com/jpillora/backoff v1.0.0 h1:uvFg412JmmHBHw7iwprIxkPMI+sGQ4kzOWsMeHnm2EA= diff --git a/main.go b/main.go index 5e4c6da0..bb15dc54 100644 --- a/main.go +++ b/main.go @@ -25,6 +25,7 @@ import ( "github.com/alecthomas/kingpin/v2" "github.com/go-kit/log/level" + "github.com/google/uuid" "github.com/prometheus-community/elasticsearch_exporter/collector" "github.com/prometheus-community/elasticsearch_exporter/pkg/clusterinfo" "github.com/prometheus-community/elasticsearch_exporter/pkg/roundtripper" @@ -37,6 +38,17 @@ import ( const name = "elasticsearch_exporter" +type withDefaultHeader struct { + rt http.RoundTripper +} + +func (h *withDefaultHeader) RoundTrip(req *http.Request) (*http.Response, error) { + req.Header.Add("X-Opaque-Id", uuid.New().String()) + req.Header.Add("x-request-path", "/o11y/prometheus/es_exporter") + req.Header.Add("x-request-from-head-app-name", "elasticsearch_exporter") + return h.rt.RoundTrip(req) +} + type transportWithAPIKey struct { underlyingTransport http.RoundTripper apiKey string @@ -163,8 +175,10 @@ func main() { } httpClient := &http.Client{ - Timeout: *esTimeout, - Transport: httpTransport, + Timeout: *esTimeout, + Transport: &withDefaultHeader{ + rt: httpTransport, + }, } if *awsRegion != "" {