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
|
package cache
import (
"sync"
"time"
"github.com/patrickmn/go-cache"
"gitlab.com/gitlab-org/gitlab-pages/internal/config"
"gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/api"
)
type memstore struct {
store *cache.Cache
mux *sync.RWMutex
retriever *Retriever
entryRefreshTimeout time.Duration
}
func newMemStore(client api.Client, cc *config.Cache) Store {
retriever := NewRetriever(client, cc.RetrievalTimeout, cc.MaxRetrievalInterval, cc.MaxRetrievalRetries)
return &memstore{
store: cache.New(cc.CacheExpiry, cc.CacheCleanupInterval),
mux: &sync.RWMutex{},
retriever: retriever,
entryRefreshTimeout: cc.EntryRefreshTimeout,
}
}
// LoadOrCreate writes or retrieves a domain entry from the cache in a
// thread-safe way, trying to make this read-preferring RW locking.
func (m *memstore) LoadOrCreate(domain string) *Entry {
m.mux.RLock()
entry, exists := m.store.Get(domain)
m.mux.RUnlock()
if exists {
return entry.(*Entry)
}
m.mux.Lock()
defer m.mux.Unlock()
if entry, exists = m.store.Get(domain); exists {
return entry.(*Entry)
}
newEntry := newCacheEntry(domain, m.entryRefreshTimeout, m.retriever)
m.store.SetDefault(domain, newEntry)
return newEntry
}
func (m *memstore) ReplaceOrCreate(domain string, entry *Entry) *Entry {
m.mux.Lock()
defer m.mux.Unlock()
m.store.Delete(domain)
m.store.SetDefault(domain, entry)
return entry
}
|