Merge pull request #792 from multiversx/ugrade-endpoints

separate upgrade endpoint
multiversx · Dec 13, 2023 · a49c37f · a49c37f
2 parents bf85891 + d35e367
commit a49c37f
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/dex/farm-with-locked-rewards/src/lib.rs b/dex/farm-with-locked-rewards/src/lib.rs
@@ -82,7 +82,11 @@ pub trait Farm:
     }
 
     #[endpoint]
-    fn upgrade(&self) {}
+    fn upgrade(&self) {
+        // Farm position migration code
+        let farm_token_mapper = self.farm_token();
+        self.try_set_farm_position_migration_nonce(farm_token_mapper);
+    }
 
     #[payable("*")]
     #[endpoint(enterFarm)]

diff --git a/dex/farm/src/lib.rs b/dex/farm/src/lib.rs
@@ -83,7 +83,11 @@ pub trait Farm:
     }
 
     #[endpoint]
-    fn upgrade(&self) {}
+    fn upgrade(&self) {
+        // Farm position migration code
+        let farm_token_mapper = self.farm_token();
+        self.try_set_farm_position_migration_nonce(farm_token_mapper);
+    }
 
     #[payable("*")]
     #[endpoint(enterFarm)]

diff --git a/farm-staking/farm-staking/src/lib.rs b/farm-staking/farm-staking/src/lib.rs
@@ -95,7 +95,11 @@ pub trait FarmStaking:
     }
 
     #[endpoint]
-    fn upgrade(&self) {}
+    fn upgrade(&self) {
+        // Farm position migration code
+        let farm_token_mapper = self.farm_token();
+        self.try_set_farm_position_migration_nonce(farm_token_mapper);
+    }
 
     #[payable("*")]
     #[endpoint(mergeFarmTokens)]