/
control.go
89 lines (70 loc) · 2.63 KB
/
control.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
89
package accesslog
import "github.com/zalando/skipper/filters"
const (
// DisableAccessLogName is the filter name seen by the user
DisableAccessLogName = "disableAccessLog"
// EnableAccessLogName is the filter name seen by the user
EnableAccessLogName = "enableAccessLog"
// AccessLogEnabledKey is the key used in the state bag to pass the access log state to the proxy.
AccessLogEnabledKey = "statebag:access_log:proxy:enabled"
// AccessLogAdditionalDataKey is the key used in the state bag to pass extra data to access log
AccessLogAdditionalDataKey = "statebag:access_log:additional"
)
// Common filter struct for holding access log state
type AccessLogFilter struct {
Enable bool
Prefixes []int
}
func (al *AccessLogFilter) Request(ctx filters.FilterContext) {
bag := ctx.StateBag()
bag[AccessLogEnabledKey] = al
}
func (*AccessLogFilter) Response(filters.FilterContext) {}
func extractFilterValues(args []interface{}, enable bool) (filters.Filter, error) {
prefixes := make([]int, 0)
for _, prefix := range args {
var intPref int
switch p := prefix.(type) {
case float32:
intPref = int(p)
case float64:
intPref = int(p)
default:
var ok bool
intPref, ok = prefix.(int)
if !ok {
return nil, filters.ErrInvalidFilterParameters
}
}
prefixes = append(prefixes, intPref)
}
return &AccessLogFilter{Enable: enable, Prefixes: prefixes}, nil
}
type disableAccessLog struct{}
// NewDisableAccessLog creates a filter spec to disable access log for specific route.
// Optionally takes in response code prefixes as arguments. When provided, access log is disabled
// only if response code matches one of the arguments.
//
// disableAccessLog() or
// disableAccessLog(1, 20, 301) to disable logs for 1xx, 20x and 301 codes
func NewDisableAccessLog() filters.Spec {
return &disableAccessLog{}
}
func (*disableAccessLog) Name() string { return DisableAccessLogName }
func (al *disableAccessLog) CreateFilter(args []interface{}) (filters.Filter, error) {
return extractFilterValues(args, false)
}
type enableAccessLog struct{}
// NewEnableAccessLog creates a filter spec to enable access log for specific route
// Optionally takes in response code prefixes as arguments. When provided, access log is enabled
// only if response code matches one of the arguments.
//
// enableAccessLog()
// enableAccessLog(1, 20, 301) to enable logs for 1xx, 20x and 301 codes
func NewEnableAccessLog() filters.Spec {
return &enableAccessLog{}
}
func (*enableAccessLog) Name() string { return EnableAccessLogName }
func (al *enableAccessLog) CreateFilter(args []interface{}) (filters.Filter, error) {
return extractFilterValues(args, true)
}