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:41:42 +01:00
parent 858cc61f92
commit 4a648a9052
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
internal/caching/impl_ristretto.go
View file

@ -147,13 +147,19 @@ func (c *RistrettoCachePartition[K, V]) setWithCost(key K, value V, cost int64)
} }
func (c *RistrettoCachePartition[K, V]) Set(key K, value V) { func (c *RistrettoCachePartition[K, V]) Set(key K, value V) {
var cost int64 var keyCost int64
if cv, ok := any(value).(string); ok { var valueCost int64
cost = int64(len(cv)) if ck, ok := any(key).(string); ok {
keyCost = int64(len(ck))
} else { } else {
cost = int64(unsafe.Sizeof(value)) keyCost = int64(unsafe.Sizeof(key))
} }
c.setWithCost(key, value, cost) if cv, ok := any(value).(string); ok {
valueCost = int64(len(cv))
} else {
valueCost = int64(unsafe.Sizeof(value))
}
c.setWithCost(key, value, keyCost+valueCost)
} }
func (c *RistrettoCachePartition[K, V]) Unset(key K) { func (c *RistrettoCachePartition[K, V]) Unset(key K) {