Skip to content

Commit e009ff1

Browse files
committed
track weight for clear/removal
1 parent f25bdda commit e009ff1

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pallets/subtensor/src/migration.rs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,11 @@ pub fn do_migrate_fix_total_coldkey_stake<T: Config>() -> Weight {
3838
let mut weight = T::DbWeight::get().reads(1);
3939

4040
// Clear everything from the map first
41-
let _ = TotalColdkeyStake::<T>::clear(u32::MAX, None);
41+
let removal_results = TotalColdkeyStake::<T>::clear(u32::MAX, None);
42+
// 1 read/write per removal
43+
let entries_removed: u64 = removal_results.backend.into();
44+
weight =
45+
weight.saturating_add(T::DbWeight::get().reads_writes(entries_removed, entries_removed));
4246

4347
// Iterate through all staking hotkeys.
4448
for (coldkey, hotkey_vec) in StakingHotkeys::<T>::iter() {

0 commit comments

Comments
 (0)