go/cache
1
0
Fork 0
Code Issues Pull requests Releases Activity Actions

fix: deadlock in tests

Browse source
This commit is contained in:
Bruno Carlin 2025-01-18 01:37:41 +01:00
parent 9d6d9b9dff
commit 35044ad38a
Signed by: bcarlin
GPG key ID: 8E254EA0FFEB9B6D
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cache.go
View file

@ -23,14 +23,13 @@ func (e entry[V]) isExpired() bool {
// Cache is an in-memory key-value store. // Cache is an in-memory key-value store.
type Cache[K comparable, V any] struct { type Cache[K comparable, V any] struct {
data map[K]entry[V] data map[K]entry[V]
mu *sync.RWMutex mu sync.RWMutex
} }
// New instantiate a new cache. // New instantiate a new cache.
func New[K comparable, V any]() *Cache[K, V] { func New[K comparable, V any]() *Cache[K, V] {
return &Cache[K, V]{ return &Cache[K, V]{
data: make(map[K]entry[V]), data: make(map[K]entry[V]),
mu: &sync.RWMutex{},
} }
} }
@ -47,9 +46,6 @@ func (c *Cache[K, V]) Put(key K, val V) {
// //
// A 0 ttl value disables the expiration of the value. // A 0 ttl value disables the expiration of the value.
func (c *Cache[K, V]) PutTTL(key K, val V, ttl time.Duration) { func (c *Cache[K, V]) PutTTL(key K, val V, ttl time.Duration) {
c.mu.Lock()
defer c.mu.Unlock()
if ttl == 0 { if ttl == 0 {
c.Put(key, val) c.Put(key, val)
@ -58,6 +54,9 @@ func (c *Cache[K, V]) PutTTL(key K, val V, ttl time.Duration) {
exp := time.Now().Add(ttl) exp := time.Now().Add(ttl)
c.mu.Lock()
defer c.mu.Unlock()
c.data[key] = entry[V]{&exp, val} c.data[key] = entry[V]{&exp, val}
} }