status-go/services/wallet/balance/ttl_cache.go

package balance

import (
	"context"
	"math/big"
	"time"

	"github.com/jellydator/ttlcache/v3"
	"go.uber.org/zap"

	"github.com/status-im/status-go/common"
	"github.com/status-im/status-go/logutils"
)

var (
	defaultTTLValue = 5 * time.Minute
)

func NewCacherWithTTL(ttl time.Duration) Cacher {
	return newCacherImpl(newCacheWithTTL(ttl))
}

// TTL cache implementation of cacheIface
type ttlCache[K comparable, V any] struct {
	cache *ttlcache.Cache[K, V]
}

//nolint:golint,unused // linter does not detect using it via reflect
func (c *ttlCache[K, V]) get(key K) V {
	item := c.cache.Get(key)
	if item == nil {
		var v V
		return v
	}
	return item.Value()
}

//nolint:golint,unused // linter does not detect using it via reflect
func (c *ttlCache[K, V]) set(key K, value V) {
	_ = c.cache.Set(key, value, ttlcache.DefaultTTL)
}

//nolint:golint,unused // linter does not detect using it via reflect
func (c *ttlCache[K, V]) len() int {
	return c.cache.Len()
}

//nolint:golint,unused // linter does not detect using it via reflect
func (c *ttlCache[K, V]) keys() []K {
	return c.cache.Keys()
}

//nolint:golint,unused // linter does not detect using it via reflect
func (c *ttlCache[K, V]) init() {
	c.cache = ttlcache.New[K, V](
		ttlcache.WithTTL[K, V](defaultTTLValue),
	)
	c.cache.OnEviction(func(ctx context.Context, reason ttlcache.EvictionReason, item *ttlcache.Item[K, V]) {
		logutils.ZapLogger().Debug("Evicting item from balance/nonce cache",
			zap.Int("reason", int(reason)),
			zap.Any("key", item.Key),
			zap.Any("value", item.Value),
		)
	})
	go func() { // starts automatic expired item deletion
		defer common.LogOnPanic()
		c.cache.Start()
	}()
}

//nolint:golint,unused // linter does not detect using it via reflect
func (c *ttlCache[K, V]) clear() {
	c.cache.DeleteAll()
}

// specializes generic cache
type cacheWithTTL struct {
	// TODO: use ttlCache instead of mapCache for nonceRangeCache. For that we need to update sortedRanges on item eviction
	// For now, nonceRanges cache is not updated on nonces items eviction, but it should not be as big as nonceCache is
	genericCache[*ttlCache[uint64, *big.Int], *ttlCache[uint64, *int64], *mapCache[int64, nonceRange]]
}

func newCacheWithTTL(ttl time.Duration) *cacheWithTTL {
	defaultTTLValue = ttl

	return &cacheWithTTL{
		genericCache: genericCache[*ttlCache[uint64, *big.Int], *ttlCache[uint64, *int64], *mapCache[int64, nonceRange]]{
			balances:        make(addressChainMap[*ttlCache[uint64, *big.Int]]),
			nonces:          make(addressChainMap[*ttlCache[uint64, *int64]]),
			nonceRangeCache: *newNonceRangeCache[*mapCache[int64, nonceRange]](),
		},
	}
}

func (c *cacheWithTTL) Clear() {
	c.rw.Lock()
	defer c.rw.Unlock()

	// TTL cache removes expired items automatically
	// but in case we want to clear it manually we can do it here
	for _, chainCache := range c.balances {
		for _, cache := range chainCache {
			cache.clear()
		}
	}

	for _, chainCache := range c.nonces {
		for _, cache := range chainCache {
			cache.clear()
		}
	}

	c.nonceRangeCache.clear()
}