71 lines
1.2 KiB
Go
71 lines
1.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type cacheEntry struct {
|
|
data interface{}
|
|
expiresAt time.Time
|
|
}
|
|
|
|
type dashboardCache struct {
|
|
mu sync.RWMutex
|
|
entries map[string]*cacheEntry
|
|
}
|
|
|
|
func newDashboardCache() *dashboardCache {
|
|
return &dashboardCache{
|
|
entries: make(map[string]*cacheEntry),
|
|
}
|
|
}
|
|
|
|
func (c *dashboardCache) Get(key string) (interface{}, bool) {
|
|
c.mu.RLock()
|
|
entry, ok := c.entries[key]
|
|
if !ok {
|
|
c.mu.RUnlock()
|
|
return nil, false
|
|
}
|
|
if time.Now().Before(entry.expiresAt) {
|
|
data := entry.data
|
|
c.mu.RUnlock()
|
|
return data, true
|
|
}
|
|
c.mu.RUnlock()
|
|
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
entry, ok = c.entries[key]
|
|
if ok && time.Now().After(entry.expiresAt) {
|
|
delete(c.entries, key)
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func (c *dashboardCache) Set(key string, data interface{}, ttl time.Duration) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
|
|
c.entries[key] = &cacheEntry{
|
|
data: data,
|
|
expiresAt: time.Now().Add(ttl),
|
|
}
|
|
}
|
|
|
|
func (c *dashboardCache) Invalidate(prefixes ...string) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
|
|
for key := range c.entries {
|
|
for _, prefix := range prefixes {
|
|
if strings.HasPrefix(key, prefix) {
|
|
delete(c.entries, key)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|