From bd4590d7cfaf2fa554ff6c37e095975ac4eb0f02 Mon Sep 17 00:00:00 2001
From: vnugent <public@vaughnnugent.com>
Date: Sat, 5 Aug 2023 17:50:16 -0400
Subject: Public prep, and api additions

---
 .../src/RemoteBackedMemoryCache.cs                 | 49 ++++++++++++++++++++--
 1 file changed, 45 insertions(+), 4 deletions(-)

(limited to 'lib/VNLib.Plugins.Extensions.VNCache/src/RemoteBackedMemoryCache.cs')

diff --git a/lib/VNLib.Plugins.Extensions.VNCache/src/RemoteBackedMemoryCache.cs b/lib/VNLib.Plugins.Extensions.VNCache/src/RemoteBackedMemoryCache.cs
index ffe9108..ff73f19 100644
--- a/lib/VNLib.Plugins.Extensions.VNCache/src/RemoteBackedMemoryCache.cs
+++ b/lib/VNLib.Plugins.Extensions.VNCache/src/RemoteBackedMemoryCache.cs
@@ -192,20 +192,61 @@ namespace VNLib.Plugins.Extensions.VNCache
         {
             CheckConnected();
 
-            DateTime currentTime = DateTime.UtcNow;
-
             //Alloc serialzation buffer
             using AddOrUpdateBuffer buffer = new (Client.Config.BufferHeap);
 
             //Serialze the value
             serialzer.Serialize(value, buffer);
 
+            //Call update on raw data
+            await AddOrUpdateAsync(key, newKey, buffer, cancellation);
+        }
+
+        ///<inheritdoc/>
+        public override async Task GetAsync(string key, IObjectData rawData, CancellationToken cancellation)
+        {
+            CheckConnected();
+         
+            IBlobCacheBucket bucket = _memCache.GetBucket(key);
+
+            //Obtain cache handle
+            using (CacheBucketHandle handle = await bucket.WaitAsync(cancellation))
+            {
+                //Try to read the value
+                if (handle.Cache.TryGetValue(key, out CacheEntry entry))
+                {
+                    rawData.SetData(entry.GetDataSegment());
+                    return;
+                }
+            }
+
+            //Get the object from the server
+            await Client.GetObjectAsync(key, rawData, cancellation);
+
+            //See if object data was set
+            if (rawData.GetData().IsEmpty)
+            {
+                return;
+            }
+
+            //Update local cache
+            await _memCache.AddOrUpdateObjectAsync(key, null, static b => b.GetData(), rawData, DateTime.UtcNow, CancellationToken.None);
+        }
+
+        ///<inheritdoc/>
+        public override async Task AddOrUpdateAsync(string key, string? newKey, IObjectData rawData, ICacheObjectSerialzer serialzer, CancellationToken cancellation)
+        {
+            CheckConnected();
+
+            DateTime currentTime = DateTime.UtcNow;
+
             try
             {
                 //Update remote first, and if exceptions are raised, do not update local cache
-                await Client.AddOrUpdateObjectAsync(key, newKey, (IObjectData)buffer, cancellation);
+                await Client.AddOrUpdateObjectAsync(key, newKey, rawData, cancellation);
 
-                await _memCache.AddOrUpdateObjectAsync(key, newKey, static b => b.GetData(), buffer, currentTime, CancellationToken.None);
+                //Safe to update local cache
+                await _memCache.AddOrUpdateObjectAsync(key, newKey, static b => b.GetData(), rawData, currentTime, CancellationToken.None);
             }
             catch
             {
-- 
cgit