fix(migration): release storage locks

Closes #4344. Hat tip to @erwanor for spotting the omission.
penumbra-zone · May 7, 2024 · 0e4bb1a · 0e4bb1a
1 parent 15f347c
commit 0e4bb1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/crates/bin/pd/src/migrate.rs b/crates/bin/pd/src/migrate.rs
@@ -158,6 +158,7 @@ impl Migration {
                 storage.release().await;
                 let storage = Storage::load(rocksdb_dir, SUBSTORE_PREFIXES.to_vec()).await?;
                 let migrated_state = storage.latest_snapshot();
+                storage.release().await;
 
                 // The migration is complete, now we need to generate a genesis file. To do this, we need
                 // to lookup a validator view from the chain, and specify the post-upgrade app hash and
@@ -254,6 +255,7 @@ pub async fn last_block_timestamp(home: PathBuf) -> anyhow::Result<tendermint::T
         .get_block_timestamp()
         .await
         .context("error reading latest block timestamp")?;
+    storage.release().await;
     Ok(last_block_time)
 }
 

diff --git a/crates/bin/pd/src/migrate/testnet74.rs b/crates/bin/pd/src/migrate/testnet74.rs
@@ -129,6 +129,7 @@ pub async fn migrate(
     storage.release().await;
     let storage = Storage::load(rocksdb_dir, SUBSTORE_PREFIXES.to_vec()).await?;
     let migrated_state = storage.latest_snapshot();
+    storage.release().await;
 
     // The migration is complete, now we need to generate a genesis file. To do this, we need
     // to lookup a validator view from the chain, and specify the post-upgrade app hash and