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
|
package setup
import (
"strconv"
"github.com/miekg/coredns/middleware"
"github.com/miekg/coredns/middleware/cache"
)
// Cache sets up the root file path of the server.
func Cache(c *Controller) (middleware.Middleware, error) {
ttl, zones, err := cacheParse(c)
if err != nil {
return nil, err
}
return func(next middleware.Handler) middleware.Handler {
return cache.NewCache(ttl, zones, next)
}, nil
}
func cacheParse(c *Controller) (int, []string, error) {
var (
err error
ttl int
)
for c.Next() {
if c.Val() == "cache" {
// cache [ttl] [zones..]
origins := c.ServerBlockHosts
args := c.RemainingArgs()
if len(args) > 0 {
origins = args
// first args may be just a number, then it is the ttl, if not it is a zone
t := origins[0]
ttl, err = strconv.Atoi(t)
if err == nil {
origins = origins[1:]
if len(origins) == 0 {
// There was *only* the ttl, revert back to server block
origins = c.ServerBlockHosts
}
}
}
for i, _ := range origins {
origins[i] = middleware.Host(origins[i]).Normalize()
}
return ttl, origins, nil
}
}
return 0, nil, nil
}
|
