mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package synccache
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/zijiren233/gencontainer/rwmap"
|
|
)
|
|
|
|
type SyncCache[K comparable, V any] struct {
|
|
cache rwmap.RWMap[K, *entry[V]]
|
|
ticker *time.Ticker
|
|
}
|
|
|
|
func NewSyncCache[K comparable, V any](trimTime time.Duration) *SyncCache[K, V] {
|
|
sc := &SyncCache[K, V]{
|
|
ticker: time.NewTicker(trimTime),
|
|
}
|
|
go func() {
|
|
for range sc.ticker.C {
|
|
sc.trim()
|
|
}
|
|
}()
|
|
return sc
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) Releases() {
|
|
sc.ticker.Stop()
|
|
sc.cache.Clear()
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) trim() {
|
|
sc.cache.Range(func(key K, value *entry[V]) bool {
|
|
if value.IsExpired() {
|
|
sc.cache.Delete(key)
|
|
}
|
|
return true
|
|
})
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) Store(key K, value V, expire time.Duration) {
|
|
sc.LoadOrStore(key, value, expire)
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) Load(key K) (value V, loaded bool) {
|
|
e, ok := sc.cache.Load(key)
|
|
if ok && !e.IsExpired() {
|
|
return e.value, ok
|
|
}
|
|
return
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) LoadOrStore(key K, value V, expire time.Duration) (actual V, loaded bool) {
|
|
e, loaded := sc.cache.LoadOrStore(key, &entry[V]{
|
|
expiration: time.Now().Add(expire),
|
|
value: value,
|
|
})
|
|
if e.IsExpired() {
|
|
sc.cache.Store(key, &entry[V]{
|
|
expiration: time.Now().Add(expire),
|
|
value: value,
|
|
})
|
|
return value, false
|
|
}
|
|
return e.value, loaded
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) Delete(key K) {
|
|
sc.LoadAndDelete(key)
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) LoadAndDelete(key K) (value V, loaded bool) {
|
|
e, loaded := sc.cache.LoadAndDelete(key)
|
|
if loaded && !e.IsExpired() {
|
|
return e.value, loaded
|
|
}
|
|
return
|
|
}
|
|
|
|
func (sc *SyncCache[K, V]) Clear() {
|
|
sc.cache.Clear()
|
|
}
|