address PR comments

OffchainLabs · Nov 12, 2024 · 68cb86d · 68cb86d
1 parent ecbb71b
commit 68cb86d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/common/lru/blob_lru.go b/common/lru/blob_lru.go
@@ -87,12 +87,16 @@ func (c *SizeConstrainedCache[K, V]) Remove(key K) {
 	c.lock.Lock()
 	defer c.lock.Unlock()
 
-	c.lru.Remove(key)
+	if v, ok := c.lru.Peek(key); ok {
+		c.size -= uint64(len(v))
+		c.lru.Remove(key)
+	}
 }
 
 func (c *SizeConstrainedCache[K, V]) Clear() {
 	c.lock.Lock()
 	defer c.lock.Unlock()
 
 	c.lru.Purge()
+	c.size = 0
 }