fix(core): only resize db if migration is required

tari-project · Oct 11, 2022 · e6def7f · e6def7f
1 parent 702658c
commit e6def7f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/base_layer/core/src/chain_storage/lmdb_db/lmdb_db.rs b/base_layer/core/src/chain_storage/lmdb_db/lmdb_db.rs
@@ -2618,7 +2618,7 @@ fn run_migrations(db: &LMDBDatabase) -> Result<(), ChainStorageError> {
     Ok(())
 }
 
-// TODO: remove
+// TODO: this is a temporary fix, remove
 mod tari_script_execution_stack_bug_migration {
     use serde::{Deserialize, Serialize};
     use tari_common_types::types::{ComSignature, PublicKey};
@@ -2632,6 +2632,15 @@ mod tari_script_execution_stack_bug_migration {
     };
 
     pub fn migrate(db: &LMDBDatabase) -> Result<(), ChainStorageError> {
+        {
+            let txn = db.write_transaction()?;
+            // Only perform migration if necessary
+            if fetch_metadata(&txn, &db.metadata_db)?.height_of_longest_chain() == 0 ||
+                lmdb_len(&txn, &db.inputs_db)? == 0
+            {
+                return Ok(());
+            }
+        }
         unsafe {
             LMDBStore::resize(&db.env, &LMDBConfig::new(0, 1024 * 1024 * 1024, 0))?;
         }