Add GetOrAdd method

InCerryGit · Apr 26, 2023 · d4e8fc9 · d4e8fc9
1 parent d45b49c
commit d4e8fc9
Show file tree

Hide file tree

Showing 6 changed files with 594 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -64,6 +64,10 @@ cache.Set(key, "my cache sync", TimeSpan.FromMinutes(5));
 var result = cache.Get<string>(key);
 Console.WriteLine(result);
 
+// get or add
+result = cache.GetOrAdd(key, () => "my cache sync", TimeSpan.FromMinutes(5));
+Console.WriteLine(result);
+
 // delete
 cache.Delete(key);
 
@@ -75,6 +79,10 @@ await cache.SetAsync(key, "my cache async");
 result = await cache.GetAsync<string>(key);
 Console.WriteLine(result);
 
+// get or add
+result = await cache.GetOrAddAsync(key, () => "my cache async");
+Console.WriteLine(result);
+
 // delete
 await cache.DeleteAsync(key);
 

diff --git a/src/FasterKv.Cache.Core/FasterKvCache.cs b/src/FasterKv.Cache.Core/FasterKvCache.cs
@@ -104,6 +104,35 @@ public FasterKvCache(string name,
         return result.output.Get<TValue>(_valueSerializer);
     }
 
+    public TValue GetOrAdd<TValue>(string key, Func<string, TValue> factory)
+    {
+        key.ArgumentNotNullOrEmpty();
+        factory.ArgumentNotNull();
+
+        var result = Get<TValue>(key);
+        if (result is not null)
+            return result;
+
+        result = factory(key);
+        Set(key, result);
+        return result;
+    }
+
+    public TValue GetOrAdd<TValue>(string key, Func<string, TValue> factory, TimeSpan expiryTime)
+    {
+        key.ArgumentNotNullOrEmpty();
+        factory.ArgumentNotNull();
+        expiryTime.ArgumentNotNegativeOrZero();
+
+        var result = Get<TValue>(key);
+        if (result is not null)
+            return result;
+
+        result = factory(key);
+        Set(key, result, expiryTime);
+        return result;
+    }
+
     public void Delete(string key)
     {
         using var scopeSession = GetSessionWrap();
@@ -147,7 +176,36 @@ public void Set<TValue>(string key, TValue value, TimeSpan expiryTime)
 
         return result.output.Get<TValue>(_valueSerializer);
     }
+
+    public async Task<TValue> GetOrAddAsync<TValue>(string key, Func<string, Task<TValue>> factory, CancellationToken token = default)
+    {
+        key.ArgumentNotNullOrEmpty();
+        factory.ArgumentNotNull();
+
+        var result = await GetAsync<TValue>(key, token);
+        if (result is not null)
+            return result;
+
+        result = await factory(key);
+        await SetAsync(key, result, token);
+        return result;
+    }
+
+    public async Task<TValue> GetOrAddAsync<TValue>(string key, Func<string, Task<TValue>> factory, TimeSpan expiryTime, CancellationToken token = default)
+    {
+        key.ArgumentNotNullOrEmpty();
+        factory.ArgumentNotNull();
+        expiryTime.ArgumentNotNegativeOrZero();
 
+        var result = await GetAsync<TValue>(key, token);
+        if (result is not null)
+            return result;
+
+        result = await factory(key);
+        await SetAsync(key, result, expiryTime, token);
+        return result;
+    }
+
     public async Task DeleteAsync(string key, CancellationToken token = default)
     {
         key.ArgumentNotNull();
@@ -171,7 +229,7 @@ public async Task SetAsync<TValue>(string key, TValue? value, TimeSpan expiryTim
         using var sessionWrap = GetSessionWrap();
         await SetInternalAsync(sessionWrap, key, value, token, expiryTime);
     }
-    
+
     private async Task SetInternalAsync<TValue>(ClientSessionWrap sessionWrap, string key, TValue? value,
         CancellationToken cancellationToken, TimeSpan? expiryTime = null)
     {

diff --git a/src/FasterKv.Cache.Core/FasterKvStore.TValue.cs b/src/FasterKv.Cache.Core/FasterKvStore.TValue.cs
@@ -97,6 +97,32 @@ public FasterKvCache(string name,
         return result.output.Data;
     }
 
+    public TValue GetOrAdd(string key, Func<string,TValue> factory)
+    {
+        key.ArgumentNotNullOrEmpty();
+
+        var value = Get(key);
+        if (value is not null)
+            return value;
+
+        value = factory(key);
+        Set(key, value);
+        return value;
+    }
+
+    public TValue GetOrAdd(string key, Func<string,TValue> factory, TimeSpan expiryTime)
+    {
+        key.ArgumentNotNullOrEmpty();
+
+        var value = Get(key);
+        if (value is not null)
+            return value;
+
+        value = factory(key);
+        Set(key, value, expiryTime);
+        return value;
+    }
+
     public void Delete(string key)
     {
         key.ArgumentNotNull();
@@ -141,31 +167,57 @@ public void Set(string key, TValue value, TimeSpan expiryTime)
 
         return result.output.Data;
     }
-
+
+    public async Task<TValue> GetOrAddAsync(string key, Func<string,Task<TValue>> factory, CancellationToken token = default)
+    {
+        key.ArgumentNotNullOrEmpty();
+
+        var value = await GetAsync(key, token);
+        if (value is not null)
+            return value;
+
+        value = await factory(key);
+        await SetAsync(key, value, token);
+        return value;
+    }
+
+    public async Task<TValue> GetOrAddAsync(string key, Func<string,Task<TValue>> factory, TimeSpan expiryTime, CancellationToken token = default)
+    {
+        key.ArgumentNotNullOrEmpty();
+
+        var value = await GetAsync(key, token);
+        if (value is not null)
+            return value;
+
+        value = await factory(key);
+        await SetAsync(key, value, expiryTime, token);
+        return value;
+    }
+
     public async Task DeleteAsync(string key, CancellationToken token = default)
     {
         key.ArgumentNotNull();
 
         using var scopeSession = GetSessionWrap();
         (await scopeSession.Session.DeleteAsync(ref key, token: token).ConfigureAwait(false)).Complete();
     }
-    
+
     public async Task SetAsync(string key, TValue? value, CancellationToken token = default)
     {
         key.ArgumentNotNullOrEmpty();
 
         using var sessionWrap = GetSessionWrap();
         await SetInternalAsync(sessionWrap, key, value, token);
     }
-    
+
     public async Task SetAsync(string key, TValue? value, TimeSpan expiryTime, CancellationToken token = default)
     {
         key.ArgumentNotNullOrEmpty();
 
         using var sessionWrap = GetSessionWrap();
         await SetInternalAsync(sessionWrap, key, value, token, expiryTime);
     }
-    
+
 
     private void SetInternal(ClientSessionWrap<TValue> sessionWrap, string key, TValue? value,
         TimeSpan? expiryTime = null)