Skip to content
Snippets Groups Projects
Unverified Commit 4bb71f5b authored by bgallois's avatar bgallois
Browse files

remove weight accounting OnNewIdty

parent ec620b8a
No related branches found
No related tags found
1 merge request!246Fix weight accounting
...@@ -38,7 +38,7 @@ mod benchmarks { ...@@ -38,7 +38,7 @@ mod benchmarks {
fn add_certs<T: Config>(i: u32, receiver: T::IdtyIndex) -> Result<(), &'static str> { fn add_certs<T: Config>(i: u32, receiver: T::IdtyIndex) -> Result<(), &'static str> {
Pallet::<T>::remove_all_certs_received_by(RawOrigin::Root.into(), receiver)?; Pallet::<T>::remove_all_certs_received_by(RawOrigin::Root.into(), receiver)?;
for j in 1..i { for j in 1..i {
let _ = Pallet::<T>::do_add_cert_checked(j.into(), receiver, false); Pallet::<T>::do_add_cert_checked(j.into(), receiver, false);
} }
assert!( assert!(
CertsByReceiver::<T>::get(receiver).len() as u32 == i - 1, CertsByReceiver::<T>::get(receiver).len() as u32 == i - 1,
...@@ -103,7 +103,7 @@ mod benchmarks { ...@@ -103,7 +103,7 @@ mod benchmarks {
let receiver: T::IdtyIndex = 2.into(); let receiver: T::IdtyIndex = 2.into();
// try to add cert if missing, else ignore // try to add cert if missing, else ignore
// this depends on initial data // this depends on initial data
let _ = Pallet::<T>::do_add_cert_checked(issuer, receiver, false); Pallet::<T>::do_add_cert_checked(issuer, receiver, false);
#[extrinsic_call] #[extrinsic_call]
_(RawOrigin::Root, issuer, receiver); _(RawOrigin::Root, issuer, receiver);
...@@ -154,7 +154,7 @@ mod benchmarks { ...@@ -154,7 +154,7 @@ mod benchmarks {
let issuer: T::IdtyIndex = 1.into(); let issuer: T::IdtyIndex = 1.into();
let receiver: T::IdtyIndex = 0.into(); let receiver: T::IdtyIndex = 0.into();
Pallet::<T>::do_remove_cert(issuer, receiver, None); Pallet::<T>::do_remove_cert(issuer, receiver, None);
let _ = Pallet::<T>::do_add_cert_checked(issuer, receiver, false); Pallet::<T>::do_add_cert_checked(issuer, receiver, false);
let block_number = T::ValidityPeriod::get(); let block_number = T::ValidityPeriod::get();
frame_system::pallet::Pallet::<T>::set_block_number(block_number); frame_system::pallet::Pallet::<T>::set_block_number(block_number);
......
...@@ -360,21 +360,19 @@ pub mod pallet { ...@@ -360,21 +360,19 @@ pub mod pallet {
issuer: T::IdtyIndex, issuer: T::IdtyIndex,
receiver: T::IdtyIndex, receiver: T::IdtyIndex,
verify_rules: bool, verify_rules: bool,
) -> Weight { ) {
let block_number = frame_system::pallet::Pallet::<T>::block_number(); let block_number = frame_system::pallet::Pallet::<T>::block_number();
if verify_rules { if verify_rules {
// only verify internal rules if asked // only verify internal rules if asked
if let Err(e) = Self::check_add_cert_internal(issuer, receiver, block_number) { if let Err(e) = Self::check_add_cert_internal(issuer, receiver, block_number) {
sp_std::if_std! {println!("fail to force add cert: {:?}", e)} sp_std::if_std! {println!("fail to force add cert: {:?}", e)}
return T::WeightInfo::do_add_cert_checked();
} }
} }
if let Err(e) = Self::try_add_cert(block_number, issuer, receiver) { if let Err(e) = Self::try_add_cert(block_number, issuer, receiver) {
sp_std::if_std! {println!("fail to force add cert: {:?}", e)} sp_std::if_std! {println!("fail to force add cert: {:?}", e)}
} }
T::WeightInfo::do_add_cert_checked()
} }
/// perform cert addition if not existing, else CertAlreadyExists /// perform cert addition if not existing, else CertAlreadyExists
......
...@@ -222,15 +222,9 @@ where ...@@ -222,15 +222,9 @@ where
// implement identity event handler // implement identity event handler
impl<T: Config> pallet_identity::traits::OnNewIdty<T> for Pallet<T> { impl<T: Config> pallet_identity::traits::OnNewIdty<T> for Pallet<T> {
fn on_created(idty_index: &IdtyIndex, creator: &IdtyIndex) -> Weight { fn on_created(idty_index: &IdtyIndex, creator: &IdtyIndex) {
let mut weight = Weight::zero();
// identity just has been created, a cert must be added // identity just has been created, a cert must be added
weight = weight.saturating_add(<pallet_certification::Pallet<T>>::do_add_cert_checked( <pallet_certification::Pallet<T>>::do_add_cert_checked(*creator, *idty_index, true);
*creator,
*idty_index,
true,
));
weight
} }
} }
......
...@@ -36,7 +36,7 @@ pub trait IdtyNameValidator { ...@@ -36,7 +36,7 @@ pub trait IdtyNameValidator {
} }
pub trait OnNewIdty<T: Config> { pub trait OnNewIdty<T: Config> {
fn on_created(idty_index: &T::IdtyIndex, creator: &T::IdtyIndex) -> Weight; fn on_created(idty_index: &T::IdtyIndex, creator: &T::IdtyIndex);
} }
pub trait OnRemoveIdty<T: Config> { pub trait OnRemoveIdty<T: Config> {
...@@ -47,10 +47,8 @@ pub trait OnRemoveIdty<T: Config> { ...@@ -47,10 +47,8 @@ pub trait OnRemoveIdty<T: Config> {
#[impl_for_tuples(5)] #[impl_for_tuples(5)]
#[allow(clippy::let_and_return)] #[allow(clippy::let_and_return)]
impl<T: Config> OnNewIdty<T> for Tuple { impl<T: Config> OnNewIdty<T> for Tuple {
fn on_created(idty_index: &T::IdtyIndex, creator: &T::IdtyIndex) -> Weight { fn on_created(idty_index: &T::IdtyIndex, creator: &T::IdtyIndex) {
let mut weight = Weight::zero(); for_tuples!( #( Tuple::on_created(idty_index, creator); )* );
for_tuples!( #( weight = weight.saturating_add(Tuple::on_created(idty_index, creator)); )* );
weight
} }
} }
......
...@@ -339,12 +339,7 @@ fn is_eligible_for_refund<T: pallet_identity::Config>(_identity: IdtyId<T>) -> b ...@@ -339,12 +339,7 @@ fn is_eligible_for_refund<T: pallet_identity::Config>(_identity: IdtyId<T>) -> b
// implement identity event handler // implement identity event handler
impl<T: Config> pallet_identity::traits::OnNewIdty<T> for Pallet<T> { impl<T: Config> pallet_identity::traits::OnNewIdty<T> for Pallet<T> {
fn on_created(idty_index: &IdtyId<T>, _creator: &T::IdtyIndex) -> Weight { fn on_created(idty_index: &IdtyId<T>, _creator: &T::IdtyIndex) {
let mut weight = Weight::zero();
let mut add_db_reads_writes = |reads, writes| {
weight = weight.saturating_add(T::DbWeight::get().reads_writes(reads, writes));
};
IdtyQuota::<T>::insert( IdtyQuota::<T>::insert(
idty_index, idty_index,
Quota { Quota {
...@@ -352,8 +347,6 @@ impl<T: Config> pallet_identity::traits::OnNewIdty<T> for Pallet<T> { ...@@ -352,8 +347,6 @@ impl<T: Config> pallet_identity::traits::OnNewIdty<T> for Pallet<T> {
amount: BalanceOf::<T>::zero(), amount: BalanceOf::<T>::zero(),
}, },
); );
add_db_reads_writes(0, 1);
weight
} }
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment