twitch-bot/twitch/twitchAPICache.go

58 lines
954 B
Go
Raw Permalink Normal View History

package twitch
import (
"crypto/sha256"
"fmt"
"strings"
"sync"
"time"
)
type (
APICache struct {
data map[string]twitchAPICacheEntry
lock sync.RWMutex
}
twitchAPICacheEntry struct {
Data interface{}
ValidUntil time.Time
}
)
func newTwitchAPICache() *APICache {
return &APICache{
data: make(map[string]twitchAPICacheEntry),
}
}
func (t *APICache) Get(key []string) interface{} {
t.lock.RLock()
defer t.lock.RUnlock()
e := t.data[t.deriveKey(key)]
if e.ValidUntil.Before(time.Now()) {
return nil
}
return e.Data
}
func (t *APICache) Set(key []string, valid time.Duration, data interface{}) {
t.lock.Lock()
defer t.lock.Unlock()
t.data[t.deriveKey(key)] = twitchAPICacheEntry{
Data: data,
ValidUntil: time.Now().Add(valid),
}
}
func (*APICache) deriveKey(key []string) string {
sha := sha256.New()
fmt.Fprintf(sha, "%s", strings.Join(key, ":"))
return fmt.Sprintf("sha256:%x", sha.Sum(nil))
}