From eac070494702a4c44efc86cc2499d8fdd24e9fef Mon Sep 17 00:00:00 2001 From: Daniel Shiposha Date: Thu, 30 Jun 2022 22:19:15 +0000 Subject: [PATCH] fix: use get_or_insert_with --- --- a/pallets/nonfungible/src/property_guard.rs +++ b/pallets/nonfungible/src/property_guard.rs @@ -32,16 +32,13 @@ } pub fn check_collection_admin(&mut self) -> DispatchResult { - if self.collection_admin_result.is_none() { - self.collection_admin_result = - Some(self.collection.check_is_owner_or_admin(self.sender)); - } - - self.collection_admin_result.unwrap() + *self.collection_admin_result.get_or_insert_with(|| { + self.collection.check_is_owner_or_admin(self.sender) + }) } pub fn check_token_owner(&mut self) -> DispatchResult { - if self.token_owner_result.is_none() { + *self.token_owner_result.get_or_insert_with(|| { let is_owned = >::check_indirectly_owned( self.sender.clone(), self.collection.id, @@ -50,15 +47,11 @@ self.budget, )?; - let result = if is_owned { + if is_owned { Ok(()) } else { Err(>::NoPermission.into()) - }; - - self.token_owner_result = Some(result); - } - - self.token_owner_result.unwrap() + } + }) } } -- gitstuff