mirror of
https://github.com/SagerNet/sing.git
synced 2025-04-03 11:57:39 +03:00
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
package freelru_test
|
|
|
|
import (
|
|
"github.com/sagernet/sing/common"
|
|
F "github.com/sagernet/sing/common/format"
|
|
"math/rand/v2"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/sagernet/sing/contrab/freelru"
|
|
"github.com/sagernet/sing/contrab/maphash"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUpdateLifetimeOnGet(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.AddWithLifetime("hello", "world", 2*time.Second)
|
|
time.Sleep(time.Second)
|
|
_, ok := lru.GetAndRefresh("hello")
|
|
require.True(t, ok)
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok = lru.Get("hello")
|
|
require.True(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetimeOnGet1(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.AddWithLifetime("hello", "world", 2*time.Second)
|
|
time.Sleep(time.Second)
|
|
lru.Peek("hello")
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok := lru.Get("hello")
|
|
require.False(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetime(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.Add("hello", "world")
|
|
require.True(t, lru.UpdateLifetime("hello", "world", 2*time.Second))
|
|
time.Sleep(time.Second)
|
|
_, ok := lru.Get("hello")
|
|
require.True(t, ok)
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok = lru.Get("hello")
|
|
require.False(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetime1(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.Add("hello", "world")
|
|
require.False(t, lru.UpdateLifetime("hello", "not world", 2*time.Second))
|
|
time.Sleep(2*time.Second + time.Millisecond*100)
|
|
_, ok := lru.Get("hello")
|
|
require.True(t, ok)
|
|
}
|
|
|
|
func TestUpdateLifetime2(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.AddWithLifetime("hello", "world", 2*time.Second)
|
|
time.Sleep(time.Second)
|
|
require.True(t, lru.UpdateLifetime("hello", "world", 2*time.Second))
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok := lru.Get("hello")
|
|
require.True(t, ok)
|
|
time.Sleep(time.Second + time.Millisecond*100)
|
|
_, ok = lru.Get("hello")
|
|
require.False(t, ok)
|
|
}
|
|
|
|
func TestPurgeExpired(t *testing.T) {
|
|
t.Parallel()
|
|
lru, err := freelru.New[string, *string](1024, maphash.NewHasher[string]().Hash32)
|
|
require.NoError(t, err)
|
|
lru.SetLifetime(time.Second)
|
|
lru.SetOnEvict(func(s string, s2 *string) {
|
|
if s2 == nil {
|
|
t.Fail()
|
|
}
|
|
})
|
|
for i := 0; i < 100; i++ {
|
|
lru.AddWithLifetime("hello_"+F.ToString(i), common.Ptr("world_"+F.ToString(i)), time.Duration(rand.Int32N(3000))*time.Millisecond)
|
|
}
|
|
for i := 0; i < 5; i++ {
|
|
time.Sleep(time.Second)
|
|
lru.GetAndRefreshOrAdd("hellox"+F.ToString(i), func() (*string, bool) {
|
|
return common.Ptr("worldx"), true
|
|
})
|
|
}
|
|
}
|