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

Include key in cost

Browse source
This commit is contained in:
Neil Alexander 2022-07-07 11:47:01 +01:00
parent 4a648a9052
commit caecdedeed
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 5 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
internal/caching/impl_ristretto.go
View file

@ -143,23 +143,17 @@ func (c *RistrettoCachePartition[K, V]) setWithCost(key K, value V, cost int64)
panic(fmt.Sprintf("invalid use of immutable cache tries to change value of %v from %v to %v", key, v, value)) panic(fmt.Sprintf("invalid use of immutable cache tries to change value of %v from %v to %v", key, v, value))
} }
} }
c.cache.SetWithTTL(bkey, value, cost, c.MaxAge) c.cache.SetWithTTL(bkey, value, int64(len(bkey))+cost, c.MaxAge)
} }
func (c *RistrettoCachePartition[K, V]) Set(key K, value V) { func (c *RistrettoCachePartition[K, V]) Set(key K, value V) {
var keyCost int64 var cost int64
var valueCost int64
if ck, ok := any(key).(string); ok {
keyCost = int64(len(ck))
} else {
keyCost = int64(unsafe.Sizeof(key))
}
if cv, ok := any(value).(string); ok { if cv, ok := any(value).(string); ok {
valueCost = int64(len(cv)) cost = int64(len(cv))
} else { } else {
valueCost = int64(unsafe.Sizeof(value)) cost = int64(unsafe.Sizeof(value))
} }
c.setWithCost(key, value, keyCost+valueCost) c.setWithCost(key, value, cost)
} }
func (c *RistrettoCachePartition[K, V]) Unset(key K) { func (c *RistrettoCachePartition[K, V]) Unset(key K) {