-
Notifications
You must be signed in to change notification settings - Fork 69
/
request_tracing.go
70 lines (66 loc) · 2.46 KB
/
request_tracing.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
package core
import (
"net/http"
"github.com/wundergraph/graphql-go-tools/v2/pkg/engine/resolve"
)
const (
// RequestTraceHeader is the header used to enable request tracing
RequestTraceHeader = "X-WG-Trace"
// RequestTraceQueryParameter is the query parameter used to enable request tracing
RequestTraceQueryParameter = "wg_trace"
requestTraceOptionExcludeParseStats = "exclude_parse_stats"
requestTraceOptionExcludeNormalizeStats = "exclude_normalize_stats"
requestTraceOptionExcludeValidateStats = "exclude_validate_stats"
requestTraceOptionExcludePlannerStats = "exclude_planner_stats"
requestTraceOptionExcludeRawInputData = "exclude_raw_input_data"
requestTraceOptionExcludeInput = "exclude_input"
requestTraceOptionExcludeOutput = "exclude_output"
requestTraceOptionExcludeLoadStats = "exclude_load_stats"
requestTraceOptionEnablePredictableDebugTimings = "enable_predictable_debug_timings"
)
func ParseRequestTraceOptions(r *http.Request) (options resolve.TraceOptions) {
var (
values []string
)
if r.Header.Get(RequestTraceHeader) != "" {
options.Enable = true
values = r.Header.Values(RequestTraceHeader)
}
if r.URL.Query().Get(RequestTraceQueryParameter) != "" {
options.Enable = true
values = r.URL.Query()[RequestTraceQueryParameter]
}
if len(values) == 0 {
options.ExcludePlannerStats = true
options.ExcludeRawInputData = true
options.ExcludeInput = true
options.ExcludeOutput = true
options.ExcludeLoadStats = true
options.EnablePredictableDebugTimings = true
return
}
options.IncludeTraceOutputInResponseExtensions = true
for i := range values {
switch values[i] {
case requestTraceOptionExcludeParseStats:
options.ExcludeParseStats = true
case requestTraceOptionExcludeNormalizeStats:
options.ExcludeNormalizeStats = true
case requestTraceOptionExcludeValidateStats:
options.ExcludeValidateStats = true
case requestTraceOptionExcludePlannerStats:
options.ExcludePlannerStats = true
case requestTraceOptionExcludeRawInputData:
options.ExcludeRawInputData = true
case requestTraceOptionExcludeInput:
options.ExcludeInput = true
case requestTraceOptionExcludeOutput:
options.ExcludeOutput = true
case requestTraceOptionExcludeLoadStats:
options.ExcludeLoadStats = true
case requestTraceOptionEnablePredictableDebugTimings:
options.EnablePredictableDebugTimings = true
}
}
return
}