1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
package fetchers
import (
"llostre/backend/general"
"llostre/backend/notes"
"sync"
"time"
)
type CacheEntry struct {
Note *notes.Note
ExpiresAt time.Time
}
type Cache struct {
posts map[string]*CacheEntry
profiles map[string]*ProfileCacheEntry
mu sync.RWMutex
}
func NewCache() *Cache {
return &Cache{
posts: make(map[string]*CacheEntry),
profiles: make(map[string]*ProfileCacheEntry),
}
}
func (c *Cache) Get(uri string) (*notes.Note, bool) {
c.mu.Lock()
defer c.mu.Unlock()
entry, exists := c.posts[uri]
if !exists {
return nil, false
}
if time.Now().After(entry.ExpiresAt) {
delete(c.posts, uri)
return nil, false
}
return entry.Note, true
}
func (c *Cache) Set(uri string, note *notes.Note) {
c.mu.Lock()
defer c.mu.Unlock()
c.posts[uri] = &CacheEntry{
Note: note,
ExpiresAt: time.Now().Add(15 * time.Minute),
}
}
func (c *Cache) GetMany(uris []string) map[string]*notes.Note {
c.mu.Lock()
defer c.mu.Unlock()
result := make(map[string]*notes.Note)
now := time.Now()
for _, uri := range uris {
entry, exists := c.posts[uri]
if !exists {
continue
}
if now.After(entry.ExpiresAt) {
delete(c.posts, uri)
continue
}
result[uri] = entry.Note
}
return result
}
type ProfileCacheEntry struct {
Profile *general.Profile
ExpiresAt time.Time
}
func (c *Cache) GetProfile(did string) (*general.Profile, bool) {
c.mu.Lock()
defer c.mu.Unlock()
entry, exists := c.profiles[did]
if !exists {
return nil, false
}
if time.Now().After(entry.ExpiresAt) {
delete(c.profiles, did)
return nil, false
}
return entry.Profile, true
}
func (c *Cache) SetProfile(did string, profile *general.Profile) {
c.mu.Lock()
defer c.mu.Unlock()
c.profiles[did] = &ProfileCacheEntry{
Profile: profile,
ExpiresAt: time.Now().Add(15 * time.Minute),
}
}
|
