forked from p53841790/wheat-cache
103 lines
1.9 KiB
Go
103 lines
1.9 KiB
Go
package lru
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"gitee.com/timedb/wheatCache/pkg/proto"
|
|
"gitee.com/timedb/wheatCache/pkg/structure/stringx"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNewLRUCache(t *testing.T) {
|
|
cache := NewLRUCache()
|
|
v1 := stringx.NewStringSingle()
|
|
v2 := stringx.NewStringSingle()
|
|
v3 := stringx.NewStringSingle()
|
|
key1 := proto.BaseKey{
|
|
Key: "1",
|
|
}
|
|
key2 := proto.BaseKey{
|
|
Key: "2",
|
|
}
|
|
key3 := proto.BaseKey{
|
|
Key: "3",
|
|
}
|
|
cache.Add(&key1, v1)
|
|
cache.Add(&key2, v2)
|
|
cache.Add(&key3, v3)
|
|
cache.Add(&key1, v1)
|
|
fmt.Println(cache.nowSize)
|
|
cache.Del()
|
|
fmt.Println(cache.nowSize)
|
|
_, isTrue := cache.Get(&key1)
|
|
require.Equal(t, isTrue, true)
|
|
}
|
|
|
|
func TestNewLRUCache2(t *testing.T) {
|
|
//根据key删除
|
|
cache := NewLRUCache()
|
|
v1 := stringx.NewStringSingle()
|
|
v2 := stringx.NewStringSingle()
|
|
v3 := stringx.NewStringSingle()
|
|
key1 := proto.BaseKey{
|
|
Key: "1",
|
|
}
|
|
key2 := proto.BaseKey{
|
|
Key: "2",
|
|
}
|
|
key3 := proto.BaseKey{
|
|
Key: "3",
|
|
}
|
|
cache.Add(&key1, v1)
|
|
cache.Add(&key2, v2)
|
|
cache.Add(&key3, v3)
|
|
cache.DelByKey(&key1)
|
|
_, ok := cache.Get(&key1)
|
|
require.Equal(t, ok, false)
|
|
require.Error(t, cache.DelByKey(&key1))
|
|
}
|
|
|
|
func TestLruProcess(t *testing.T) {
|
|
lru := NewLRUCache()
|
|
lru.clearSize = 1000
|
|
|
|
for i := 100; i < 200; i++ {
|
|
lru.Add(&proto.BaseKey{
|
|
Key: fmt.Sprint(i),
|
|
Ttl: 20 << 2,
|
|
}, stringx.NewStringSingle())
|
|
}
|
|
|
|
// mock LruKey
|
|
for i := 0; i < 100; i++ {
|
|
lru.Add(&proto.BaseKey{
|
|
Key: fmt.Sprint(i),
|
|
Ttl: 4,
|
|
}, stringx.NewStringSingle())
|
|
}
|
|
|
|
require.Equal(t, lru.nowSize, int64(200*8))
|
|
|
|
// 自动清理测试
|
|
time.Sleep(3 * time.Second)
|
|
fmt.Println(lru.nowSize)
|
|
require.Less(t, lru.nowSize, lru.clearSize+1)
|
|
|
|
// TTL 测试
|
|
time.Sleep(2 * time.Second)
|
|
require.Equal(t, lru.li.Len(), 25)
|
|
|
|
// 过期全部的 Key
|
|
for i := 100; i < 200; i++ {
|
|
lru.UpdateTTl(&proto.BaseKey{
|
|
Key: fmt.Sprint(i),
|
|
Ttl: -1,
|
|
})
|
|
}
|
|
|
|
time.Sleep(2 * time.Second)
|
|
require.Equal(t, lru.nowSize, int64(0))
|
|
}
|