open-im-server/pkg/common/localcache/local/cache.go

package local

import (
	"hash/fnv"
	"time"
	"unsafe"
)

type Cache[V any] interface {
	Get(key string, fetch func() (V, error)) (V, error)
	Del(key string) bool
}

func NewCache[V any](slotNum, slotSize int, successTTL, failedTTL time.Duration, target Target, onEvict EvictCallback[string, V]) Cache[V] {
	c := &slot[V]{
		n:      uint64(slotNum),
		slots:  make([]*LRU[string, V], slotNum),
		target: target,
	}
	for i := 0; i < slotNum; i++ {
		c.slots[i] = NewLRU[string, V](slotSize, successTTL, failedTTL, c.target, onEvict)
	}
	return c
}

type slot[V any] struct {
	n      uint64
	slots  []*LRU[string, V]
	target Target
}

func (c *slot[V]) index(s string) uint64 {
	h := fnv.New64a()
	_, _ = h.Write(*(*[]byte)(unsafe.Pointer(&s)))
	return h.Sum64() % c.n
}

func (c *slot[V]) Get(key string, fetch func() (V, error)) (V, error) {
	return c.slots[c.index(key)].Get(key, fetch)
}

func (c *slot[V]) Del(key string) bool {
	if c.slots[c.index(key)].Del(key) {
		c.target.IncrDelHit()
		return true
	} else {
		c.target.IncrDelNotFound()
		return false
	}
}