Optimize saturating_add_signed

rust-lang · Oct 4, 2021 · 47edde1 · 47edde1
1 parent 4846fd9
commit 47edde1
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/library/core/src/num/uint_macros.rs b/library/core/src/num/uint_macros.rs
@@ -1037,10 +1037,13 @@ macro_rules! uint_impl {
                       without modifying the original"]
         #[inline]
         pub const fn saturating_add_signed(self, rhs: $SignedT) -> Self {
-            if rhs >= 0 {
-                self.saturating_add(rhs as Self)
+            let (res, overflow) = self.overflowing_add(rhs as Self);
+            if overflow == (rhs < 0) {
+                res
+            } else if overflow {
+                Self::MAX
             } else {
-                self.saturating_sub(rhs.unsigned_abs())
+                0
             }
         }