chore: sync dvn code

LayerZero-Labs · Oct 18, 2024 · cd66df2 · cd66df2
1 parent 05b9fa8
commit cd66df2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/packages/layerzero-v2/solana/programs/programs/dvn/src/instructions/admin/set_config.rs b/packages/layerzero-v2/solana/programs/programs/dvn/src/instructions/admin/set_config.rs
@@ -32,6 +32,7 @@ pub enum AdminConfig {
     DefaultMultiplierBps(u16),
     DstConfigs(Vec<DstConfig>),
     PriceFeed(Pubkey),
+    RemoveDstConfigs(Vec<u32>),
 }
 
 impl AdminConfig {
@@ -49,6 +50,9 @@ impl AdminConfig {
             AdminConfig::PriceFeed(price_feed) => {
                 config.price_feed = *price_feed;
             },
+            AdminConfig::RemoveDstConfigs(dst_eids) => {
+                config.remove_dst_configs(dst_eids.clone())?;
+            },
         }
         Ok(())
     }

diff --git a/packages/layerzero-v2/solana/programs/programs/dvn/src/state/dvn_config.rs b/packages/layerzero-v2/solana/programs/programs/dvn/src/state/dvn_config.rs
@@ -74,6 +74,13 @@ impl DvnConfig {
         }
         Ok(())
     }
+
+    pub fn remove_dst_configs(&mut self, dst_eids: Vec<u32>) -> Result<()> {
+        for eid in dst_eids {
+            sorted_list_helper::remove_from_sorted_list_by_eid(&mut self.dst_configs, eid)?;
+        }
+        Ok(())
+    }
 }
 
 impl sorted_list_helper::EID for DstConfig {