From 03bfd16939847b0b20b3d5c6941280a21f58e901 Mon Sep 17 00:00:00 2001
From: Joel Nygren <joel.nygren@norce.io>
Date: Tue, 28 Oct 2025 12:28:33 +0100
Subject: [PATCH] added async tag

---
 HzMemoryCache/HzCacheMemoryLocker.cs | 8 ++++----
 HzMemoryCache/HzMemoryCacheAsync.cs  | 9 ++++-----
 2 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/HzMemoryCache/HzCacheMemoryLocker.cs b/HzMemoryCache/HzCacheMemoryLocker.cs
index 7f2ee72..606b9ca 100644
--- a/HzMemoryCache/HzCacheMemoryLocker.cs
+++ b/HzMemoryCache/HzCacheMemoryLocker.cs
@@ -82,7 +82,7 @@ private SemaphoreSlim GetSemaphore(string cacheName, string cacheInstanceId, str
         public async ValueTask<object> AcquireLockAsync(string cacheName, string cacheInstanceId, string operationId, string key, TimeSpan timeout, ILogger? logger,
             CancellationToken token)
         {
-            using var activity = HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.AcquireLock, HzActivities.Area.HzCacheMemoryLocker, key: key);
+            using var activity = HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.AcquireLock, HzActivities.Area.HzCacheMemoryLocker, key: key, async: true);
             var semaphore = GetSemaphore(cacheName, cacheInstanceId, key, logger);
 
             if (logger?.IsEnabled(LogLevel.Trace) ?? false)
@@ -91,7 +91,8 @@ public async ValueTask<object> AcquireLockAsync(string cacheName, string cacheIn
                     cacheInstanceId, operationId, key);
             }
             var acquired = false;
-            using (var waitForSemaphore = HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.WaitForSemaphore, HzActivities.Area.HzCacheMemoryLocker, key: key)){
+            using (var waitForSemaphore = HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.WaitForSemaphore, HzActivities.Area.HzCacheMemoryLocker, key: key, async: true))
+            {
                 acquired = await semaphore.WaitAsync(timeout, token).ConfigureAwait(false);
             }
 
@@ -135,7 +136,6 @@ public async ValueTask<object> AcquireLockAsync(string cacheName, string cacheIn
                    HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.WaitForSemaphore,
                        HzActivities.Area.HzCacheMemoryLocker, key: key))
             {
-
                 acquired = semaphore.Wait(timeout, token);
             }
 
@@ -188,4 +188,4 @@ public void ReleaseLock(string cacheName, string cacheInstanceId, string operati
             }
         }
     }
-}
+}
\ No newline at end of file
diff --git a/HzMemoryCache/HzMemoryCacheAsync.cs b/HzMemoryCache/HzMemoryCacheAsync.cs
index 2633dc2..4b018c2 100644
--- a/HzMemoryCache/HzMemoryCacheAsync.cs
+++ b/HzMemoryCache/HzMemoryCacheAsync.cs
@@ -52,7 +52,7 @@ public Task SetAsync<T>(string key, T? value, TimeSpan ttl)
 
                 using (var executeActivity =
                        HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.ExecuteFactory,
-                           HzActivities.Area.HzMemoryCache,async:true, key: key))
+                           HzActivities.Area.HzMemoryCache, async: true, key: key))
                 {
                     value = await valueFactory(key).ConfigureAwait(false);
                 }
@@ -71,7 +71,6 @@ public Task SetAsync<T>(string key, T? value, TimeSpan ttl)
             return value;
         }
 
-
         public async Task<IList<T>> GetOrSetBatchAsync<T>(IList<string> keys, Func<IList<string>, Task<List<KeyValuePair<string, T>>>> valueFactory)
         {
             return await GetOrSetBatchAsync(keys, valueFactory, options.defaultTTL).ConfigureAwait(false);
@@ -86,7 +85,7 @@ public async Task<IList<T>> GetOrSetBatchAsync<T>(IList<string> keys, Func<IList
             Dictionary<string, T> factoryRetrievedItems;
             using (var executeActivity =
                    HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.ExecuteFactory,
-                       HzActivities.Area.HzMemoryCache, key: string.Join(",", missingKeys ?? new List<string>())))
+                       HzActivities.Area.HzMemoryCache, key: string.Join(",", missingKeys ?? new List<string>()), async: true))
             {
                 factoryRetrievedItems =
                     (await valueFactory(missingKeys).ConfigureAwait(false)).ToDictionary(kv => kv.Key, kv => kv.Value);
@@ -129,7 +128,7 @@ public async Task<bool> RemoveAsync(string key)
 
         public async Task RemoveByPatternAsync(string pattern, bool sendNotification = true)
         {
-            using var activity = HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.RemoveByPattern, HzActivities.Area.HzMemoryCache, async: true, pattern: pattern,sendNotification:sendNotification);
+            using var activity = HzActivities.Source.StartActivityWithCommonTags(HzActivities.Names.RemoveByPattern, HzActivities.Area.HzMemoryCache, async: true, pattern: pattern, sendNotification: sendNotification);
             var myPattern = pattern;
             if (pattern[0] != '*')
             {
@@ -202,4 +201,4 @@ private async Task<bool> RemoveItemAsync(string key, CacheItemChangeType changeT
             return result;
         }
     }
-}
+}
\ No newline at end of file