perf/go/cache/local/local.go

// Package local implements cache.Cache with an in-memory cache.
package local

import (
	lru "github.com/hashicorp/golang-lru"
	"go.skia.org/infra/go/skerr"
	"go.skia.org/infra/perf/go/cache"
)

// Cache implements the cache.Cache interface.
type Cache struct {
	cache *lru.Cache
}

// New returns a new in-memory cache of the given size.
func New(size int) (*Cache, error) {
	c, err := lru.New(size)
	if err != nil {
		return nil, skerr.Wrapf(err, "failed to create local cache of size: %d", size)
	}
	return &Cache{
		cache: c,
	}, nil
}

// Add implements the cache.Cache interface.
func (c *Cache) Add(key string, value interface{}) {
	_ = c.cache.Add(key, value)
}

// Get implements the cache.Cache interface.
func (c *Cache) Get(key string) (interface{}, bool) {
	return c.cache.Get(key)
}

// Exists implements the cache.Cache interface.
func (c *Cache) Exists(key string) bool {
	return c.cache.Contains(key)
}

// Confirm we implement the interface.
var _ cache.Cache = (*Cache)(nil)