mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-07 06:03: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) {
var cost int64
if cv, ok := any(value).(string); ok {
cost = int64(len(cv))
var keyCost int64
var valueCost int64
if ck, ok := any(key).(string); ok {
keyCost = int64(len(ck))
} 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) {