Skip to content

Commit

Permalink
Remove Storage getters from pallets (#907)
Browse files Browse the repository at this point in the history
* Remove getters from pallet echo

* Remove getters from pallet-evm-chain-id

* Remove getters from pallet-token-approvals

* Remove getters from pallet-dex

* Remove getters from pallet-xrpl-bridge

* Remove getters from pallet-marketplace

* fmt

* Fix CI issues
  • Loading branch information
JasonTulp authored Dec 3, 2024
1 parent 4ab9b18 commit 08b4c54
Show file tree
Hide file tree
Showing 19 changed files with 242 additions and 259 deletions.
8 changes: 4 additions & 4 deletions evm-precompiles/erc1155/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -322,7 +322,7 @@ where
let operator: Runtime::AccountId = H160::from(operator).into();

handle.record_cost(RuntimeHelper::<Runtime>::db_read_gas_cost())?;
let is_approved = pallet_token_approvals::Pallet::<Runtime>::erc1155_approvals_for_all(
let is_approved = pallet_token_approvals::ERC1155ApprovalsForAll::<Runtime>::get(
owner,
(collection_id, operator),
)
Expand Down Expand Up @@ -582,7 +582,7 @@ where
// Check approvals
if from != handle.context().caller {
handle.record_cost(RuntimeHelper::<Runtime>::db_read_gas_cost())?;
let is_approved = pallet_token_approvals::Pallet::<Runtime>::erc1155_approvals_for_all(
let is_approved = pallet_token_approvals::ERC1155ApprovalsForAll::<Runtime>::get(
Runtime::AccountId::from(from),
(collection_id, Runtime::AccountId::from(handle.context().caller)),
)
Expand Down Expand Up @@ -629,7 +629,7 @@ where
// Check approvals
if operator != handle.context().caller {
handle.record_cost(RuntimeHelper::<Runtime>::db_read_gas_cost())?;
let is_approved = pallet_token_approvals::Pallet::<Runtime>::erc1155_approvals_for_all(
let is_approved = pallet_token_approvals::ERC1155ApprovalsForAll::<Runtime>::get(
Runtime::AccountId::from(operator),
(collection_id, Runtime::AccountId::from(handle.context().caller)),
)
Expand Down Expand Up @@ -697,7 +697,7 @@ where
// Check approvals
if operator != handle.context().caller {
handle.record_cost(RuntimeHelper::<Runtime>::db_read_gas_cost())?;
let is_approved = pallet_token_approvals::Pallet::<Runtime>::erc1155_approvals_for_all(
let is_approved = pallet_token_approvals::ERC1155ApprovalsForAll::<Runtime>::get(
Runtime::AccountId::from(operator),
(collection_id, Runtime::AccountId::from(handle.context().caller)),
)
Expand Down
10 changes: 4 additions & 6 deletions evm-precompiles/erc20/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -217,12 +217,10 @@ where
let spender: Runtime::AccountId = H160::from(spender).into();

// Fetch info.
let amount: U256 = pallet_token_approvals::Pallet::<Runtime>::erc20_approvals(
(&owner, &asset_id),
&spender,
)
.unwrap_or_default()
.into();
let amount: U256 =
pallet_token_approvals::ERC20Approvals::<Runtime>::get((&owner, &asset_id), &spender)
.unwrap_or_default()
.into();

// Build output.
Ok(succeed(EvmDataWriter::new().write(amount).build()))
Expand Down
16 changes: 7 additions & 9 deletions evm-precompiles/erc721/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -581,14 +581,12 @@ where

// Return either the approved account or zero address if no account is approved
handle.record_cost(RuntimeHelper::<Runtime>::db_read_gas_cost())?;
let approved_account: H160 =
match pallet_token_approvals::Pallet::<Runtime>::erc721_approvals((
collection_id,
serial_number,
)) {
Some(approved_account) => (approved_account).into(),
None => H160::default(),
};
let approved_account: H160 = match pallet_token_approvals::ERC721Approvals::<Runtime>::get(
(collection_id, serial_number),
) {
Some(approved_account) => (approved_account).into(),
None => H160::default(),
};

Ok(succeed(EvmDataWriter::new().write(Address::from(approved_account)).build()))
}
Expand All @@ -605,7 +603,7 @@ where
let operator: Runtime::AccountId = H160::from(operator).into();

handle.record_cost(RuntimeHelper::<Runtime>::db_read_gas_cost())?;
let is_approved = pallet_token_approvals::Pallet::<Runtime>::erc721_approvals_for_all(
let is_approved = pallet_token_approvals::ERC721ApprovalsForAll::<Runtime>::get(
owner,
(collection_id, operator),
)
Expand Down
31 changes: 13 additions & 18 deletions pallet/dex/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -224,25 +224,20 @@ pub mod pallet {

/// FeeTo account where network fees are deposited
#[pallet::storage]
#[pallet::getter(fn fee_to)]
pub type FeeTo<T: Config> = StorageValue<_, Option<T::AccountId>, ValueQuery, DefaultFeeTo<T>>;

#[pallet::storage]
#[pallet::getter(fn lp_token_id)]
pub type TradingPairLPToken<T: Config> =
StorageMap<_, Twox64Concat, TradingPair, Option<AssetId>, ValueQuery>;

#[pallet::storage]
#[pallet::getter(fn liquidity_pool)]
pub type LiquidityPool<T: Config> =
StorageMap<_, Twox64Concat, TradingPair, (Balance, Balance), ValueQuery>;

#[pallet::storage]
#[pallet::getter(fn liquidity_pool_last_k)]
pub type LiquidityPoolLastK<T: Config> = StorageMap<_, Twox64Concat, AssetId, U256, ValueQuery>;

#[pallet::storage]
#[pallet::getter(fn trading_pair_statuses)]
pub type TradingPairStatuses<T: Config> =
StorageMap<_, Twox64Concat, TradingPair, TradingPairStatus, ValueQuery>;

Expand Down Expand Up @@ -466,11 +461,11 @@ pub mod pallet {
let trading_pair = TradingPair::new(token_a, token_b);

ensure!(
Self::lp_token_id(&trading_pair).is_some(),
TradingPairLPToken::<T>::get(&trading_pair).is_some(),
Error::<T>::LiquidityProviderTokenNotCreated
);

match Self::trading_pair_statuses(&trading_pair) {
match TradingPairStatuses::<T>::get(&trading_pair) {
TradingPairStatus::Enabled => return Err(Error::<T>::MustBeNotEnabled.into()),
// will enabled Disabled trading_pair
TradingPairStatus::NotEnabled => {
Expand Down Expand Up @@ -500,11 +495,11 @@ pub mod pallet {
let trading_pair = TradingPair::new(token_a, token_b);

ensure!(
Self::lp_token_id(&trading_pair).is_some(),
TradingPairLPToken::<T>::get(&trading_pair).is_some(),
Error::<T>::LiquidityProviderTokenNotCreated
);

match Self::trading_pair_statuses(&trading_pair) {
match TradingPairStatuses::<T>::get(&trading_pair) {
// will disable Enabled trading_pair
TradingPairStatus::Enabled => {
TradingPairStatuses::<T>::insert(trading_pair, TradingPairStatus::NotEnabled);
Expand Down Expand Up @@ -602,13 +597,13 @@ where

let trading_pair = TradingPair::new(token_a, token_b);
// create trading pair & lp token if non-existent
let lp_share_asset_id = match Self::lp_token_id(trading_pair) {
let lp_share_asset_id = match TradingPairLPToken::<T>::get(trading_pair) {
Some(lp_id) => lp_id,
None => Self::create_lp_token(&trading_pair)?,
};

ensure!(
matches!(Self::trading_pair_statuses(&trading_pair), TradingPairStatus::Enabled),
matches!(TradingPairStatuses::<T>::get(&trading_pair), TradingPairStatus::Enabled),
Error::<T>::MustBeEnabled,
);

Expand Down Expand Up @@ -769,7 +764,7 @@ where

let trading_pair = TradingPair::new(token_a, token_b);
let lp_share_asset_id =
Self::lp_token_id(trading_pair).ok_or(Error::<T>::InvalidAssetId)?;
TradingPairLPToken::<T>::get(trading_pair).ok_or(Error::<T>::InvalidAssetId)?;

ensure!(token_a != token_b, Error::<T>::IdenticalTokenAddress);

Expand Down Expand Up @@ -876,12 +871,12 @@ where
token_b: AssetId,
) -> sp_std::result::Result<AssetId, DispatchError> {
let trading_pair = TradingPair::new(token_a, token_b);
Self::lp_token_id(trading_pair).ok_or(Error::<T>::InvalidAssetId.into())
TradingPairLPToken::<T>::get(trading_pair).ok_or(Error::<T>::InvalidAssetId.into())
}

pub fn get_liquidity(token_a: AssetId, token_b: AssetId) -> (Balance, Balance) {
let trading_pair = TradingPair::new(token_a, token_b);
let (reserve_0, reserve_1) = Self::liquidity_pool(trading_pair);
let (reserve_0, reserve_1) = LiquidityPool::<T>::get(trading_pair);
if token_a == trading_pair.0 {
(reserve_0, reserve_1)
} else {
Expand All @@ -891,7 +886,7 @@ where

pub fn get_trading_pair_status(token_a: AssetId, token_b: AssetId) -> TradingPairStatus {
let trading_pair = TradingPair::new(token_a, token_b);
Self::trading_pair_statuses(trading_pair)
TradingPairStatuses::<T>::get(trading_pair)
}

/// Given an input amount of an asset and pair reserves, returns the maximum output amount of
Expand Down Expand Up @@ -965,7 +960,7 @@ where
// trading pair in path must be enabled
ensure!(
matches!(
Self::trading_pair_statuses(&TradingPair::new(path[i], path[i + 1])),
TradingPairStatuses::<T>::get(&TradingPair::new(path[i], path[i + 1])),
TradingPairStatus::Enabled
),
Error::<T>::MustBeEnabled
Expand Down Expand Up @@ -1006,7 +1001,7 @@ where
// trading pair in path must be enabled
ensure!(
matches!(
Self::trading_pair_statuses(&TradingPair::new(path[i - 1], path[i])),
TradingPairStatuses::<T>::get(&TradingPair::new(path[i - 1], path[i])),
TradingPairStatus::Enabled
),
Error::<T>::MustBeEnabled
Expand Down Expand Up @@ -1160,7 +1155,7 @@ where
if FeeTo::<T>::get().is_some() {
// get the lp shared asset id
let lp_share_asset_id =
Self::lp_token_id(trading_pair).ok_or(Error::<T>::InvalidAssetId)?;
TradingPairLPToken::<T>::get(trading_pair).ok_or(Error::<T>::InvalidAssetId)?;

// get the updated reserve values of the trading pair
let (reserve_a, reserve_b) = LiquidityPool::<T>::get(trading_pair);
Expand Down
Loading

0 comments on commit 08b4c54

Please sign in to comment.