Fix Bug

neo-project · Feb 10, 2024 · c74a04f · c74a04f
1 parent 561a8dc
commit c74a04f
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/Neo/Persistence/MemorySnapshot.cs b/src/Neo/Persistence/MemorySnapshot.cs
@@ -65,12 +65,22 @@ public void Put(byte[] key, byte[] value)
 
         public byte[] TryGet(byte[] key)
         {
-            immutableData.TryGetValue(key, out byte[] value);
+            if (writeBatch.TryGetValue(key, out byte[] value))
+            {
+                return value;
+            }
+
+            immutableData.TryGetValue(key, out value);
             return value?[..];
         }
 
         public bool Contains(byte[] key)
         {
+            if (writeBatch.TryGetValue(key, out byte[] value) && value != null)
+            {
+                return true;
+            }
+
             return immutableData.ContainsKey(key);
         }
     }