优化ttlcache相关代码

internal/utils/ttlcache/piece_test.go

@@ -0,0 +1,70 @@
+package ttlcache
+
+import (
+	"github.com/iwind/TeaGo/rands"
+	"testing"
+	"time"
+)
+
+func TestPiece_Add(t *testing.T) {
+	piece := NewPiece[int](10)
+	piece.Add(1, &Item[int]{expiresAt: time.Now().Unix() + 3600})
+	piece.Add(2, &Item[int]{})
+	piece.Add(3, &Item[int]{})
+	piece.Delete(3)
+	for key, item := range piece.m {
+		t.Log(key, item.Value)
+	}
+	t.Log(piece.Read(1))
+}
+
+func TestPiece_Add_Same(t *testing.T) {
+	var piece = NewPiece[int](10)
+	piece.Add(1, &Item[int]{expiresAt: time.Now().Unix() + 3600})
+	piece.Add(1, &Item[int]{expiresAt: time.Now().Unix() + 3600})
+	for key, item := range piece.m {
+		t.Log(key, item.Value)
+	}
+	t.Log(piece.Read(1))
+}
+
+func TestPiece_MaxItems(t *testing.T) {
+	piece := NewPiece[int](10)
+	for i := 0; i < 1000; i++ {
+		piece.Add(uint64(i), &Item[int]{expiresAt: time.Now().Unix() + 3600})
+	}
+	t.Log(len(piece.m))
+}
+
+func TestPiece_GC(t *testing.T) {
+	piece := NewPiece[int](10)
+	piece.Add(1, &Item[int]{Value: 1, expiresAt: time.Now().Unix() + 1})
+	piece.Add(2, &Item[int]{Value: 2, expiresAt: time.Now().Unix() + 1})
+	piece.Add(3, &Item[int]{Value: 3, expiresAt: time.Now().Unix() + 1})
+	t.Log("before gc ===")
+	for key, item := range piece.m {
+		t.Log(key, item.Value)
+	}
+
+	time.Sleep(1 * time.Second)
+	piece.GC()
+
+	t.Log("after gc ===")
+	for key, item := range piece.m {
+		t.Log(key, item.Value)
+	}
+}
+
+func TestPiece_GC2(t *testing.T) {
+	piece := NewPiece[int](10)
+	for i := 0; i < 10_000; i++ {
+		piece.Add(uint64(i), &Item[int]{Value: 1, expiresAt: time.Now().Unix() + int64(rands.Int(1, 10))})
+	}
+
+	time.Sleep(1 * time.Second)
+
+	before := time.Now()
+	piece.GC()
+	t.Log(time.Since(before).Seconds()*1000, "ms")
+	t.Log(piece.Count())
+}