fixed Set method in simple package

Author: Code-Hex

simple/example_test.go

@@ -23,11 +23,16 @@ func ExampleCache() {
 	if !aok2 {
 		fmt.Println("key 'a' has been deleted")
 	}
+	// update
+	c.Set("b", 3)
+	newbv, _ := c.Get("b")
+	fmt.Println(newbv)
 	// Output:
 	// 1 true
 	// 2 true
 	// 0 false
 	// key 'a' has been deleted
+	// 3
 }
 
 func ExampleCacheKeys() {

simple/simple.go

@@ -26,8 +26,13 @@ func NewCache[K comparable, V any]() *Cache[K, V] {
 // The default item never expires.
 func (c *Cache[K, V]) Set(k K, v V, opts ...cache.ItemOption) {
 	c.mu.Lock()
+	defer c.mu.Unlock()
+	if item, ok := c.items[k]; ok {
+		item.Value = v
+		item.Referenced()
+		return
+	}
 	c.items[k] = cache.NewItem(k, v, opts...)
-	c.mu.Unlock()
 }
 
 // Get gets an item from the cache.