Add login and post cache helpers
This commit is contained in:
parent
8ab29d89da
commit
59ea5da521
1 changed files with 59 additions and 0 deletions
59
cache.go
Normal file
59
cache.go
Normal file
|
@ -0,0 +1,59 @@
|
||||||
|
package writefreely
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
postsCacheTime = 4 * time.Second
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
postsCacheItem struct {
|
||||||
|
Expire time.Time
|
||||||
|
Posts *[]PublicPost
|
||||||
|
ready chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
AuthCache struct {
|
||||||
|
Alias, Pass, Token string
|
||||||
|
BadPasses map[string]bool
|
||||||
|
|
||||||
|
expire time.Time
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
userPostsCache = struct {
|
||||||
|
sync.RWMutex
|
||||||
|
users map[int64]postsCacheItem
|
||||||
|
}{
|
||||||
|
users: map[int64]postsCacheItem{},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func CachePosts(userID int64, p *[]PublicPost) {
|
||||||
|
close(userPostsCache.users[userID].ready)
|
||||||
|
userPostsCache.Lock()
|
||||||
|
userPostsCache.users[userID] = postsCacheItem{
|
||||||
|
Expire: time.Now().Add(postsCacheTime),
|
||||||
|
Posts: p,
|
||||||
|
}
|
||||||
|
userPostsCache.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetPostsCache(userID int64) *[]PublicPost {
|
||||||
|
userPostsCache.RLock()
|
||||||
|
pci, ok := userPostsCache.users[userID]
|
||||||
|
userPostsCache.RUnlock()
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if pci.Expire.Before(time.Now()) {
|
||||||
|
// Cache is expired
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return pci.Posts
|
||||||
|
}
|
Loading…
Add table
Reference in a new issue