Add on_initialize logs.

Analog-Labs · Oct 17, 2024 · 61559ec · 61559ec
1 parent d010149
commit 61559ec
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 2 deletions.
diff --git a/pallets/elections/src/lib.rs b/pallets/elections/src/lib.rs
@@ -157,13 +157,15 @@ pub mod pallet {
 	#[pallet::hooks]
 	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {
 		fn on_initialize(_: BlockNumberFor<T>) -> Weight {
+			log::info!("on_initialize begin");
 			let mut weight = Weight::default();
 			for (network, _, _) in Unassigned::<T>::iter() {
 				weight = weight
 					// 1 Read of Unassigned per Loop
 					.saturating_add(T::DbWeight::get().reads(1))
 					.saturating_add(Self::try_elect_shard(network));
 			}
+			log::info!("on_initialize end");
 			weight
 		}
 	}

diff --git a/pallets/members/src/lib.rs b/pallets/members/src/lib.rs
@@ -149,6 +149,7 @@ pub mod pallet {
 	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {
 		///  `on_initialize`: Handles periodic heartbeat checks and manages member online/offline statuses.
 		fn on_initialize(n: BlockNumberFor<T>) -> Weight {
+			log::info!("on_initialize begin");
 			let mut weight = Weight::default();
 			if n % T::HeartbeatTimeout::get() == BlockNumberFor::<T>::default() {
 				for (member, _) in MemberOnline::<T>::iter() {
@@ -168,6 +169,7 @@ pub mod pallet {
 					T::DbWeight::get().writes(Heartbeat::<T>::drain().count() as u64),
 				);
 			}
+			log::info!("on_initialize end");
 			weight
 		}
 	}

diff --git a/pallets/shards/src/lib.rs b/pallets/shards/src/lib.rs
@@ -354,6 +354,7 @@ pub mod pallet {
 	#[pallet::hooks]
 	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {
 		fn on_initialize(n: BlockNumberFor<T>) -> Weight {
+			log::info!("on_initialize begin");
 			let mut writes = 0;
 			let mut reads = 0;
 			// use DkgTimeout instead
@@ -373,6 +374,7 @@ pub mod pallet {
 					reads += 1;
 				}
 			});
+			log::info!("on_initialize end");
 			T::DbWeight::get().reads_writes(reads, writes)
 		}
 	}

diff --git a/pallets/tasks/src/lib.rs b/pallets/tasks/src/lib.rs
@@ -290,7 +290,10 @@ pub mod pallet {
 	#[pallet::hooks]
 	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {
 		fn on_initialize(_: BlockNumberFor<T>) -> Weight {
-			Self::prepare_batches().saturating_add(Self::schedule_tasks())
+			log::info!("on_initialize begin");
+			let weight = Self::prepare_batches().saturating_add(Self::schedule_tasks());
+			log::info!("on_initialize end");
+			weight
 		}
 	}
 

diff --git a/pallets/tasks/src/mock.rs b/pallets/tasks/src/mock.rs
@@ -323,7 +323,6 @@ pub fn roll(n: u64) {
 
 fn next_block() {
 	let mut now = System::block_number();
-	Tasks::on_finalize(now);
 	now += 1;
 	System::set_block_number(now);
 	Shards::on_initialize(now);