Merge branch 'add-methodsto-sizeconstrainedcache' into include-timebo…

…osted-receiptsfield
OffchainLabs · Nov 14, 2024 · 01a4427 · 01a4427
2 parents 5c7cde2 + 68cb86d
commit 01a4427
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/common/lru/blob_lru.go b/common/lru/blob_lru.go
@@ -82,3 +82,21 @@ func (c *SizeConstrainedCache[K, V]) Get(key K) (V, bool) {
 
 	return c.lru.Get(key)
 }
+
+func (c *SizeConstrainedCache[K, V]) Remove(key K) {
+	c.lock.Lock()
+	defer c.lock.Unlock()
+
+	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
+}