Refactor rewarders structure

AstarNetwork · Oct 18, 2022 · 8e2dd3a · 8e2dd3a
1 parent bcc6192
commit 8e2dd3a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/farming/logics/traits/data.rs b/farming/logics/traits/data.rs
@@ -47,7 +47,7 @@ pub struct Data {
     pub lp_tokens: Vec<AccountId>,
 
     /// Address of each `rewarder` contract in MasterChef.
-    pub rewarders: Vec<Option<AccountId>>,
+    pub rewarders: Mapping<u32, AccountId>,
 
     /// Total allocation points. Must be the sum of all allocation points in all pools.
     pub total_alloc_point: u32,

diff --git a/farming/logics/traits/farming.rs b/farming/logics/traits/farming.rs
@@ -47,9 +47,12 @@ pub trait Farming: Storage<Data> + Storage<ownable::Data> + FarmingGetters + Far
             .checked_add(alloc_point)
             .ok_or(FarmingError::AddOverflow2)?;
         self.data::<Data>().lp_tokens.push(lp_token);
-        self.data::<Data>().rewarders.push(rewarder);
         let pool_length = self.pool_length();
-
+        if let Some(rewarder_address) = rewarder {
+            self.data::<Data>()
+                .rewarders
+                .insert(&pool_length, &rewarder_address);
+        }
         self.data::<Data>().pool_info.insert(
             &pool_length,
             &Pool {