diff options
Diffstat (limited to 'middleware/cache/handler.go')
-rw-r--r-- | middleware/cache/handler.go | 119 |
1 files changed, 0 insertions, 119 deletions
diff --git a/middleware/cache/handler.go b/middleware/cache/handler.go deleted file mode 100644 index 90c63d93a..000000000 --- a/middleware/cache/handler.go +++ /dev/null @@ -1,119 +0,0 @@ -package cache - -import ( - "time" - - "github.com/coredns/coredns/middleware" - "github.com/coredns/coredns/request" - - "github.com/miekg/dns" - "github.com/prometheus/client_golang/prometheus" - "golang.org/x/net/context" -) - -// ServeDNS implements the middleware.Handler interface. -func (c *Cache) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { - state := request.Request{W: w, Req: r} - - qname := state.Name() - qtype := state.QType() - zone := middleware.Zones(c.Zones).Matches(qname) - if zone == "" { - return c.Next.ServeDNS(ctx, w, r) - } - - do := state.Do() // TODO(): might need more from OPT record? Like the actual bufsize? - - now := time.Now().UTC() - - i, ttl := c.get(now, qname, qtype, do) - if i != nil && ttl > 0 { - resp := i.toMsg(r) - - state.SizeAndDo(resp) - resp, _ = state.Scrub(resp) - w.WriteMsg(resp) - - i.Freq.Update(c.duration, now) - - pct := 100 - if i.origTTL != 0 { // you'll never know - pct = int(float64(ttl) / float64(i.origTTL) * 100) - } - - if c.prefetch > 0 && i.Freq.Hits() > c.prefetch && pct < c.percentage { - // When prefetching we loose the item i, and with it the frequency - // that we've gathered sofar. See we copy the frequencies info back - // into the new item that was stored in the cache. - prr := &ResponseWriter{ResponseWriter: w, Cache: c, prefetch: true} - middleware.NextOrFailure(c.Name(), c.Next, ctx, prr, r) - - if i1, _ := c.get(now, qname, qtype, do); i1 != nil { - i1.Freq.Reset(now, i.Freq.Hits()) - } - } - - return dns.RcodeSuccess, nil - } - - crr := &ResponseWriter{ResponseWriter: w, Cache: c} - return middleware.NextOrFailure(c.Name(), c.Next, ctx, crr, r) -} - -// Name implements the Handler interface. -func (c *Cache) Name() string { return "cache" } - -func (c *Cache) get(now time.Time, qname string, qtype uint16, do bool) (*item, int) { - k := hash(qname, qtype, do) - - if i, ok := c.ncache.Get(k); ok { - cacheHits.WithLabelValues(Denial).Inc() - return i.(*item), i.(*item).ttl(now) - } - - if i, ok := c.pcache.Get(k); ok { - cacheHits.WithLabelValues(Success).Inc() - return i.(*item), i.(*item).ttl(now) - } - cacheMisses.Inc() - return nil, 0 -} - -var ( - cacheSize = prometheus.NewGaugeVec(prometheus.GaugeOpts{ - Namespace: middleware.Namespace, - Subsystem: subsystem, - Name: "size", - Help: "The number of elements in the cache.", - }, []string{"type"}) - - cacheCapacity = prometheus.NewGaugeVec(prometheus.GaugeOpts{ - Namespace: middleware.Namespace, - Subsystem: subsystem, - Name: "capacity", - Help: "The cache's capacity.", - }, []string{"type"}) - - cacheHits = prometheus.NewCounterVec(prometheus.CounterOpts{ - Namespace: middleware.Namespace, - Subsystem: subsystem, - Name: "hits_total", - Help: "The count of cache hits.", - }, []string{"type"}) - - cacheMisses = prometheus.NewCounter(prometheus.CounterOpts{ - Namespace: middleware.Namespace, - Subsystem: subsystem, - Name: "misses_total", - Help: "The count of cache misses.", - }) -) - -const subsystem = "cache" - -func init() { - prometheus.MustRegister(cacheSize) - prometheus.MustRegister(cacheCapacity) - prometheus.MustRegister(cacheHits) - prometheus.MustRegister(cacheMisses) -} |