ristretto: use capacity in MB

This commit is contained in:
Martin Holst Swende 2019-09-27 14:08:27 +02:00
parent fa166501ad
commit 3002f867fa
No known key found for this signature in database
GPG key ID: 683B438C05A5DDF0

View file

@ -29,10 +29,12 @@ type RistrettoCache struct {
cache *ristretto.Cache cache *ristretto.Cache
} }
// NewRistrettoCache create a new ristretto cache with the given
// capacity in MB
func NewRistrettoCache(capacity int) (*RistrettoCache, error) { func NewRistrettoCache(capacity int) (*RistrettoCache, error) {
cache, err := ristretto.NewCache(&ristretto.Config{ cache, err := ristretto.NewCache(&ristretto.Config{
NumCounters: int64(capacity * 10), NumCounters: int64(capacity * 10),
MaxCost: int64(capacity), MaxCost: int64(1000 * 1000 * capacity),
BufferItems: 64, BufferItems: 64,
Metrics: false, Metrics: false,
KeyToHash: func(key interface{}) uint64 { KeyToHash: func(key interface{}) uint64 {