From 3e10ee224628003a86f816a78145560bccc21a2c Mon Sep 17 00:00:00 2001 From: Dmitry Kolupaev Date: Wed, 20 Mar 2024 12:04:57 +0400 Subject: [PATCH] fix: load user info safely #ntrn-240 --- contracts/lockdrop-pcl/src/contract.rs | 4 +--- contracts/lockdrop/src/contract.rs | 8 ++------ 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/contracts/lockdrop-pcl/src/contract.rs b/contracts/lockdrop-pcl/src/contract.rs index 4581dab1..0eaf15fb 100644 --- a/contracts/lockdrop-pcl/src/contract.rs +++ b/contracts/lockdrop-pcl/src/contract.rs @@ -437,9 +437,7 @@ pub fn handle_claim_rewards_and_unlock_for_lockup( // CHECK ::: Is LP Token Pool supported or not ? let pool_info = ASSET_POOLS.load(deps.storage, pool_type)?; - let mut user_info = USER_INFO - .may_load(deps.storage, &user_address)? - .unwrap_or_default(); + let mut user_info = USER_INFO.load(deps.storage, &user_address)?; // If user's total NTRN rewards == 0 :: We update all of the user's lockup positions to calculate NTRN rewards and for each alongwith their equivalent Astroport LP Shares if user_info.total_ntrn_rewards == Uint128::zero() { diff --git a/contracts/lockdrop/src/contract.rs b/contracts/lockdrop/src/contract.rs index a275602f..23271bf0 100644 --- a/contracts/lockdrop/src/contract.rs +++ b/contracts/lockdrop/src/contract.rs @@ -530,9 +530,7 @@ pub fn handle_migrate_liquidity_to_pcl_pools( let config = CONFIG.load(deps.storage)?; let state = STATE.load(deps.storage)?; - let mut user_info = USER_INFO - .may_load(deps.storage, &user_address)? - .unwrap_or_default(); + let mut user_info = USER_INFO.load(deps.storage, &user_address)?; // If user's total NTRN rewards == 0 :: We update all of the user's lockup positions to // calculate NTRN rewards and for each alongwith their equivalent Astroport LP Shares if user_info.total_ntrn_rewards == Uint128::zero() { @@ -1190,9 +1188,7 @@ pub fn handle_claim_rewards_and_unlock_for_lockup( // CHECK ::: Is LP Token Pool supported or not ? let pool_info = ASSET_POOLS.load(deps.storage, pool_type)?; - let mut user_info = USER_INFO - .may_load(deps.storage, &user_address)? - .unwrap_or_default(); + let mut user_info = USER_INFO.load(deps.storage, &user_address)?; // If user's total NTRN rewards == 0 :: We update all of the user's lockup positions to calculate NTRN rewards and for each alongwith their equivalent Astroport LP Shares if user_info.total_ntrn_rewards == Uint128::zero() {