Fixed FunctionModifier code in Solidity tutorial

odradev · Jul 3, 2024 · 88a6529 · 88a6529
1 parent 8e3374b
commit 88a6529
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/docusaurus/docs/tutorials/odra-sol.md b/docusaurus/docs/tutorials/odra-sol.md
@@ -743,7 +743,7 @@ In Odra, you don't need to specify `view` or `pure` functions explicitly. All fu
 <TabItem value="rust" label="Odra">
 
 ```rust showLineNumbers
-use odra::Var;
+use odra::{module::Module, Var};
 
 #[odra::module]
 pub struct FunctionModifier {
@@ -753,7 +753,7 @@ pub struct FunctionModifier {
 
 #[odra::module]
 impl FunctionModifier {
-    pub fn decrement(&mut self, i: u32) -> u32 {
+    pub fn decrement(&mut self, i: u32) {
         self.lock();
         self.x.set(self.x.get_or_default() - i);
 
@@ -766,7 +766,7 @@ impl FunctionModifier {
     #[inline]
     fn lock(&mut self) {
         if self.locked.get_or_default() {
-            self.env().revert("No reentrancy");
+            self.env().revert(Error::NoReentrancy);
         }
 
         self.locked.set(true);
@@ -778,6 +778,11 @@ impl FunctionModifier {
     }
 }
 
+#[odra::odra_error]
+pub enum Error {
+    NoReentrancy = 1,
+}
+
 ```
 </TabItem>
 

diff --git a/docusaurus/versioned_docs/version-1.1.0/tutorials/odra-sol.md b/docusaurus/versioned_docs/version-1.1.0/tutorials/odra-sol.md
@@ -743,7 +743,7 @@ In Odra, you don't need to specify `view` or `pure` functions explicitly. All fu
 <TabItem value="rust" label="Odra">
 
 ```rust showLineNumbers
-use odra::Var;
+use odra::{module::Module, Var};
 
 #[odra::module]
 pub struct FunctionModifier {
@@ -753,7 +753,7 @@ pub struct FunctionModifier {
 
 #[odra::module]
 impl FunctionModifier {
-    pub fn decrement(&mut self, i: u32) -> u32 {
+    pub fn decrement(&mut self, i: u32) {
         self.lock();
         self.x.set(self.x.get_or_default() - i);
 
@@ -766,7 +766,7 @@ impl FunctionModifier {
     #[inline]
     fn lock(&mut self) {
         if self.locked.get_or_default() {
-            self.env().revert("No reentrancy");
+            self.env().revert(Error::NoReentrancy);
         }
 
         self.locked.set(true);
@@ -778,6 +778,11 @@ impl FunctionModifier {
     }
 }
 
+#[odra::odra_error]
+pub enum Error {
+    NoReentrancy = 1,
+}
+
 ```
 </TabItem>