datastores.go

package datastores

import "sync"

// DataStore ###########################
// Memory storage structures
//###########################
// Generic data store
type DataStore[K comparable, V any] struct {
	sync.RWMutex
	cache map[K]V
}

func NewDataStore[K comparable, V any]() *DataStore[K, V] {
	return &DataStore[K, V]{
		cache: make(map[K]V),
	}
}

func (ds *DataStore[K, V]) Set(key K, value V) {
	ds.Lock()
	defer ds.Unlock()

	ds.cache[key] = value
}

func (ds *DataStore[K, V]) Get(key K) V {
	ds.RLock()
	defer ds.RUnlock()

	return ds.cache[key]
}

func (ds *DataStore[K, V]) SetAll(data map[K]V) {
	ds.Lock()
	defer ds.Unlock()

	ds.cache = data
}
func (ds *DataStore[K, V]) GetAll() map[K]V {
	ds.RLock()
	defer ds.RUnlock()

	return ds.cache
}

func (ds *DataStore[K, V]) Unset(key K) bool {
	ds.Lock()
	defer ds.Unlock()

	_, ok := ds.cache[key]
	if ok {
		delete(ds.cache, key)
		return true
	}
	return false
}