Extra safety

mgnsk · Feb 13, 2021 · 7079ed1 · 7079ed1
1 parent 74d7252
commit 7079ed1
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/cache.go b/cache.go
@@ -165,7 +165,10 @@ func (c *Cache) Evict(key interface{}) (interface{}, bool) {
 		// don't lock it to prevent deadlock.
 		return nil, false
 	}
-	c.records.Delete(key)
+	old, ok := c.records.LoadAndDelete(key)
+	if !ok || old.(*record) != r {
+		panic("evcache: inconsistent map state")
+	}
 	return c.finalize(key, r.(*record))
 }
 
@@ -354,7 +357,10 @@ func (c *Cache) deleteIfEqualsLocked(key interface{}, r *record) bool {
 	if old.(*record) != r {
 		return false
 	}
-	c.records.Delete(key)
+	old, ok = c.records.LoadAndDelete(key)
+	if !ok || old.(*record) != r {
+		panic("evcache: inconsistent map state")
+	}
 	return true
 }