mirror of
https://github.com/coredns/coredns.git
synced 2025-11-01 10:43:17 -04:00
Remove the word middleware (#1067)
* Rename middleware to plugin first pass; mostly used 'sed', few spots where I manually changed text. This still builds a coredns binary. * fmt error * Rename AddMiddleware to AddPlugin * Readd AddMiddleware to remain backwards compat
This commit is contained in:
55
plugin/cache/freq/freq.go
vendored
Normal file
55
plugin/cache/freq/freq.go
vendored
Normal file
@@ -0,0 +1,55 @@
|
||||
// Package freq keeps track of last X seen events. The events themselves are not stored
|
||||
// here. So the Freq type should be added next to the thing it is tracking.
|
||||
package freq
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Freq tracks the frequencies of things.
|
||||
type Freq struct {
|
||||
// Last time we saw a query for this element.
|
||||
last time.Time
|
||||
// Number of this in the last time slice.
|
||||
hits int
|
||||
|
||||
sync.RWMutex
|
||||
}
|
||||
|
||||
// New returns a new initialized Freq.
|
||||
func New(t time.Time) *Freq {
|
||||
return &Freq{last: t, hits: 0}
|
||||
}
|
||||
|
||||
// Update updates the number of hits. Last time seen will be set to now.
|
||||
// If the last time we've seen this entity is within now - d, we increment hits, otherwise
|
||||
// we reset hits to 1. It returns the number of hits.
|
||||
func (f *Freq) Update(d time.Duration, now time.Time) int {
|
||||
earliest := now.Add(-1 * d)
|
||||
f.Lock()
|
||||
defer f.Unlock()
|
||||
if f.last.Before(earliest) {
|
||||
f.last = now
|
||||
f.hits = 1
|
||||
return f.hits
|
||||
}
|
||||
f.last = now
|
||||
f.hits++
|
||||
return f.hits
|
||||
}
|
||||
|
||||
// Hits returns the number of hits that we have seen, according to the updates we have done to f.
|
||||
func (f *Freq) Hits() int {
|
||||
f.RLock()
|
||||
defer f.RUnlock()
|
||||
return f.hits
|
||||
}
|
||||
|
||||
// Reset resets f to time t and hits to hits.
|
||||
func (f *Freq) Reset(t time.Time, hits int) {
|
||||
f.Lock()
|
||||
defer f.Unlock()
|
||||
f.last = t
|
||||
f.hits = hits
|
||||
}
|
||||
36
plugin/cache/freq/freq_test.go
vendored
Normal file
36
plugin/cache/freq/freq_test.go
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
package freq
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestFreqUpdate(t *testing.T) {
|
||||
now := time.Now().UTC()
|
||||
f := New(now)
|
||||
window := 1 * time.Minute
|
||||
|
||||
f.Update(window, time.Now().UTC())
|
||||
f.Update(window, time.Now().UTC())
|
||||
f.Update(window, time.Now().UTC())
|
||||
hitsCheck(t, f, 3)
|
||||
|
||||
f.Reset(now, 0)
|
||||
history := time.Now().UTC().Add(-3 * time.Minute)
|
||||
f.Update(window, history)
|
||||
hitsCheck(t, f, 1)
|
||||
}
|
||||
|
||||
func TestReset(t *testing.T) {
|
||||
f := New(time.Now().UTC())
|
||||
f.Update(1*time.Minute, time.Now().UTC())
|
||||
hitsCheck(t, f, 1)
|
||||
f.Reset(time.Now().UTC(), 0)
|
||||
hitsCheck(t, f, 0)
|
||||
}
|
||||
|
||||
func hitsCheck(t *testing.T, f *Freq, expected int) {
|
||||
if x := f.Hits(); x != expected {
|
||||
t.Fatalf("Expected hits to be %d, got %d", expected, x)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user