50 lines
729 B
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)
|
|
}
|