implement caching #1
64
relay/cache/cache.go
vendored
Normal file
64
relay/cache/cache.go
vendored
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
package cache
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type item struct {
|
||||||
|
value interface{}
|
||||||
|
expiration int64
|
||||||
|
}
|
||||||
|
|
||||||
|
type Cache struct {
|
||||||
|
items map[string]item
|
||||||
|
mu sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
func New() *Cache {
|
||||||
|
cache := &Cache{
|
||||||
|
items: make(map[string]item),
|
||||||
|
}
|
||||||
|
go cache.startCleanup()
|
||||||
|
return cache
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Cache) Set(key string, value interface{}, duration time.Duration) {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
|
||||||
|
c.items[key] = item{
|
||||||
|
value: value,
|
||||||
|
expiration: time.Now().Add(duration).UnixNano(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Cache) Get(key string) (interface{}, bool) {
|
||||||
|
c.mu.RLock()
|
||||||
|
defer c.mu.RUnlock()
|
||||||
|
|
||||||
|
item, exists := c.items[key]
|
||||||
|
if !exists {
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
|
||||||
|
if time.Now().UnixNano() > item.expiration {
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
|
||||||
|
return item.value, true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Cache) startCleanup() {
|
||||||
|
ticker := time.NewTicker(time.Minute)
|
||||||
|
for range ticker.C {
|
||||||
|
c.mu.Lock()
|
||||||
|
now := time.Now().UnixNano()
|
||||||
|
for k, v := range c.items {
|
||||||
|
if now > v.expiration {
|
||||||
|
delete(c.items, k)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
c.mu.Unlock()
|
||||||
|
}
|
||||||
|
}
|
@ -4,6 +4,8 @@ import (
|
|||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"git.highperfocused.tech/highperfocused/lumina-relay/relay/cache"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Post struct {
|
type Post struct {
|
||||||
@ -16,8 +18,17 @@ type Post struct {
|
|||||||
ReactionCount int `json:"reaction_count"`
|
ReactionCount int `json:"reaction_count"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
trendingCache = cache.New()
|
||||||
|
cacheDuration = 5 * time.Minute
|
||||||
|
)
|
||||||
|
|
||||||
// GetTrendingKind20 returns the top 20 trending posts of kind 20 from the last 24 hours
|
// GetTrendingKind20 returns the top 20 trending posts of kind 20 from the last 24 hours
|
||||||
func GetTrendingKind20(db *sql.DB) ([]Post, error) {
|
func GetTrendingKind20(db *sql.DB) ([]Post, error) {
|
||||||
|
if cached, ok := trendingCache.Get("trending_kind_20"); ok {
|
||||||
|
return cached.([]Post), nil
|
||||||
|
}
|
||||||
|
|
||||||
query := `
|
query := `
|
||||||
WITH reactions AS (
|
WITH reactions AS (
|
||||||
SELECT
|
SELECT
|
||||||
@ -65,5 +76,6 @@ func GetTrendingKind20(db *sql.DB) ([]Post, error) {
|
|||||||
trendingPosts = append(trendingPosts, post)
|
trendingPosts = append(trendingPosts, post)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
trendingCache.Set("trending_kind_20", trendingPosts, cacheDuration)
|
||||||
return trendingPosts, nil
|
return trendingPosts, nil
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user