package data

import (
	"errors"
	"sync"
)

var _cacheLock = sync.Mutex{}

// GetCachedValue will return a value for key from the cache.
func (ds *Datastore) GetCachedValue(key string) ([]byte, error) {
	_cacheLock.Lock()
	defer _cacheLock.Unlock()

	// Check for a cached value
	if val, ok := ds.cache[key]; ok {
		return val, nil
	}

	return nil, errors.New(key + " not found in cache")
}

// SetCachedValue will set a value for key in the cache.
func (ds *Datastore) SetCachedValue(key string, b []byte) {
	_cacheLock.Lock()
	defer _cacheLock.Unlock()

	ds.cache[key] = b
}