Use checked_add and checked_sub in Add and Sub impls

chronotope · Jun 8, 2023 · 965ab29 · 965ab29
1 parent e1cb286
commit 965ab29
Showing 1 changed file with 2 additions and 14 deletions.
diff --git a/src/duration.rs b/src/duration.rs
@@ -368,27 +368,15 @@ impl Add for Duration {
     type Output = Duration;
 
     fn add(self, rhs: Duration) -> Duration {
-        let mut secs = self.secs + rhs.secs;
-        let mut nanos = self.nanos + rhs.nanos;
-        if nanos >= NANOS_PER_SEC {
-            nanos -= NANOS_PER_SEC;
-            secs += 1;
-        }
-        Duration { secs, nanos }
+        self.checked_add(&rhs).expect("`Duration + Duration` overflowed")
     }
 }
 
 impl Sub for Duration {
     type Output = Duration;
 
     fn sub(self, rhs: Duration) -> Duration {
-        let mut secs = self.secs - rhs.secs;
-        let mut nanos = self.nanos - rhs.nanos;
-        if nanos < 0 {
-            nanos += NANOS_PER_SEC;
-            secs -= 1;
-        }
-        Duration { secs, nanos }
+        self.checked_sub(&rhs).expect("`Duration - Duration` overflowed")
     }
 }