From f52337afd8963daf3dfda7beb8c296f07b935e15 Mon Sep 17 00:00:00 2001 From: Mosharraf Date: Fri, 25 Sep 2026 00:46:56 +0530 Subject: [PATCH] feat(yield-vault): add storage key enums --- contracts/contracts/yield_vault/src/lib.rs | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/contracts/contracts/yield_vault/src/lib.rs b/contracts/contracts/yield_vault/src/lib.rs index 75e42a8..937f577 100644 --- a/contracts/contracts/yield_vault/src/lib.rs +++ b/contracts/contracts/yield_vault/src/lib.rs @@ -11,8 +11,6 @@ const TOKEN_KEY: Symbol = symbol_short!("token"); const APY_KEY: Symbol = symbol_short!("apy"); const PROPOSED_APY_KEY: Symbol = symbol_short!("prop_apy"); const APY_ACTIVATION_KEY: Symbol = symbol_short!("apy_act"); -const SHARES_KEY: Symbol = symbol_short!("tot_shr"); -const ASSETS_KEY: Symbol = symbol_short!("tot_ast"); const IDX_KEY: Symbol = symbol_short!("yld_idx"); const IDX_LED_KEY: Symbol = symbol_short!("idx_led"); const PROTO_BAL_KEY: Symbol = symbol_short!("p_bal"); @@ -154,11 +152,10 @@ mod sandbox_protocol { #[contracttype] enum DataKey { UserShares(Address), - /// Reentrancy lock for withdraw. Held in temporary storage for the - /// duration of the token transfer so a callback cannot re-enter. Locked, - /// Track cumulative active deposited principal for each user. UserDeposit(Address), + TotalShares, + TotalAssets, } #[contract]