forked from sourcegraph/track
/
handlers.go
88 lines (77 loc) · 2.04 KB
/
handlers.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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package track
import (
"github.com/gorilla/mux"
"github.com/sourcegraph/go-nnz/nnz"
"log"
"net/http"
"time"
)
// ViewIDHeader is the HTTP request header ("X-Track-View") that contains the view
// instance and sequence number.
const ViewIDHeader = "X-Track-View"
// TrackAPICall wraps an API endpoint handler and records incoming API calls.
func TrackAPICall(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
c := &Call{
URL: r.URL.String(),
Route: mux.CurrentRoute(r).GetName(),
RouteParams: mapStringStringAsParams(mux.Vars(r)),
QueryParams: mapStringSliceOfStringAsParams(r.URL.Query()),
Date: time.Now(),
}
// Try to get the current view info from the X-Track-View header.
viewID, err := GetViewID(r)
if err != nil {
log.Printf("GetViewID failed: %s", err)
w.WriteHeader(http.StatusInternalServerError)
return
}
if viewID != nil {
c.Instance = viewID.Instance
c.ViewSeq = nnz.Int(viewID.Seq)
}
// Otherwise, try to get the instance from the request context.
if c.Instance == 0 {
if i := GetInstance(r); i != 0 {
c.Instance = i
}
}
err = InsertCall(c)
if err != nil {
log.Printf("InsertCall failed: %s", err)
w.WriteHeader(http.StatusInternalServerError)
return
}
status := CallStatus{
CallID: c.ID,
Panicked: true, // assume the worst, set false if no panic
}
start := time.Now()
rw := newRecorder(w)
defer func() {
status.Duration = time.Since(start).Nanoseconds()
status.BodyLength = rw.BodyLength
status.HTTPStatusCode = rw.Code
err := InsertCallStatus(&status)
if err != nil {
log.Printf("warn: UpdateCallStatus failed (ID=%d): %s", c.ID, err)
}
}()
h.ServeHTTP(rw, r)
status.Panicked = false
})
}
func mapStringStringAsParams(m map[string]string) (p Params) {
p = make(Params)
for k, v := range m {
p[k] = v
}
return
}
func mapStringSliceOfStringAsParams(m map[string][]string) (p Params) {
p = make(Params)
for k, v := range m {
p[k] = v
}
return
}