country-geo-locations/pkg/geoloc/cache.go

50 lines
729 B
Go

package geoloc
import (
"sync"
"time"
)
var ipCache *Cache
type Cache struct {
store sync.Map
}
func NewCache() *Cache {
return &Cache{}
}
func (c *Cache) Set(key, value uint, ttl time.Duration) {
c.store.Store(key, value)
time.AfterFunc(ttl, func() {
c.store.Delete(key)
})
}
func (c *Cache) Get(key uint) (uint, bool) {
output, _ := c.store.Load(key)
value, ok := output.(uint)
return value, ok
}
func createCache() {
ipCache = NewCache()
}
func GetCacheContent(key uint) (uint, bool) {
if ipCache == nil {
createCache()
return 0, false
}
return ipCache.Get(key)
}
func SetCacheContent(key, value uint, ttl time.Duration) {
if ipCache == nil {
createCache()
}
ipCache.Set(key, value, ttl)
}