Correct txn get

serai-dex · Nov 30, 2023 · 3f04bbd · 3f04bbd
1 parent 566b1b4
commit 3f04bbd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/common/db/src/parity_db.rs b/common/db/src/parity_db.rs
@@ -8,7 +8,13 @@ pub struct Transaction<'a>(&'a Arc<ParityDb>, Vec<(u8, Vec<u8>, Option<Vec<u8>>)
 
 impl Get for Transaction<'_> {
   fn get(&self, key: impl AsRef<[u8]>) -> Option<Vec<u8>> {
-    self.0.get(key)
+    let mut res = self.0.get(&key);
+    for change in &self.1 {
+      if change.1 == key.as_ref() {
+        res = change.2.clone();
+      }
+    }
+    res
   }
 }
 impl DbTxn for Transaction<'_> {