fixed critical sections in DeleteExpired

Author: Code-Hex

cache.go

@@ -201,12 +201,18 @@ func (c *Cache[K, V]) Get(key K) (value V, ok bool) {
 
 // DeleteExpired all expired items from the cache.
 func (c *Cache[K, V]) DeleteExpired() {
-	for _, key := range c.cache.Keys() {
+	c.mu.Lock()
+	keys := c.cache.Keys()
+	c.mu.Unlock()
+
+	for _, key := range keys {
+		c.mu.Lock()
 		// if is expired, delete it and return nil instead
 		item, ok := c.cache.Get(key)
 		if ok && item.Expired() {
 			c.cache.Delete(key)
 		}
+		c.mu.Unlock()
 	}
 }
 

cache_test.go

@@ -2,7 +2,6 @@ package cache_test
 
 import (
 	"math/rand"
-	"runtime"
 	"sync"
 	"testing"
 	"time"
@@ -121,6 +120,4 @@ func TestCallJanitor(t *testing.T) {
 	if len(keys) != 0 {
 		t.Errorf("want items is empty but got %d", len(keys))
 	}
-
-	runtime.GC()
 }