jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-02-06 15:10:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Simon Ding
2024-08-07 11:20:21 +08:00
parent f48b3c657e
commit 1aa3dca2c6
1 changed files with 9 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
pkg/cache/cache.go vendored
View File

@@ -4,35 +4,20 @@ import (
"polaris/log" "polaris/log"
"polaris/pkg/utils" "polaris/pkg/utils"
"time" "time"
"github.com/robfig/cron"
) )
func NewCache[T comparable, S any](timeout time.Duration) *Cache[T, S] { func NewCache[T comparable, S any](timeout time.Duration) *Cache[T, S] {
c := &Cache[T, S]{ c := &Cache[T, S]{
m: utils.Map[T, inner[S]]{}, m: utils.Map[T, inner[S]]{},
timeout: timeout, timeout: timeout,
cr: cron.New(),
} }
c.cr.AddFunc("@ervery 1m", func() {
c.m.Range(func(key T, value inner[S]) bool {
if time.Since(value.t) > c.timeout {
log.Debugf("delete old cache: %v", key)
c.m.Delete(key)
}
return true
})
})
c.cr.Start()
return c return c
} }
type Cache[T comparable, S any] struct { type Cache[T comparable, S any] struct {
m utils.Map[T, inner[S]] m utils.Map[T, inner[S]]
timeout time.Duration timeout time.Duration
cr *cron.Cron
} }
type inner[S any] struct { type inner[S any] struct {
@@ -45,6 +30,15 @@ func (c *Cache[T, S]) Set(key T, value S) {
} }
func (c *Cache[T, S]) Get(key T) (S, bool) { func (c *Cache[T, S]) Get(key T) (S, bool) {
c.m.Range(func(key T, value inner[S]) bool {
if time.Since(value.t) > c.timeout {
log.Debugf("delete old cache: %v", key)
c.m.Delete(key)
}
return true
})
v, ok := c.m.Load(key) v, ok := c.m.Load(key)
if !ok { if !ok {
return getZero[S](), ok return getZero[S](), ok