nginx-ui/internal/cache/cache.go
2024-10-24 18:22:45 +08:00

35 lines
753 B
Go

package cache
import (
"github.com/dgraph-io/ristretto"
"github.com/uozi-tech/cosy/logger"
"time"
)
var cache *ristretto.Cache[string, any]
func Init() {
var err error
cache, err = ristretto.NewCache[string, any](&ristretto.Config[string, any]{
NumCounters: 1e7, // number of keys to track frequency of (10M).
MaxCost: 1 << 30, // maximum cost of cache (1GB).
BufferItems: 64, // number of keys per Get buffer.
})
if err != nil {
logger.Fatal("initializing local cache err", err)
}
}
func Set(key string, value interface{}, ttl time.Duration) {
cache.SetWithTTL(key, value, 0, ttl)
cache.Wait()
}
func Get(key string) (value interface{}, ok bool) {
return cache.Get(key)
}
func Del(key string) {
cache.Del(key)
}