-
Notifications
You must be signed in to change notification settings - Fork 0
/
resp.go
83 lines (71 loc) · 1.86 KB
/
resp.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
package cacherules
import (
"net/http"
"strings"
"time"
)
func (ir *InternalRules) IsRespCachable(resp *http.Response) (ttl time.Duration, ok bool) {
ttl, ok, last := ir.getRespStatusCodeTTL(resp)
if !ok || last {
// If the response code is not cachable
return
}
// Rules base in the response
if ok = ir.isValidRespHeader(resp); !ok {
return
}
// Rules base in the response
if ttl, ok = ir.getRespContentTypeTTL(ttl, resp); ok {
return
}
// By default, we cache the results
ok = true
return
}
// getRespStatusCodeTTL Will check it the response code is in the list of cacable codes
func (ir *InternalRules) getRespStatusCodeTTL(resp *http.Response) (time.Duration, bool, bool) {
if len(ir.RespStatusCodeTTL) == 0 {
return defaultTTL, true, false
}
if v, ok := ir.RespStatusCodeTTL[resp.StatusCode]; ok {
if resp.StatusCode == http.StatusOK {
// Response with the TTL, that TTL can be overwrite by the next rules
return v, true, false
}
// This is the last rule that can be apply
return v, true, true
}
// The response code is not in the list of the cachable content
return defaultTTL, false, true
}
func (ir *InternalRules) isValidRespHeader(resp *http.Response) bool {
if len(ir.RespHeadersBlackList) == 0 {
return true
}
for k, v := range ir.RespHeadersBlackList {
headValue := resp.Header.Get(k)
if headValue == "" {
continue
}
for _, s := range v {
if strings.Contains(headValue, s) {
return false
}
}
}
return true
}
func (ir *InternalRules) getRespContentTypeTTL(ttl time.Duration, resp *http.Response) (time.Duration, bool) {
if resp.Header.Get("Content-Type") == "" {
return ttl, false
}
if len(ir.RespContentTypeTTL) == 0 {
return ttl, false
}
for k, v := range ir.RespContentTypeTTL {
if strings.Contains(resp.Header.Get("Content-Type"), k) {
return ttl + v, true
}
}
return ttl, false
}