Update the example

informalsystems · Aug 21, 2023 · 99e35cb · 99e35cb
1 parent fbf4ab4
commit 99e35cb
Showing 1 changed file with 20 additions and 18 deletions.
diff --git a/doc/rfcs/rfc001-sum-types/rfc001-erc-sum-types-example.qnt b/doc/rfcs/rfc001-sum-types/rfc001-erc-sum-types-example.qnt
@@ -34,26 +34,28 @@ module erc20 {
 
     // Example of a sum type representing a
     // NOTE: Once we support parametric sum-types we can generalize this alias and the map operators
-    type Erc20Result = {
-         Ok(Erc20State),
-         Error(str),
-    }
-
-    pure def mapState(r:Erc20Result, f:Erc20State => Erc20State): Erc20Result = r match {
-         Ok(s)    => Ok(f(s)),
-         Error(_) => r
-    }
-
-    pure def flatMap(r:Erc20Result, f:Erc20State => Erc20Result): Erc20Result = r match {
-         Ok(s)    => f(s),
-         Error(_) => r
-    }
+    type Erc20Result =
+         | Ok(Erc20State)
+         | Error(str)
+
+    pure def mapState(r:Erc20Result, f:Erc20State => Erc20State): Erc20Result =
+         match r {
+            | Ok(s)    => Ok(f(s))
+            | Error(_) => r
+        }
+
+    pure def flatMap(r:Erc20Result, f:Erc20State => Erc20Result): Erc20Result =
+         match r {
+            | Ok(s)    => f(s)
+            | Error(_) => r
+        }
 
     // An auxilliary definition similar to Solidity's require
-    pure def require(r: Erc20Result, cond: bool, msg: str): Erc20State = match r {
-        Error(_) => s,
-        Ok(s)    => if (cond) Ok(s) else Error(msg)
-    }
+    pure def require(r: Erc20Result, cond: bool, msg: str): Erc20Result =
+        match r {
+            | Error(_) => r
+            | _        => if (cond) r else Error(msg)
+        }
 
     // contract initialization
     pure def newErc20(sender: Address, initialSupply: Uint): Erc20State = {