feature: add Cap interface{}

Author: zhuliquan

arc/arc.go

@@ -202,6 +202,11 @@ func (c *ARCCache[K, V]) Len() int {
 	return c.t1.Len() + c.t2.Len()
 }
 
+// Cap returns the capacity of cache
+func (c *ARCCache[K, V]) Cap() int {
+	return c.Cap()
+}
+
 // Keys returns all the cached keys
 func (c *ARCCache[K, V]) Keys() []K {
 	c.lock.RLock()

arc/arc_test.go

@@ -313,6 +313,9 @@ func TestARC(t *testing.T) {
 	if l.Len() != 128 {
 		t.Fatalf("bad len: %v", l.Len())
 	}
+	if l.Cap() != 128 {
+		t.Fatalf("expect %d, but %d", 128, l.Cap())
+	}
 
 	for i, k := range l.Keys() {
 		if v, ok := l.Get(k); !ok || v != k || v != i+128 {
@@ -340,6 +343,9 @@ func TestARC(t *testing.T) {
 			t.Fatalf("should be deleted")
 		}
 	}
+	if l.Cap() != 128 {
+		t.Fatalf("expect %d, but %d", 128, l.Cap())
+	}
 
 	l.Purge()
 	if l.Len() != 0 {
@@ -348,6 +354,9 @@ func TestARC(t *testing.T) {
 	if _, ok := l.Get(200); ok {
 		t.Fatalf("should contain nothing")
 	}
+	if l.Cap() != 128 {
+		t.Fatalf("expect %d, but %d", 128, l.Cap())
+	}
 }
 
 // Test that Contains doesn't update recent-ness

lru.go

@@ -248,3 +248,8 @@ func (c *Cache[K, V]) Len() int {
 	c.lock.RUnlock()
 	return length
 }
+
+// Cap returns the capacity of the cache
+func (c *Cache[K, V]) Cap() int {
+	return c.lru.Cap()
+}

lru_test.go

@@ -85,6 +85,9 @@ func TestLRU(t *testing.T) {
 	if l.Len() != 128 {
 		t.Fatalf("bad len: %v", l.Len())
 	}
+	if l.Cap() != 128 {
+		t.Fatalf("expect %d, but %d", 128, l.Cap())
+	}
 
 	if evictCounter != 128 {
 		t.Fatalf("bad evict count: %v", evictCounter)

simplelru/lru.go

@@ -148,6 +148,11 @@ func (c *LRU[K, V]) Len() int {
 	return c.evictList.length()
 }
 
+// Cap returns the capacity of the cache
+func (c *LRU[K, V]) Cap() int {
+	return c.size
+}
+
 // Resize changes the cache size.
 func (c *LRU[K, V]) Resize(size int) (evicted int) {
 	diff := c.Len() - size

simplelru/lru_interface.go

@@ -38,6 +38,9 @@ type LRUCache[K comparable, V any] interface {
 	// Returns the number of items in the cache.
 	Len() int
 
+	// Returns the capacity of the cache.
+	Cap() int
+
 	// Clears all cache entries.
 	Purge()
 

simplelru/lru_test.go

@@ -24,6 +24,9 @@ func TestLRU(t *testing.T) {
 	if l.Len() != 128 {
 		t.Fatalf("bad len: %v", l.Len())
 	}
+	if l.Cap() != 128 {
+		t.Fatalf("expect %d, but %d", 128, l.Cap())
+	}
 
 	if evictCounter != 128 {
 		t.Fatalf("bad evict count: %v", evictCounter)