git.delta.rocks / unique-network / refs/commits / e48f2d1c76b5

1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617use frame_benchmarking::{account, benchmarks};18use pallet_common::{19	bench_init,20	benchmarking::{21		create_collection_raw, property_key, property_value, load_is_admin_and_property_permissions,22	},23	CommonCollectionOperations,24};25use sp_std::prelude::*;26use up_data_structs::{27	budget::Unlimited, CollectionMode, PropertyPermission, MAX_ITEMS_PER_BATCH,28	MAX_PROPERTIES_PER_ITEM,29};3031use super::*;32use crate::{Config, NonfungibleHandle, Pallet};3334const SEED: u32 = 1;3536fn create_max_item_data<T: Config>(owner: T::CrossAccountId) -> CreateItemData<T> {37	CreateItemData::<T> {38		owner,39		properties: Default::default(),40	}41}42fn create_max_item<T: Config>(43	collection: &NonfungibleHandle<T>,44	sender: &T::CrossAccountId,45	owner: T::CrossAccountId,46) -> Result<TokenId, DispatchError> {47	<Pallet<T>>::create_item(48		collection,49		sender,50		create_max_item_data::<T>(owner),51		&Unlimited,52	)?;53	Ok(TokenId(<TokensMinted<T>>::get(collection.id)))54}5556fn create_collection<T: Config>(57	owner: T::CrossAccountId,58) -> Result<NonfungibleHandle<T>, DispatchError> {59	create_collection_raw(60		owner,61		CollectionMode::NFT,62		|owner: T::CrossAccountId, data| <Pallet<T>>::init_collection(owner.clone(), owner, data),63		NonfungibleHandle::cast,64	)65}6667benchmarks! {68	create_item {69		bench_init!{70			owner: sub; collection: collection(owner);71			sender: cross_from_sub(owner); to: cross_sub;72		};73	}: {create_max_item(&collection, &sender, to.clone())?}7475	create_multiple_items {76		let b in 0..MAX_ITEMS_PER_BATCH;77		bench_init!{78			owner: sub; collection: collection(owner);79			sender: cross_from_sub(owner); to: cross_sub;80		};81		let data = (0..b).map(|_| create_max_item_data::<T>(to.clone())).collect();82	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}8384	create_multiple_items_ex {85		let b in 0..MAX_ITEMS_PER_BATCH;86		bench_init!{87			owner: sub; collection: collection(owner);88			sender: cross_from_sub(owner);89		};90		let data = (0..b).map(|i| {91			bench_init!(to: cross_sub(i););92			create_max_item_data::<T>(to)93		}).collect();94	}: {<Pallet<T>>::create_multiple_items(&collection, &sender, data, &Unlimited)?}9596	burn_item {97		bench_init!{98			owner: sub; collection: collection(owner);99			sender: cross_from_sub(owner); burner: cross_sub;100		};101		let item = create_max_item(&collection, &sender, burner.clone())?;102	}: {<Pallet<T>>::burn(&collection, &burner, item)?}103104	burn_recursively_self_raw {105		bench_init!{106			owner: sub; collection: collection(owner);107			sender: cross_from_sub(owner); burner: cross_sub;108		};109		let item = create_max_item(&collection, &sender, burner.clone())?;110	}: {<Pallet<T>>::burn_recursively(&collection, &burner, item, &Unlimited, &Unlimited)?}111112	burn_recursively_breadth_plus_self_plus_self_per_each_raw {113		let b in 0..200;114		bench_init!{115			owner: sub; collection: collection(owner);116			sender: cross_from_sub(owner); burner: cross_sub;117		};118		let item = create_max_item(&collection, &sender, burner.clone())?;119		for i in 0..b {120			create_max_item(&collection, &sender, T::CrossTokenAddressMapping::token_to_address(collection.id, item))?;121		}122	}: {<Pallet<T>>::burn_recursively(&collection, &burner, item, &Unlimited, &Unlimited)?}123124	transfer_raw {125		bench_init!{126			owner: sub; collection: collection(owner);127			owner: cross_from_sub; sender: cross_sub; receiver: cross_sub;128		};129		let item = create_max_item(&collection, &owner, sender.clone())?;130	}: {<Pallet<T>>::transfer(&collection, &sender, &receiver, item, &Unlimited)?}131132	approve {133		bench_init!{134			owner: sub; collection: collection(owner);135			owner: cross_from_sub; sender: cross_sub; spender: cross_sub;136		};137		let item = create_max_item(&collection, &owner, sender.clone())?;138	}: {<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?}139140	approve_from {141		bench_init!{142			owner: sub; collection: collection(owner);143			owner: cross_from_sub; sender: cross_sub; spender: cross_sub;144		};145		let owner_eth = T::CrossAccountId::from_eth(*sender.as_eth());146		let item = create_max_item(&collection, &owner, owner_eth.clone())?;147	}: {<Pallet<T>>::set_allowance_from(&collection, &sender, &owner_eth, item, Some(&spender))?}148149	check_allowed_raw {150		bench_init!{151			owner: sub; collection: collection(owner);152			owner: cross_from_sub; sender: cross_sub; spender: cross_sub; receiver: cross_sub;153		};154		let item = create_max_item(&collection, &owner, sender.clone())?;155		<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&spender))?;156	}: {<Pallet<T>>::check_allowed(&collection, &spender, &sender, item, &Unlimited)?}157158	burn_from {159		bench_init!{160			owner: sub; collection: collection(owner);161			owner: cross_from_sub; sender: cross_sub; burner: cross_sub;162		};163		let item = create_max_item(&collection, &owner, sender.clone())?;164		<Pallet<T>>::set_allowance(&collection, &sender, item, Some(&burner))?;165	}: {<Pallet<T>>::burn_from(&collection, &burner, &sender, item, &Unlimited)?}166167	set_token_property_permissions {168		let b in 0..MAX_PROPERTIES_PER_ITEM;169		bench_init!{170			owner: sub; collection: collection(owner);171			owner: cross_from_sub;172		};173		let perms = (0..b).map(|k| PropertyKeyPermission {174			key: property_key(k as usize),175			permission: PropertyPermission {176				mutable: false,177				collection_admin: false,178				token_owner: false,179			},180		}).collect::<Vec<_>>();181	}: {<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?}182183	set_token_properties {184		let b in 0..MAX_PROPERTIES_PER_ITEM;185		bench_init!{186			owner: sub; collection: collection(owner);187			owner: cross_from_sub;188		};189		let perms = (0..b).map(|k| PropertyKeyPermission {190			key: property_key(k as usize),191			permission: PropertyPermission {192				mutable: false,193				collection_admin: true,194				token_owner: true,195			},196		}).collect::<Vec<_>>();197		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;198		let props = (0..b).map(|k| Property {199			key: property_key(k as usize),200			value: property_value(),201		}).collect::<Vec<_>>();202		let item = create_max_item(&collection, &owner, owner.clone())?;203	}: {<Pallet<T>>::set_token_properties(&collection, &owner, item, props.into_iter(), &Unlimited)?}204205	init_token_properties {206		let b in 0..MAX_PROPERTIES_PER_ITEM;207		bench_init!{208			owner: sub; collection: collection(owner);209			owner: cross_from_sub;210		};211212		let perms = (0..b).map(|k| PropertyKeyPermission {213			key: property_key(k as usize),214			permission: PropertyPermission {215				mutable: false,216				collection_admin: true,217				token_owner: true,218			},219		}).collect::<Vec<_>>();220		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;221		let props = (0..b).map(|k| Property {222			key: property_key(k as usize),223			value: property_value(),224		}).collect::<Vec<_>>();225		let item = create_max_item(&collection, &owner, owner.clone())?;226227		let (is_collection_admin, property_permissions) = load_is_admin_and_property_permissions(&collection, &owner);228	}: {229		let mut property_writer = pallet_common::collection_info_loaded_property_writer(230			&collection,231			is_collection_admin,232			property_permissions,233		);234235		property_writer.write_token_properties(236			true,237			item,238			props.into_iter(),239			crate::erc::ERC721TokenEvent::TokenChanged {240				token_id: item.into(),241			}242			.to_log(T::ContractAddress::get()),243		)?244	}245246	delete_token_properties {247		let b in 0..MAX_PROPERTIES_PER_ITEM;248		bench_init!{249			owner: sub; collection: collection(owner);250			owner: cross_from_sub;251		};252		let perms = (0..b).map(|k| PropertyKeyPermission {253			key: property_key(k as usize),254			permission: PropertyPermission {255				mutable: true,256				collection_admin: true,257				token_owner: true,258			},259		}).collect::<Vec<_>>();260		<Pallet<T>>::set_token_property_permissions(&collection, &owner, perms)?;261		let props = (0..b).map(|k| Property {262			key: property_key(k as usize),263			value: property_value(),264		}).collect::<Vec<_>>();265		let item = create_max_item(&collection, &owner, owner.clone())?;266		<Pallet<T>>::set_token_properties(&collection, &owner, item, props.into_iter(), &Unlimited)?;267		let to_delete = (0..b).map(|k| property_key(k as usize)).collect::<Vec<_>>();268	}: {<Pallet<T>>::delete_token_properties(&collection, &owner, item, to_delete.into_iter(), &Unlimited)?}269270	token_owner {271		bench_init!{272			owner: sub; collection: collection(owner);273			owner: cross_from_sub;274		};275		let item = create_max_item(&collection, &owner, owner.clone())?;276	}: {collection.token_owner(item).unwrap()}277278	set_allowance_for_all {279		bench_init!{280			owner: sub; collection: collection(owner); owner: cross_from_sub;281			operator: cross_sub;282		};283	}: {<Pallet<T>>::set_allowance_for_all(&collection, &owner, &operator, true)?}284285	allowance_for_all {286		bench_init!{287			owner: sub; collection: collection(owner); owner: cross_from_sub;288			operator: cross_sub;289		};290	}: {<Pallet<T>>::allowance_for_all(&collection, &owner, &operator)}291292	repair_item {293		bench_init!{294			owner: sub; collection: collection(owner);295			owner: cross_from_sub;296		};297		let item = create_max_item(&collection, &owner, owner.clone())?;298	}: {<Pallet<T>>::repair_item(&collection, item)?}299}