git.delta.rocks / unique-network / refs/commits / 26734e956758

difftreelog

refactor switch to new prefix removal methods

Yaroslav Bolyukin2022-07-21parent: #cdfb9bd.patch.diff
in: master
New methods allows to call `remove_prefix` with limit multiple times
in the same block
However, we don't use prefix removal limits, so upgrade is
straightforward

5 files changed

modifiedpallets/common/src/lib.rsdiffbeforeafterboth
--- a/pallets/common/src/lib.rs
+++ b/pallets/common/src/lib.rs
@@ -958,8 +958,8 @@
 		<DestroyedCollectionCount<T>>::put(destroyed_collections);
 		<CollectionById<T>>::remove(collection.id);
 		<AdminAmount<T>>::remove(collection.id);
-		<IsAdmin<T>>::remove_prefix((collection.id,), None);
-		<Allowlist<T>>::remove_prefix((collection.id,), None);
+		let _ = <IsAdmin<T>>::clear_prefix((collection.id,), u32::MAX, None);
+		let _ = <Allowlist<T>>::clear_prefix((collection.id,), u32::MAX, None);
 		<CollectionProperties<T>>::remove(collection.id);
 
 		<Pallet<T>>::deposit_event(Event::CollectionDestroyed(collection.id));
modifiedpallets/fungible/src/lib.rsdiffbeforeafterboth
--- a/pallets/fungible/src/lib.rs
+++ b/pallets/fungible/src/lib.rs
@@ -231,8 +231,8 @@
 		PalletCommon::destroy_collection(collection.0, sender)?;
 
 		<TotalSupply<T>>::remove(id);
-		<Balance<T>>::remove_prefix((id,), None);
-		<Allowance<T>>::remove_prefix((id,), None);
+		let _ = <Balance<T>>::clear_prefix((id,), u32::MAX, None);
+		let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);
 		Ok(())
 	}
modifiedpallets/nonfungible/src/lib.rsdiffbeforeafterboth
429429
430		PalletCommon::destroy_collection(collection.0, sender)?;430		PalletCommon::destroy_collection(collection.0, sender)?;
431431
432		<TokenData<T>>::remove_prefix((id,), None);432		let _ = <TokenData<T>>::clear_prefix((id,), u32::MAX, None);
433		<TokenChildren<T>>::remove_prefix((id,), None);433		let _ = <TokenChildren<T>>::clear_prefix((id,), u32::MAX, None);
434		<Owned<T>>::remove_prefix((id,), None);434		let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);
435		<TokensMinted<T>>::remove(id);435		<TokensMinted<T>>::remove(id);
436		<TokensBurnt<T>>::remove(id);436		<TokensBurnt<T>>::remove(id);
437		<Allowance<T>>::remove_prefix((id,), None);437		let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);
438		<AccountBalance<T>>::remove_prefix((id,), None);438		let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);
439		Ok(())439		Ok(())
440	}440	}
441441
487		<TokensBurnt<T>>::insert(collection.id, burnt);487		<TokensBurnt<T>>::insert(collection.id, burnt);
488		<TokenData<T>>::remove((collection.id, token));488		<TokenData<T>>::remove((collection.id, token));
489		<TokenProperties<T>>::remove((collection.id, token));489		<TokenProperties<T>>::remove((collection.id, token));
490		<TokenAuxProperties<T>>::remove_prefix((collection.id, token), None);490		let _ = <TokenAuxProperties<T>>::clear_prefix((collection.id, token), u32::MAX, None);
491		let old_spender = <Allowance<T>>::take((collection.id, token));491		let old_spender = <Allowance<T>>::take((collection.id, token));
492492
493		if let Some(old_spender) = old_spender {493		if let Some(old_spender) = old_spender {
modifiedpallets/refungible/src/lib.rsdiffbeforeafterboth
--- a/pallets/refungible/src/lib.rs
+++ b/pallets/refungible/src/lib.rs
@@ -398,11 +398,11 @@
 
 		<TokensMinted<T>>::remove(id);
 		<TokensBurnt<T>>::remove(id);
-		<TotalSupply<T>>::remove_prefix((id,), None);
-		<Balance<T>>::remove_prefix((id,), None);
-		<Allowance<T>>::remove_prefix((id,), None);
-		<Owned<T>>::remove_prefix((id,), None);
-		<AccountBalance<T>>::remove_prefix((id,), None);
+		let _ = <TotalSupply<T>>::clear_prefix((id,), u32::MAX, None);
+		let _ = <Balance<T>>::clear_prefix((id,), u32::MAX, None);
+		let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);
+		let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);
+		let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);
 		Ok(())
 	}
 
@@ -424,6 +424,7 @@
 		<TokensBurnt<T>>::insert(collection.id, burnt);
 		<TokenProperties<T>>::remove((collection.id, token_id));
 		<TotalSupply<T>>::remove((collection.id, token_id));
+<<<<<<< HEAD
 		<Balance<T>>::remove_prefix((collection.id, token_id), None);
 		<Allowance<T>>::remove_prefix((collection.id, token_id), None);
 
@@ -435,6 +436,11 @@
 			}
 			.to_log(collection_id_to_address(collection.id)),
 		);
+=======
+		let _ = <Balance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);
+		let _ = <Allowance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);
+		// TODO: ERC721 transfer event
+>>>>>>> 5d9665e0... refactor: switch to new prefix removal methods
 		Ok(())
 	}
modifiedpallets/unique/src/lib.rsdiffbeforeafterboth
--- a/pallets/unique/src/lib.rs
+++ b/pallets/unique/src/lib.rs
@@ -373,13 +373,16 @@
 
 			T::CollectionDispatch::destroy(sender, collection)?;
 
-			<NftTransferBasket<T>>::remove_prefix(collection_id, None);
-			<FungibleTransferBasket<T>>::remove_prefix(collection_id, None);
-			<ReFungibleTransferBasket<T>>::remove_prefix((collection_id,), None);
+			// TODO: basket cleanup should be moved elsewhere
+			// Maybe runtime dispatch.rs should perform it?
 
-			<NftApproveBasket<T>>::remove_prefix(collection_id, None);
-			<FungibleApproveBasket<T>>::remove_prefix(collection_id, None);
-			<RefungibleApproveBasket<T>>::remove_prefix((collection_id,), None);
+			let _ = <NftTransferBasket<T>>::clear_prefix(collection_id, u32::MAX, None);
+			let _ = <FungibleTransferBasket<T>>::clear_prefix(collection_id, u32::MAX, None);
+			let _ = <ReFungibleTransferBasket<T>>::clear_prefix((collection_id,), u32::MAX, None);
+
+			let _ = <NftApproveBasket<T>>::clear_prefix(collection_id, u32::MAX, None);
+			let _ = <FungibleApproveBasket<T>>::clear_prefix(collection_id, u32::MAX, None);
+			let _ = <RefungibleApproveBasket<T>>::clear_prefix((collection_id,), u32::MAX, None);
 
 			Ok(())
 		}