From 1a00c512fdb31cd36ca9c46a21bd4c150e5f021f Mon Sep 17 00:00:00 2001
From: Mark Cilia Vincenti <markciliavincenti@gmail.com>
Date: Sat, 10 Aug 2024 12:50:46 +0200
Subject: [PATCH] More cleaning.

---
 .../DistributedLock/RefCounter.cs             | 59 -------------------
 .../Internal/InMemoryCaching.cs               | 14 +----
 2 files changed, 2 insertions(+), 71 deletions(-)
 delete mode 100644 src/EasyCaching.Core/DistributedLock/RefCounter.cs

diff --git a/src/EasyCaching.Core/DistributedLock/RefCounter.cs b/src/EasyCaching.Core/DistributedLock/RefCounter.cs
deleted file mode 100644
index d5f8c189..00000000
--- a/src/EasyCaching.Core/DistributedLock/RefCounter.cs
+++ /dev/null
@@ -1,59 +0,0 @@
-using System;
-using System.Collections.Generic;
-using System.Runtime.CompilerServices;
-using System.Threading;
-
-namespace EasyCaching.Core.DistributedLock
-{
-    internal class RefCounter<T>
-    {
-        private int _refCount = 1;
-
-        public RefCounter(T value) => Value = value;
-
-        public T Value { get; }
-
-        [MethodImpl(MethodImplOptions.AggressiveInlining)]
-        public int Increment() => Interlocked.Increment(ref _refCount);
-
-        [MethodImpl(MethodImplOptions.AggressiveInlining)]
-        public int Decrement() => Interlocked.Decrement(ref _refCount);
-    }
-
-    internal class RefCounterPool<TKey, TValue> where TValue : class
-    {
-        private readonly IDictionary<TKey, RefCounter<TValue>> _dictionary;
-
-        public RefCounterPool() => _dictionary = new Dictionary<TKey, RefCounter<TValue>>();
-
-        public TValue GetOrAdd(TKey key, Func<TKey, TValue> valueFactory)
-        {
-            if (valueFactory == null) throw new ArgumentNullException(nameof(valueFactory));
-
-            RefCounter<TValue> item;
-            lock (_dictionary)
-            {
-                if (!_dictionary.TryGetValue(key, out item))
-                    return (_dictionary[key] = new RefCounter<TValue>(valueFactory(key))).Value;
-            }
-
-            item.Increment();
-
-            return item.Value;
-        }
-
-        public TValue TryRemove(TKey key)
-        {
-            RefCounter<TValue> item;
-
-            lock (_dictionary)
-            {
-                if (!_dictionary.TryGetValue(key, out item) || item.Decrement() > 0) return null;
-
-                _dictionary.Remove(key);
-            }
-
-            return item.Value;
-        }
-    }
-}
diff --git a/src/EasyCaching.InMemory/Internal/InMemoryCaching.cs b/src/EasyCaching.InMemory/Internal/InMemoryCaching.cs
index 913ebb10..3c49ee09 100644
--- a/src/EasyCaching.InMemory/Internal/InMemoryCaching.cs
+++ b/src/EasyCaching.InMemory/Internal/InMemoryCaching.cs
@@ -305,11 +305,7 @@ private static bool FilterByPattern(string key, string searchKey, SearchKeyPatte
 
         public IDictionary<string, CacheValue<T>> GetAll<T>(IEnumerable<string> keys)
         {
-            var map = new Dictionary<string, CacheValue<T>>();
-            foreach (string key in keys)
-                map[key] = Get<T>(key);
-
-            return map;
+            return keys.ToDictionary(key => key, key => Get<T>(key));
         }
 
         public IDictionary<string, CacheValue<T>> GetAll<T>(string prefix = "")
@@ -323,13 +319,7 @@ public IDictionary<string, CacheValue<T>> GetAll<T>(string prefix = "")
 
         public int SetAll<T>(IDictionary<string, T> values, TimeSpan? expiresIn = null)
         {
-            if (values == null || values.Count == 0) return 0;
-
-            var list = new List<bool>();
-
-            foreach (var entry in values) list.Add(Set(entry.Key, entry.Value, expiresIn));
-
-            return list.Count(r => r);
+            return values?.Count(entry => Set(entry.Key, entry.Value, expiresIn)) ?? 0;
         }
 
         public bool Replace<T>(string key, T value, TimeSpan? expiresIn = null)