mirror of
https://github.com/kubernetes-sigs/external-dns.git
synced 2025-08-06 01:26:59 +02:00
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
/*
|
|
Copyright 2025 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
// ref: https://github.com/linki/instrumented_http/blob/master/client.go
|
|
|
|
package http
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
var (
|
|
requestDuration = prometheus.NewSummaryVec(
|
|
prometheus.SummaryOpts{
|
|
Name: "request_duration_seconds",
|
|
Help: "The HTTP request latencies in seconds.",
|
|
Subsystem: "http",
|
|
ConstLabels: prometheus.Labels{"handler": "instrumented_http"},
|
|
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
|
|
},
|
|
[]string{"scheme", "host", "path", "method", "status"},
|
|
)
|
|
)
|
|
|
|
func init() {
|
|
prometheus.MustRegister(requestDuration)
|
|
}
|
|
|
|
type CustomRoundTripper struct {
|
|
next http.RoundTripper
|
|
}
|
|
|
|
// CancelRequest is a no-op to satisfy interfaces that require it.
|
|
// https://github.com/kubernetes/client-go/blob/34f52c14eaed7e50c845cc14e85e1c4c91e77470/transport/transport.go#L346
|
|
func (r *CustomRoundTripper) CancelRequest(_ *http.Request) {
|
|
}
|
|
|
|
func (r *CustomRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
start := time.Now()
|
|
resp, err := r.next.RoundTrip(req)
|
|
|
|
status := ""
|
|
if resp != nil {
|
|
status = fmt.Sprintf("%d", resp.StatusCode)
|
|
}
|
|
|
|
labels := prometheus.Labels{
|
|
"scheme": req.URL.Scheme,
|
|
"host": req.URL.Host,
|
|
"path": pathProcessor(req.URL.Path),
|
|
"method": req.Method,
|
|
"status": status,
|
|
}
|
|
requestDuration.With(labels).Observe(time.Since(start).Seconds())
|
|
return resp, err
|
|
}
|
|
|
|
func NewInstrumentedClient(next *http.Client) *http.Client {
|
|
if next == nil {
|
|
next = http.DefaultClient
|
|
}
|
|
|
|
next.Transport = NewInstrumentedTransport(next.Transport)
|
|
|
|
return next
|
|
}
|
|
|
|
func NewInstrumentedTransport(next http.RoundTripper) http.RoundTripper {
|
|
if next == nil {
|
|
next = http.DefaultTransport
|
|
}
|
|
|
|
return &CustomRoundTripper{next: next}
|
|
}
|
|
|
|
func pathProcessor(path string) string {
|
|
parts := strings.Split(path, "/")
|
|
return parts[len(parts)-1]
|
|
}
|