| ... | ... | @@ -7,11 +7,28 @@ const testing = std.testing; |
| 7 | 7 | const SpinLock = std.SpinLock; |
| 8 | 8 | const ResetEvent = std.ResetEvent; |
| 9 | 9 | |
| 10 | | /// Lock may be held only once. If the same thread |
| 11 | | /// tries to acquire the same mutex twice, it deadlocks. |
| 12 | | /// This type supports static initialization and is at most `@sizeOf(usize)` in size. |
| 13 | | /// When an application is built in single threaded release mode, all the functions are |
| 14 | | /// no-ops. In single threaded debug mode, there is deadlock detection. |
| 10 | /// Lock may be held only once. If the same thread tries to acquire |
| 11 | /// the same mutex twice, it deadlocks. This type supports static |
| 12 | /// initialization and is at most `@sizeOf(usize)` in size. When an |
| 13 | /// application is built in single threaded release mode, all the |
| 14 | /// functions are no-ops. In single threaded debug mode, there is |
| 15 | /// deadlock detection. |
| 16 | /// |
| 17 | /// Example usage: |
| 18 | /// var m = Mutex.init(); |
| 19 | /// defer m.deinit(); |
| 20 | /// |
| 21 | /// const lock = m.acquire(); |
| 22 | /// defer lock.release(); |
| 23 | /// ... critical code |
| 24 | /// |
| 25 | /// Non-blocking: |
| 26 | /// if (m.tryAcquire) |lock| { |
| 27 | /// defer lock.release(); |
| 28 | /// // ... critical section |
| 29 | /// } else { |
| 30 | /// // ... lock not acquired |
| 31 | /// } |
| 15 | 32 | pub const Mutex = if (builtin.single_threaded) |
| 16 | 33 | struct { |
| 17 | 34 | lock: @TypeOf(lock_init), |
| ... | ... | @@ -28,14 +45,20 @@ pub const Mutex = if (builtin.single_threaded) |
| 28 | 45 | } |
| 29 | 46 | }; |
| 30 | 47 | |
| 48 | /// Create a new mutex in unlocked state. |
| 31 | 49 | pub fn init() Mutex { |
| 32 | 50 | return Mutex{ .lock = lock_init }; |
| 33 | 51 | } |
| 34 | 52 | |
| 53 | /// Free a mutex created with init. Calling this while the |
| 54 | /// mutex is held is illegal behavior. |
| 35 | 55 | pub fn deinit(self: *Mutex) void { |
| 36 | 56 | self.* = undefined; |
| 37 | 57 | } |
| 38 | 58 | |
| 59 | /// Try to acquire the mutex without blocking. Returns null if |
| 60 | /// the mutex is unavailable. Otherwise returns Held. Call |
| 61 | /// release on Held. |
| 39 | 62 | pub fn tryAcquire(self: *Mutex) ?Held { |
| 40 | 63 | if (std.debug.runtime_safety) { |
| 41 | 64 | if (self.lock) return null; |
| ... | ... | @@ -44,6 +67,8 @@ pub const Mutex = if (builtin.single_threaded) |
| 44 | 67 | return Held{ .mutex = self }; |
| 45 | 68 | } |
| 46 | 69 | |
| 70 | /// Acquire the mutex. Will deadlock if the mutex is already |
| 71 | /// held by the calling thread. |
| 47 | 72 | pub fn acquire(self: *Mutex) Held { |
| 48 | 73 | return self.tryAcquire() orelse @panic("deadlock detected"); |
| 49 | 74 | } |
| ... | ... | @@ -220,9 +245,12 @@ else if (builtin.link_libc or builtin.os == .linux) |
| 220 | 245 | } |
| 221 | 246 | } |
| 222 | 247 | |
| 248 | /// Returned when the lock is acquired. Call release to |
| 249 | /// release. |
| 223 | 250 | pub const Held = struct { |
| 224 | 251 | mutex: *Mutex, |
| 225 | 252 | |
| 253 | /// Release the held lock. |
| 226 | 254 | pub fn release(self: Held) void { |
| 227 | 255 | // first, remove the lock bit so another possibly parallel acquire() can succeed. |
| 228 | 256 | // use .Sub since it can be usually compiled down more efficiency |