Merge pull request opentensor#691 from opentensor/fix-identity-migration

fix identity migration
backend-developers-ltd · Jul 24, 2024 · 6dca681 · 6dca681
2 parents f991965 + 3233133
commit 6dca681
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/pallets/subtensor/src/macros/hooks.rs b/pallets/subtensor/src/macros/hooks.rs
@@ -68,7 +68,9 @@ mod hooks {
                 .saturating_add(migrations::migrate_populate_staking_hotkeys::migrate_populate_staking_hotkeys::<T>())
                 // Fix total coldkey stake.
                 // Storage version v8 -> v9
-                .saturating_add(migrations::migrate_fix_total_coldkey_stake::migrate_fix_total_coldkey_stake::<T>());
+                .saturating_add(migrations::migrate_fix_total_coldkey_stake::migrate_fix_total_coldkey_stake::<T>())
+                // Migrate Delegate Ids on chain
+                .saturating_add(migrations::migrate_chain_identity::migrate_set_hotkey_identities::<T>());
             weight
         }
 

diff --git a/pallets/subtensor/src/migrations/migrate_chain_identity.rs b/pallets/subtensor/src/migrations/migrate_chain_identity.rs
@@ -31,7 +31,7 @@ fn string_to_bounded_vec(input: &str) -> Result<BoundedVec<u8, ConstU32<1024>>,
 }
 
 pub fn migrate_set_hotkey_identities<T: Config>() -> Weight {
-    let migration_name = b"fix_total_coldkey_stake_v7".to_vec();
+    let migration_name = b"migrate_identities".to_vec();
 
     // Initialize the weight with one read operation.
     let mut weight = T::DbWeight::get().reads(1);
@@ -140,17 +140,22 @@ pub fn migrate_set_hotkey_identities<T: Config>() -> Weight {
                 && identity.description.len() <= 1024
                 && identity.additional.len() <= 1024;
             if !is_valid {
+                log::info!("Bytes not correct");
                 continue;
             }
 
             // Get the owning coldkey.
             let coldkey = Owner::<T>::get(decoded_hotkey.clone());
+            log::info!("ColdKey: {:?}", decoded_hotkey);
+
             weight = weight.saturating_add(T::DbWeight::get().reads(1));
 
             // Sink into the map.
             Identities::<T>::insert(coldkey.clone(), identity.clone());
             weight = weight.saturating_add(T::DbWeight::get().writes(1));
         }
+    } else {
+        log::info!("Failed to decode JSON");
     }
     // Mark the migration as completed
     HasMigrationRun::<T>::insert(&migration_name, true);