metrics: add fetch_add for AtomicU64 (#4453)

tokio-rs · Jan 31, 2022 · 77468ae · 77468ae
1 parent 2cee1db
commit 77468ae
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tokio/src/loom/std/atomic_u64.rs b/tokio/src/loom/std/atomic_u64.rs
@@ -34,6 +34,13 @@ cfg_not_has_atomic_u64! {
             *self.inner.lock() = val;
         }
 
+        pub(crate) fn fetch_add(&self, val: u64, _: Ordering) -> u64 {
+            let mut lock = self.inner.lock();
+            let prev = *lock;
+            *lock = prev + val;
+            prev
+        }
+
         pub(crate) fn fetch_or(&self, val: u64, _: Ordering) -> u64 {
             let mut lock = self.inner.lock();
             let prev = *lock;