| 17 | ) |
| 18 | |
| 19 | func NewRetryWare(maxRetries int, incrementRetriedRequestWeight bool, registerer prometheus.Registerer) Middleware { |
| 20 | retriesCount := promauto.With(registerer).NewHistogram(prometheus.HistogramOpts{ |
| 21 | Namespace: "tempo", |
| 22 | Name: "query_frontend_retries", |
| 23 | Help: "Number of times a request is retried.", |
| 24 | Buckets: []float64{0, 1, 2, 3, 4, 5}, |
| 25 | NativeHistogramBucketFactor: 1.1, |
| 26 | NativeHistogramMaxBucketNumber: 100, |
| 27 | NativeHistogramMinResetDuration: 1 * time.Hour, |
| 28 | }) |
| 29 | |
| 30 | return MiddlewareFunc(func(next RoundTripper) RoundTripper { |
| 31 | return retryWare{ |
| 32 | next: next, |
| 33 | maxRetries: maxRetries, |
| 34 | retriesCount: retriesCount, |
| 35 | incrementRetriedRequestWeight: incrementRetriedRequestWeight, |
| 36 | } |
| 37 | }) |
| 38 | } |
| 39 | |
| 40 | type retryWare struct { |
| 41 | next RoundTripper |