aboutsummaryrefslogtreecommitdiff
path: root/lib/std/atomic/int.zig
diff options
context:
space:
mode:
Diffstat (limited to 'lib/std/atomic/int.zig')
-rw-r--r--lib/std/atomic/int.zig41
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);
+ }
+ };
+}