Refactor AddOrUpdate method to accept separate factory functions for adding and updating items in cache

Author: alex-jitbit

FastCache/FastCache.cs

@@ -125,13 +125,14 @@ public void AddOrUpdate(TKey key, TValue value, TimeSpan ttl)
 		/// Factory pattern overload. Adds an item to cache if it does not exist, updates the existing item otherwise. Updating an item resets its TTL, essentially "sliding expiration".
 		/// </summary>
 		/// <param name="key">The key to add or update</param>
-		/// <param name="valueFactory">The factory function used to generate the item for the key</param>
+		/// <param name="addValueFactory">The factory function used to generate the item for the key</param>
+		/// <param name="updateValueFactory">The factory function used to update the item for the key</param>
 		/// <param name="ttl">TTL of the item</param>
-		public void AddOrUpdate(TKey key, Func<TValue> valueFactory, TimeSpan ttl)
+		public void AddOrUpdate(TKey key, Func<TKey, TValue> addValueFactory, Func<TKey, TValue, TValue> updateValueFactory, TimeSpan ttl)
 		{
 			_dict.AddOrUpdate(key,
-				addValueFactory: _ => new TtlValue(valueFactory(), ttl),
-				updateValueFactory: (_, _) => new TtlValue(valueFactory(), ttl));
+				addValueFactory: k => new TtlValue(addValueFactory(k), ttl),
+				updateValueFactory: (k, v) => new TtlValue(updateValueFactory(k, v.Value), ttl));
 		}
 
 		/// <summary>

UnitTests/UnitTests2.cs

@@ -234,7 +234,7 @@ public void AddOrUpdate_WithFactory()
             int callCount = 0;
 
             // Act
-            cache.AddOrUpdate("key1", () => { callCount++; return 42; }, TimeSpan.FromMinutes(1));
+            cache.AddOrUpdate("key1", _ => { callCount++; return 42; }, (_, _) => { callCount++; return 43; }, TimeSpan.FromMinutes(1));
             bool exists = cache.TryGet("key1", out int value);
 
             // Assert
@@ -243,10 +243,10 @@ public void AddOrUpdate_WithFactory()
             Assert.AreEqual(1, callCount); // Factory should be called exactly once
 
             callCount = 0;
-            cache.AddOrUpdate("key1", () => { callCount++; return 43; }, TimeSpan.FromMinutes(1));
+            cache.AddOrUpdate("key1", _ => { callCount++; return 44; }, (_, _) => { callCount++; return 45; }, TimeSpan.FromMinutes(1));
             exists = cache.TryGet("key1", out value);
             Assert.IsTrue(exists);
-            Assert.AreEqual(43, value);
+            Assert.AreEqual(45, value);
             Assert.AreEqual(1, callCount); // Factory should be called exactly once
         }
     }