-
Notifications
You must be signed in to change notification settings - Fork 3
/
round_trippers.go
71 lines (58 loc) · 1.59 KB
/
round_trippers.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package main
import (
"net/http"
"path"
"strings"
)
// TransportMiddleware wraps http.RoundTripper.
type TransportMiddleware func(http.RoundTripper) http.RoundTripper
var _ http.RoundTripper = (*pyroscopeTransport)(nil)
// pyroscopeTransport overrides Content-Type for some endpoints.
//
// See https://github.com/grafana/pyroscope/pull/1969.
type pyroscopeTransport struct {
next http.RoundTripper
}
func newPyroscopeTransport(next http.RoundTripper) http.RoundTripper {
return &pyroscopeTransport{next: next}
}
func (t *pyroscopeTransport) RoundTrip(req *http.Request) (*http.Response, error) {
next := t.next
if next == nil {
next = http.DefaultTransport
}
resp, err := next.RoundTrip(req)
if err != nil {
return resp, err
}
if last := path.Base(req.URL.Path); last == "labels" || last == "label-values" {
resp.Header.Set("Content-Type", "application/json")
}
return resp, nil
}
var _ http.RoundTripper = (*tempoTransport)(nil)
// tempoTransport sets Accept for some endpoints.
//
// FIXME(tdakkota): probably, we need to add an Accept header.
type tempoTransport struct {
next http.RoundTripper
}
func newTempoTransport(next http.RoundTripper) http.RoundTripper {
return &tempoTransport{next: next}
}
func (t *tempoTransport) RoundTrip(req *http.Request) (*http.Response, error) {
next := t.next
if next == nil {
next = http.DefaultTransport
}
if strings.Contains(req.URL.Path, "api/traces/") {
if req.Header.Get("Accept") == "" {
req.Header.Set("Accept", "application/protobuf")
}
}
resp, err := next.RoundTrip(req)
if err != nil {
return resp, err
}
return resp, nil
}