LruCache faster snapshot (#5397)

NethermindEth · Mar 8, 2023 · a3d2c44 · a3d2c44
1 parent e1bc1be
commit a3d2c44
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/Nethermind/Nethermind.Core/Caching/LruCache.cs b/src/Nethermind/Nethermind.Core/Caching/LruCache.cs
@@ -123,7 +123,17 @@ public bool Delete(TKey key)
         public bool Contains(TKey key) => _cacheMap.ContainsKey(key);
 
         [MethodImpl(MethodImplOptions.Synchronized)]
-        public IDictionary<TKey, TValue> Clone() => _cacheMap.ToDictionary(i => i.Key, i => i.Value.Value.Value);
+        public KeyValuePair<TKey, TValue>[] ToArray()
+        {
+            int i = 0;
+            KeyValuePair<TKey, TValue>[] array = new KeyValuePair<TKey, TValue>[_cacheMap.Count];
+            foreach (KeyValuePair<TKey, LinkedListNode<LruCacheItem>> kvp in _cacheMap)
+            {
+                array[i++] = new KeyValuePair<TKey, TValue>(kvp.Key, kvp.Value.Value.Value);
+            }
+
+            return array;
+        }
 
         private void Replace(TKey key, TValue value)
         {

diff --git a/src/Nethermind/Nethermind.Trie/CachingStore.cs b/src/Nethermind/Nethermind.Trie/CachingStore.cs
@@ -57,7 +57,7 @@ public byte[]? this[byte[] key]
 
         public void PersistCache(IKeyValueStore pruningContext)
         {
-            IDictionary<byte[], byte[]> clone = _cache.Clone();
+            KeyValuePair<byte[], byte[]>[] clone = _cache.ToArray();
             Task.Run(() =>
             {
                 foreach (KeyValuePair<byte[], byte[]> kvp in clone)