From 22e3c3cd44439d45b42bb65b4e49a016703029ec Mon Sep 17 00:00:00 2001
From: Gianfranco Tasteri <gianfra33@gmail.com>
Date: Thu, 11 Jan 2024 12:56:30 -0300
Subject: [PATCH] implement type that set the proper storage version for pallet
 vesting and transaction payment

---
 frame/transaction-payment/src/lib.rs        |  2 ++
 frame/transaction-payment/src/migrations.rs | 22 ++++++++++++
 frame/vesting/src/migrations.rs             | 37 ++++++---------------
 3 files changed, 35 insertions(+), 26 deletions(-)
 create mode 100644 frame/transaction-payment/src/migrations.rs

diff --git a/frame/transaction-payment/src/lib.rs b/frame/transaction-payment/src/lib.rs
index 83ff69428d..d7c960f77d 100644
--- a/frame/transaction-payment/src/lib.rs
+++ b/frame/transaction-payment/src/lib.rs
@@ -78,6 +78,8 @@ mod tests;
 mod payment;
 mod types;
 
+pub mod migrations;
+
 pub use pallet::*;
 pub use payment::*;
 pub use types::{FeeDetails, InclusionFee, RuntimeDispatchInfo};
diff --git a/frame/transaction-payment/src/migrations.rs b/frame/transaction-payment/src/migrations.rs
new file mode 100644
index 0000000000..065604b2fb
--- /dev/null
+++ b/frame/transaction-payment/src/migrations.rs
@@ -0,0 +1,22 @@
+
+
+
+// temporary code to set the migration to V1
+pub mod v1 {
+	use super::*;
+
+	/// Manually setting the version to V1
+	pub struct ForceSetVersionToV2<T>(sp_std::marker::PhantomData<T>);
+	impl<T: Config> OnRuntimeUpgrade for ForceSetVersionToV2<T> {
+		fn on_runtime_upgrade() -> Weight {
+
+			if StorageVersion::<T>::get() == Releases::V1 {
+				StorageVersion::<T>::put(Releases::V1Ancient);
+			}
+			
+			T::DbWeight::get().reads_writes(1, 1)
+		}
+	
+	}
+
+}
\ No newline at end of file
diff --git a/frame/vesting/src/migrations.rs b/frame/vesting/src/migrations.rs
index 69bbc97296..ba627a5636 100644
--- a/frame/vesting/src/migrations.rs
+++ b/frame/vesting/src/migrations.rs
@@ -35,33 +35,18 @@ pub mod v1 {
 		Ok(())
 	}
 
-	/// Migrate from single schedule to multi schedule storage.
-	/// WARNING: This migration will delete schedules if `MaxVestingSchedules < 1`.
-	pub fn migrate<T: Config>() -> Weight {
-		let mut reads_writes = 0;
+	/// Manually setting the version to V1
+	pub struct ForceSetVersionToV1<T>(sp_std::marker::PhantomData<T>);
+	impl<T: Config> OnRuntimeUpgrade for ForceSetVersionToV1<T> {
+		fn on_runtime_upgrade() -> Weight {
 
-		Vesting::<T>::translate::<VestingInfo<BalanceOf<T>, T::BlockNumber>, _>(
-			|_key, vesting_info| {
-				reads_writes += 1;
-				let v: Option<
-					BoundedVec<
-						VestingInfo<BalanceOf<T>, T::BlockNumber>,
-						MaxVestingSchedulesGet<T>,
-					>,
-				> = vec![vesting_info].try_into().ok();
-
-				if v.is_none() {
-					log::warn!(
-						target: "runtime::vesting",
-						"migration: Failed to move a vesting schedule into a BoundedVec"
-					);
-				}
-
-				v
-			},
-		);
-
-		T::DbWeight::get().reads_writes(reads_writes, reads_writes)
+			if StorageVersion::<T>::get() == Releases::V0 {
+				StorageVersion::<T>::put(Releases::V1);
+			}
+			
+			T::DbWeight::get().reads_writes(1, 1)
+		}
+	
 	}
 
 	#[cfg(feature = "try-runtime")]