Add saturating_add methods

aurora-is-near · Jan 9, 2024 · a85cc2a · a85cc2a
1 parent 84bc698
commit a85cc2a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/engine-types/src/types/balance.rs b/engine-types/src/types/balance.rs
@@ -91,6 +91,10 @@ impl Yocto {
     pub const fn as_u128(self) -> u128 {
         self.0
     }
+
+    pub const fn saturating_add(self, rhs: Yocto) -> Yocto {
+        Yocto(self.0.saturating_add(rhs.0))
+    }
 }
 
 impl Serialize for Yocto {

diff --git a/engine-types/src/types/gas.rs b/engine-types/src/types/gas.rs
@@ -49,6 +49,10 @@ impl NearGas {
     pub const fn as_u64(self) -> u64 {
         self.0
     }
+
+    pub const fn saturating_add(self, rhs: NearGas) -> NearGas {
+        NearGas(self.0.saturating_add(rhs.0))
+    }
 }
 
 #[derive(