save

Author: sudeepdino008

db/state/cache.go

@@ -19,8 +19,7 @@ type u128 struct{ hi, lo uint64 }      //nolint
 type u192 struct{ hi, lo, ext uint64 } //nolint
 
 type DomainGetFromFileCache struct {
-	sync.RWMutex
-	*freelru.LRU[uint64, domainGetFromFileCacheItem]
+	*freelru.ShardedLRU[uint64, domainGetFromFileCacheItem]
 	enabled, trace bool
 	limit          uint32
 }
@@ -33,37 +32,31 @@ type domainGetFromFileCacheItem struct {
 
 var (
 	domainGetFromFileCacheLimit   = uint32(dbg.EnvInt("D_LRU", 1_000_000))
-	domainGetFromFileCacheTrace   = dbg.EnvBool("D_LRU_TRACE", false)
+	domainGetFromFileCacheTrace   = dbg.EnvBool("D_LRU_TRACE", true)
 	domainGetFromFileCacheEnabled = dbg.EnvBool("D_LRU_ENABLED", true)
 )
 
 func NewDomainGetFromFileCache(limit uint32) *DomainGetFromFileCache {
-	c, err := freelru.New[uint64, domainGetFromFileCacheItem](limit, u64noHash)
+	c, err := freelru.NewSharded[uint64, domainGetFromFileCacheItem](limit, u64noHash)
 	if err != nil {
 		panic(err)
 	}
-	return &DomainGetFromFileCache{LRU: c, enabled: domainGetFromFileCacheEnabled, trace: domainGetFromFileCacheTrace, limit: limit}
+	return &DomainGetFromFileCache{ShardedLRU: c, enabled: domainGetFromFileCacheEnabled, trace: domainGetFromFileCacheTrace, limit: limit}
 }
 
 func (c *DomainGetFromFileCache) Add(key uint64, value domainGetFromFileCacheItem) (evicted bool) {
-	c.Lock()
-	defer c.Unlock()
-	return c.LRU.Add(key, value)
+	return c.ShardedLRU.Add(key, value)
 }
 
 func (c *DomainGetFromFileCache) Get(key uint64) (value domainGetFromFileCacheItem, ok bool) {
-	c.Lock() // get upates cache vars
-	defer c.Unlock()
-	return c.LRU.Get(key)
+	return c.ShardedLRU.Get(key)
 }
 
 func (c *DomainGetFromFileCache) SetTrace(v bool) { c.trace = v }
 func (c *DomainGetFromFileCache) LogStats(dt kv.Domain) {
 	if c == nil {
 		return
 	}
-	c.RLock()
-	defer c.RUnlock()
 	if !c.enabled || !c.trace {
 		return
 	}