add Sum<&U128> (#93)

* add Sum<&U128> * format
ontio · Sep 13, 2020 · 0374c21 · 0374c21
1 parent 44c1285
commit 0374c21
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/ontio-std/src/types/num.rs b/ontio-std/src/types/num.rs
@@ -60,6 +60,21 @@ impl Sum for U128 {
     }
 }
 
+impl<'a> Sum<&'a U128> for U128 {
+    fn sum<I: Iterator<Item = &'a Self>>(iter: I) -> Self {
+        iter.fold(U128::new(0), Add::add)
+    }
+}
+
+impl<'a> Add<&'a U128> for U128 {
+    type Output = U128;
+
+    #[track_caller]
+    fn add(self, rhs: &'a U128) -> Self::Output {
+        U128(self.0.checked_add(rhs.0).unwrap())
+    }
+}
+
 impl Add<U128> for U128 {
     type Output = U128;