Add lock/unlock in SetEvictionFunc

dims · dims · commit 7a270de31e55 · 2024-11-04T11:27:16.000-05:00
Signed-off-by: Davanum Srinivas &lt;davanum@gmail.com&gt;
diff --git a/lru/lru.go b/lru/lru.go
@@ -47,6 +47,8 @@ func NewWithEvictionFunc(size int, f EvictionFunc) *Cache {
 
 // SetEvictionFunc updates the eviction func
 func (c *Cache) SetEvictionFunc(f EvictionFunc) error {
+	c.lock.Lock()
+	defer c.lock.Unlock()
 	if c.cache.OnEvicted != nil {
 		return fmt.Errorf("lru cache eviction function is already set")
 	}

Original file line number	Diff line number	Diff line change
`@@ -47,6 +47,8 @@ func NewWithEvictionFunc(size int, f EvictionFunc) *Cache {`
`47`	`47`
`48`	`48`	`// SetEvictionFunc updates the eviction func`
`49`	`49`	`func (c *Cache) SetEvictionFunc(f EvictionFunc) error {`
	`50`	`+ c.lock.Lock()`
	`51`	`+ defer c.lock.Unlock()`
`50`	`52`	`if c.cache.OnEvicted != nil {`
`51`	`53`	`return fmt.Errorf("lru cache eviction function is already set")`
`52`	`54`	`}`