Fix runtime migration (#453)

bifrost-io · Dec 23, 2021 · 755fe00 · 755fe00
1 parent 9290d4f
commit 755fe00
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/runtime/bifrost/src/lib.rs b/runtime/bifrost/src/lib.rs
@@ -1979,14 +1979,22 @@ impl OnRuntimeUpgrade for CustomOnRuntimeUpgrade {
 		Ok(())
 	}
 
-	#[cfg(feature = "try-runtime")]
 	fn on_runtime_upgrade() -> Weight {
 		log::info!("Bifrost `on_runtime_upgrade`...");
-		log::info!("Bifrost `on_runtime_upgrade finished`");
 		Salp::set_multisig_account(ConfirmMuitiSigAccount::get());
 		SalpLite::set_multisig_account(PolkaConfirmAsMultiSig::get());
+		log::info!("Bifrost `on_runtime_upgrade finished`");
 		RocksDbWeight::get().writes(1)
 	}
+
+	#[cfg(feature = "try-runtime")]
+	fn post_upgrade() -> Result<(), &'static str> {
+		let mut account = Salp::multisig_confirm_account();
+		log::info!("after salp migration multisig account as {}", account);
+		account = SalpLite::multisig_confirm_account();
+		log::info!("after salp-lite migration multisig account as {}", account);
+		Ok(())
+	}
 }
 
 struct CheckInherents;