diff options
Diffstat (limited to 'lib/std/atomic/int.zig')
| -rw-r--r-- | lib/std/atomic/int.zig | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/lib/std/atomic/int.zig b/lib/std/atomic/int.zig new file mode 100644 index 0000000000..94985b914f --- /dev/null +++ b/lib/std/atomic/int.zig @@ -0,0 +1,41 @@ +const builtin = @import("builtin"); +const AtomicOrder = builtin.AtomicOrder; + +/// Thread-safe, lock-free integer +pub fn Int(comptime T: type) type { + return struct { + unprotected_value: T, + + pub const Self = @This(); + + pub fn init(init_val: T) Self { + return Self{ .unprotected_value = init_val }; + } + + /// Returns previous value + pub fn incr(self: *Self) T { + return @atomicRmw(T, &self.unprotected_value, builtin.AtomicRmwOp.Add, 1, AtomicOrder.SeqCst); + } + + /// Returns previous value + pub fn decr(self: *Self) T { + return @atomicRmw(T, &self.unprotected_value, builtin.AtomicRmwOp.Sub, 1, AtomicOrder.SeqCst); + } + + pub fn get(self: *Self) T { + return @atomicLoad(T, &self.unprotected_value, AtomicOrder.SeqCst); + } + + pub fn set(self: *Self, new_value: T) void { + _ = self.xchg(new_value); + } + + pub fn xchg(self: *Self, new_value: T) T { + return @atomicRmw(T, &self.unprotected_value, builtin.AtomicRmwOp.Xchg, new_value, AtomicOrder.SeqCst); + } + + pub fn fetchAdd(self: *Self, op: T) T { + return @atomicRmw(T, &self.unprotected_value, builtin.AtomicRmwOp.Add, op, AtomicOrder.SeqCst); + } + }; +} |
