sharded.go

package loadingcache

import (
	"context"
	"fmt"

	"github.com/devzero-inc/loadingcache/stats"
	"github.com/pkg/errors"
)

type shardedCache[Key comparable, Value any] struct {
	cacheOptions[Key, Value]
	shards []Cache[Key, Value]
}

func (s *shardedCache[K, V]) Get(ctx context.Context, key K) (V, error) {
	val, err := s.shards[s.hashCodeFunc(key)%uint64(len(s.shards))].Get(ctx, key)
	return val, errors.Wrap(err, "")
}

func (s *shardedCache[K, V]) Put(key K, value V) {
	s.shards[s.hashCodeFunc(key)%uint64(len(s.shards))].Put(key, value)
}

func (s *shardedCache[K, V]) Invalidate(keys ...K) {
	for _, k := range keys {
		s.shards[s.hashCodeFunc(k)%uint64(len(s.shards))].Invalidate(k)
	}
}

func (s *shardedCache[K, V]) InvalidateAll() {
	for _, shard := range s.shards {
		shard.InvalidateAll()
	}
}

func (s *shardedCache[K, V]) Close() {
	for _, shard := range s.shards {
		shard.Close()
	}
}

func (s *shardedCache[K, V]) Stats() Stats {
	statsSum := &stats.InternalStats{}
	for _, shard := range s.shards {
		switch typedCache := shard.(type) {
		case *genericCache[K, V]:
			statsSum = statsSum.Add(typedCache.stats)
		default:
			panic(fmt.Sprintf("unsupported cache type %T", shard))
		}
	}
	return statsSum
}