-
Notifications
You must be signed in to change notification settings - Fork 2
/
cached_doc.go
76 lines (67 loc) · 2.13 KB
/
cached_doc.go
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
package db
import (
"config-service/db/mongo"
"config-service/types"
"context"
"fmt"
"sync"
"time"
"go.mongodb.org/mongo-driver/bson"
"go.uber.org/zap"
)
var cachedDocuments = sync.Map{}
func AddCachedDocument[T types.DocContent](cacheKey, collection string, filterBuilder *FilterBuilder, updateInterval time.Duration) {
cachedDocuments.Store(cacheKey, newCachedDocument[T](collection, filterBuilder.get(), updateInterval))
}
func GetCachedDocument[T types.DocContent](cacheKey string) (T, error) {
if i, ok := cachedDocuments.Load(cacheKey); ok {
if cachedDoc, ok := i.(*cachedDocument[T]); ok {
return cachedDoc.get()
}
return nil, fmt.Errorf("documented cached with key: %s does not match parametric type", cacheKey)
}
return nil, fmt.Errorf("cached document %s not found", cacheKey)
}
type cachedDocument[T types.DocContent] struct {
doc T
lastRefreshError error
timeUpdated time.Time
mutex sync.RWMutex
updateInterval time.Duration
queryFilter bson.D
collection string
}
func newCachedDocument[T types.DocContent](collection string, queryFilter bson.D, updateInterval time.Duration) *cachedDocument[T] {
return &cachedDocument[T]{
doc: nil,
updateInterval: updateInterval,
queryFilter: queryFilter,
collection: collection,
mutex: sync.RWMutex{},
timeUpdated: time.Time{},
}
}
func (c *cachedDocument[T]) get() (T, error) {
c.refresh()
c.mutex.RLock()
defer c.mutex.RUnlock()
return c.doc, c.lastRefreshError
}
func (c *cachedDocument[T]) refresh() {
if time.Since(c.timeUpdated) > c.updateInterval {
c.mutex.Lock()
defer c.mutex.Unlock()
//check if not updated by another thread
if time.Since(c.timeUpdated) > c.updateInterval {
var doc T
if err := mongo.GetReadCollection(c.collection).FindOne(context.Background(), c.queryFilter).Decode(&doc); err != nil {
zap.L().Error("Failed to refresh cached document", zap.Error(err), zap.String("collection", c.collection), zap.Any("queryFilter", c.queryFilter))
c.lastRefreshError = err
return
}
c.doc = doc
c.lastRefreshError = nil
c.timeUpdated = time.Now()
}
}
}