mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-01 03:03:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix bugs

Browse source
This commit is contained in:
Neil Alexander 2022-03-01 16:28:39 +00:00
parent 2e2d40b7ba
commit 8e71afb828
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 12 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
internal/caching/impl_inmemorylru.go
View file

@ -126,10 +126,12 @@ func NewInMemoryLRUCachePartition(name string, mutable bool, maxEntries int, max
func (c *InMemoryLRUCachePartition) Set(key string, value interface{}) {
if !c.mutable {
if peek, ok := c.lru.Peek(key); ok && peek != value {
if peek, ok := c.lru.Peek(key); ok {
if entry, ok := peek.(*inMemoryLRUCacheEntry); ok && entry.value != value {
panic(fmt.Sprintf("invalid use of immutable cache tries to mutate existing value of %q", key))
}
}
}
c.lru.Add(key, &inMemoryLRUCacheEntry{
value: value,
created: time.Now(),
@ -149,12 +151,15 @@ func (c *InMemoryLRUCachePartition) Get(key string) (value interface{}, ok bool)
return nil, false
}
entry, ok := v.(*inMemoryLRUCacheEntry)
if !ok {
return nil, false
}
if c.maxAge > 0 && time.Since(entry.created) > c.maxAge {
switch {
case ok && c.maxAge == CacheNoMaxAge:
return entry.value, ok // There's no maximum age policy
case ok && time.Since(entry.created) < c.maxAge:
return entry.value, ok // The value for the key isn't stale
default:
// Either the key was found and it was stale, or the key
// wasn't found at all
c.lru.Remove(key)
return nil, false
}
return entry.value, true
}