65 lines
1.0 KiB
Go
65 lines
1.0 KiB
Go
package pokecache
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// Cache -
|
|
type Cache struct {
|
|
cache map[string]cacheEntry
|
|
mux *sync.Mutex
|
|
}
|
|
|
|
type cacheEntry struct {
|
|
createdAt time.Time
|
|
val []byte
|
|
}
|
|
|
|
// NewCache -
|
|
func NewCache(interval time.Duration) Cache {
|
|
c := Cache{
|
|
cache: make(map[string]cacheEntry),
|
|
mux: &sync.Mutex{},
|
|
}
|
|
|
|
go c.reapLoop(interval)
|
|
|
|
return c
|
|
}
|
|
|
|
// Add -
|
|
func (c *Cache) Add(key string, value []byte) {
|
|
c.mux.Lock()
|
|
defer c.mux.Unlock()
|
|
c.cache[key] = cacheEntry{
|
|
createdAt: time.Now().UTC(),
|
|
val: value,
|
|
}
|
|
}
|
|
|
|
// Get -
|
|
func (c *Cache) Get(key string) ([]byte, bool) {
|
|
c.mux.Lock()
|
|
defer c.mux.Unlock()
|
|
val, ok := c.cache[key]
|
|
return val.val, ok
|
|
}
|
|
|
|
func (c *Cache) reapLoop(interval time.Duration) {
|
|
ticker := time.NewTicker(interval)
|
|
for range ticker.C {
|
|
c.reap(time.Now().UTC(), interval)
|
|
}
|
|
}
|
|
|
|
func (c *Cache) reap(now time.Time, last time.Duration) {
|
|
c.mux.Lock()
|
|
defer c.mux.Unlock()
|
|
for k, v := range c.cache {
|
|
if v.createdAt.Before(now.Add(-last)) {
|
|
delete(c.cache, k)
|
|
}
|
|
}
|
|
}
|