add isExpired bool to OnEvicted callback signature
This commit is contained in:
parent
7ac151875f
commit
44fb2d5981
8
cache.go
8
cache.go
|
@ -41,7 +41,7 @@ type cache struct {
|
|||
defaultExpiration time.Duration
|
||||
items map[string]Item
|
||||
mu sync.RWMutex
|
||||
onEvicted func(string, interface{})
|
||||
onEvicted func(string, interface{}, bool)
|
||||
janitor *janitor
|
||||
}
|
||||
|
||||
|
@ -907,7 +907,7 @@ func (c *cache) Delete(k string) {
|
|||
v, evicted := c.delete(k)
|
||||
c.mu.Unlock()
|
||||
if evicted {
|
||||
c.onEvicted(k, v)
|
||||
c.onEvicted(k, v, false)
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -943,14 +943,14 @@ func (c *cache) DeleteExpired() {
|
|||
}
|
||||
c.mu.Unlock()
|
||||
for _, v := range evictedItems {
|
||||
c.onEvicted(v.key, v.value)
|
||||
c.onEvicted(v.key, v.value, true)
|
||||
}
|
||||
}
|
||||
|
||||
// Sets an (optional) function that is called with the key and value when an
|
||||
// item is evicted from the cache. (Including when it is deleted manually, but
|
||||
// not when it is overwritten.) Set to nil to disable.
|
||||
func (c *cache) OnEvicted(f func(string, interface{})) {
|
||||
func (c *cache) OnEvicted(f func(string, interface{}, bool)) {
|
||||
c.mu.Lock()
|
||||
c.onEvicted = f
|
||||
c.mu.Unlock()
|
||||
|
|
|
@ -1231,8 +1231,8 @@ func TestOnEvicted(t *testing.T) {
|
|||
t.Fatal("tc.onEvicted is not nil")
|
||||
}
|
||||
works := false
|
||||
tc.OnEvicted(func(k string, v interface{}) {
|
||||
if k == "foo" && v.(int) == 3 {
|
||||
tc.OnEvicted(func(k string, v interface{}, expired bool) {
|
||||
if k == "foo" && v.(int) == 3 && expired == false {
|
||||
works = true
|
||||
}
|
||||
tc.Set("bar", 4, DefaultExpiration)
|
||||
|
@ -1247,6 +1247,29 @@ func TestOnEvicted(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestOnEvictedWithExpired(t *testing.T) {
|
||||
tc := New(DefaultExpiration, 0)
|
||||
tc.Set("foo", 3, 1)
|
||||
if tc.onEvicted != nil {
|
||||
t.Fatal("tc.onEvicted is not nil")
|
||||
}
|
||||
works := false
|
||||
tc.OnEvicted(func(k string, v interface{}, expired bool) {
|
||||
if k == "foo" && v.(int) == 3 && expired == true {
|
||||
works = true
|
||||
}
|
||||
tc.Set("bar", 4, DefaultExpiration)
|
||||
})
|
||||
tc.DeleteExpired()
|
||||
|
||||
x, _ := tc.Get("bar")
|
||||
if !works {
|
||||
t.Error("works bool not true")
|
||||
}
|
||||
if x.(int) != 4 {
|
||||
t.Error("bar was not 4")
|
||||
}
|
||||
}
|
||||
func TestCacheSerialization(t *testing.T) {
|
||||
tc := New(DefaultExpiration, 0)
|
||||
testFillAndSerialize(t, tc)
|
||||
|
|
Loading…
Reference in New Issue