refactor: remove DB access when generate Journal

godwokenrises · Nov 2, 2022 · 9b089e4 · 9b089e4
1 parent cac1e4e
commit 9b089e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crates/store/src/state/state_db.rs b/crates/store/src/state/state_db.rs
@@ -458,7 +458,7 @@ impl<S: CodeStore> CodeStore for StateDB<S> {
     fn insert_script(&mut self, script_hash: H256, script: packed::Script) {
         self.journal.push(JournalEntry::InsertScript {
             script_hash,
-            prev_exist: self.get_script(&script_hash).is_some(),
+            prev_exist: self.dirty_scripts.contains_key(&script_hash),
         });
         self.dirty_scripts.insert(script_hash, script);
     }
@@ -480,7 +480,7 @@ impl<S: CodeStore> CodeStore for StateDB<S> {
         }
         self.journal.push(JournalEntry::InsertData {
             data_hash,
-            prev_exist: self.get_data(&data_hash).is_some(),
+            prev_exist: self.dirty_data.contains_key(&data_hash),
         });
         self.dirty_data.insert(data_hash, code);
     }