fix query_exchange_rate x2

hadronlabs-org · Nov 26, 2024 · c0529b2 · c0529b2
1 parent 3eaf9a5
commit c0529b2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/contracts/lazy-staking/src/contract.rs b/contracts/lazy-staking/src/contract.rs
@@ -73,10 +73,12 @@ fn query_exchange_rate(deps: Deps, env: Env) -> StdResult<Decimal> {
     ))?;
     let lazy_denom: String = DENOM.load(deps.storage)?;
     let lazy_total_supply = deps.querier.query_supply(lazy_denom)?;
-    let exchange_rate = asset_contract_balance.checked_mul(Decimal::from_ratio(
-        lazy_total_supply.amount,
-        Uint128::one(),
-    ))?;
+    let exchange_rate = asset_contract_balance
+        .checked_div(Decimal::from_ratio(
+            lazy_total_supply.amount,
+            Uint128::one(),
+        ))
+        .unwrap();
     Ok(exchange_rate)
 }
 

diff --git a/contracts/lazy-staking/src/error.rs b/contracts/lazy-staking/src/error.rs
@@ -1,4 +1,4 @@
-use cosmwasm_std::{CheckedFromRatioError, OverflowError, StdError};
+use cosmwasm_std::{CheckedFromRatioError, DivideByZeroError, OverflowError, StdError};
 use cw_ownable::OwnershipError;
 use cw_utils::PaymentError;
 use neutron_sdk::NeutronError;
@@ -28,6 +28,8 @@ pub enum ContractError {
     PaymentError(#[from] PaymentError),
     #[error("{0}")]
     CheckedFromRatioError(#[from] CheckedFromRatioError),
+    #[error("{0}")]
+    DivideByZeroError(#[from] DivideByZeroError),
 }
 
 impl From<semver::Error> for ContractError {