refactor amm cache

Author: moosecat2

programs/drift/src/instructions/lp_admin.rs

@@ -4,7 +4,7 @@ use crate::ids::{admin_hot_wallet, lp_pool_swap_wallet};
 use crate::instructions::optional_accounts::{get_token_mint, load_maps, AccountMaps};
 use crate::math::constants::{PRICE_PRECISION_U64, QUOTE_SPOT_MARKET_INDEX};
 use crate::math::safe_math::SafeMath;
-use crate::state::amm_cache::{AmmCache, AMM_POSITIONS_CACHE};
+use crate::state::amm_cache::{AmmCache, CacheInfo, AMM_POSITIONS_CACHE};
 use crate::state::lp_pool::{
     AmmConstituentDatum, AmmConstituentMapping, Constituent, ConstituentCorrelations,
     ConstituentTargetBase, LPPool, TargetsDatum, AMM_MAP_PDA_SEED,
@@ -937,6 +937,21 @@ pub fn handle_update_initial_amm_cache_info<'c: 'info, 'info>(
 
     Ok(())
 }
+
+pub fn handle_reset_amm_cache(ctx: Context<ResetAmmCache>) -> Result<()> {
+    let state = &ctx.accounts.state;
+    let amm_cache = &mut ctx.accounts.amm_cache;
+
+    amm_cache.cache.clear();
+    amm_cache
+        .cache
+        .resize_with(state.number_of_markets as usize, CacheInfo::default);
+    amm_cache.validate(state)?;
+
+    msg!("AMM cache reset. markets: {}", state.number_of_markets);
+    Ok(())
+}
+
 #[derive(Debug, Clone, Copy, AnchorSerialize, AnchorDeserialize, PartialEq, Eq)]
 pub struct OverrideAmmCacheParams {
     pub quote_owed_from_lp_pool: Option<i64>,
@@ -1368,3 +1383,23 @@ pub struct UpdateInitialAmmCacheInfo<'info> {
     )]
     pub amm_cache: Box<Account<'info, AmmCache>>,
 }
+
+#[derive(Accounts)]
+pub struct ResetAmmCache<'info> {
+    #[account(
+        mut,
+        constraint = admin.key() == admin_hot_wallet::id() || admin.key() == state.admin
+    )]
+    pub admin: Signer<'info>,
+    pub state: Box<Account<'info, State>>,
+    #[account(
+        mut,
+        seeds = [AMM_POSITIONS_CACHE.as_ref()],
+        bump = amm_cache.bump,
+        realloc = AmmCache::space(state.number_of_markets as usize),
+        realloc::payer = admin,
+        realloc::zero = false,
+    )]
+    pub amm_cache: Box<Account<'info, AmmCache>>,
+    pub system_program: Program<'info, System>,
+}

programs/drift/src/lib.rs

@@ -2066,6 +2066,12 @@ pub mod drift {
         handle_override_amm_cache_info(ctx, market_index, override_params)
     }
 
+    pub fn reset_amm_cache<'c: 'info, 'info>(
+        ctx: Context<'_, '_, 'c, 'info, ResetAmmCache<'info>>,
+    ) -> Result<()> {
+        handle_reset_amm_cache(ctx)
+    }
+
     pub fn lp_pool_swap<'c: 'info, 'info>(
         ctx: Context<'_, '_, 'c, 'info, LPPoolSwap<'info>>,
         in_market_index: u16,

programs/drift/src/state/amm_cache.rs

@@ -57,11 +57,11 @@ pub struct CacheInfo {
     pub oracle_validity: u8,
     pub lp_status_for_perp_market: u8,
     pub amm_position_scalar: u8,
-    _padding: [u8; 28],
+    _padding: [u8; 36],
 }
 
 impl Size for CacheInfo {
-    const SIZE: usize = 224;
+    const SIZE: usize = 230;
 }
 
 impl Default for CacheInfo {
@@ -86,7 +86,7 @@ impl Default for CacheInfo {
             quote_owed_from_lp_pool: 0i64,
             lp_status_for_perp_market: 0u8,
             amm_position_scalar: 0u8,
-            _padding: [0u8; 28],
+            _padding: [0u8; 36],
         }
     }
 }

sdk/src/idl/drift.json

@@ -8427,6 +8427,32 @@
         }
       ]
     },
+    {
+      "name": "resetAmmCache",
+      "accounts": [
+        {
+          "name": "admin",
+          "isMut": true,
+          "isSigner": true
+        },
+        {
+          "name": "state",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "ammCache",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "systemProgram",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": []
+    },
     {
       "name": "lpPoolSwap",
       "accounts": [
@@ -12193,7 +12219,7 @@
             "type": {
               "array": [
                 "u8",
-                28
+                36
               ]
             }
           }