Commit 84e370d0b2ce2240356ebd0a6007aec8620dfa10 | zig

author	kprotty <kbutcher6200@gmail.com> 2019-11-05 09:31:29-06:00
committer	kprotty <kbutcher6200@gmail.com> 2019-11-05 09:31:29-06:00
log	84e370d0b2ce2240356ebd0a6007aec8620dfa10
tree	4315b502adbf5c7ae2cbc153495a70f569cd43f9
parent	0d4f4aad9e44043b329de0f2e556c3b61cdaafa9

Move hybrid yielding into its own struct

1 files changed, 28 insertions(+), 16 deletions(-)

lib
- std
  - spinlock.zig

lib/std/spinlock.zig+28-16

 @@ -21,23 +21,10 @@ pub const SpinLock = struct {
         return SpinLock{ .lock = 0 };
     }
-    // Hybrid spinning from
-    // http://www.1024cores.net/home/lock-free-algorithms/tricks/spinning
     pub fn acquire(self: *SpinLock) Held {
-        var backoff: usize = 0;
+        var backoff = Backoff.init();
-        while (@atomicRmw(u8, &self.lock, .Xchg, 1, .Acquire) != 0) : (backoff +%= 1) {
+        while (@atomicRmw(u8, &self.lock, .Xchg, 1, .Acquire) != 0)
-            if (backoff < 10) {
+            backoff.yield();
-                yieldCpu();
-            } else if (backoff < 20) {
-                for (([30]void)(undefined)) |_| yieldCpu();
-            } else if (backoff < 24) {
-                yieldThread();
-            } else if (backoff < 26) {
-                time.sleep(1 * time.millisecond);
-            } else {
-                time.sleep(10 * time.millisecond);
-            }
-        }
         return Held{ .spinlock = self };
     }
 @@ -56,6 +43,31 @@ pub const SpinLock = struct {
             else => time.sleep(1 * time.microsecond),
         }
     }
+    const Backoff = struct {
+        iteration: usize,
+        fn init() @This() {
+            return @This(){ .iteration = 0 };
+        }
+        // Hybrid yielding from
+        // http://www.1024cores.net/home/lock-free-algorithms/tricks/spinning
+        fn yield(self: *@This()) void {
+            defer self.iteration +%= 1;
+            if (self.iteration < 10) {
+                yieldCpu();
+            } else if (self.iteration < 20) {
+                for (([30]void)(undefined)) |_| yieldCpu();
+            } else if (self.iteration < 24) {
+                yieldThread();
+            } else if (self.iteration < 26) {
+                time.sleep(1 * time.millisecond);
+            } else {
+                time.sleep(10 * time.millisecond);
+            }
+        }
+    };
 };
 test "spinlock" {