diff options
author | cinap_lenrek <cinap_lenrek@localhost> | 2011-05-03 11:25:13 +0000 |
---|---|---|
committer | cinap_lenrek <cinap_lenrek@localhost> | 2011-05-03 11:25:13 +0000 |
commit | 458120dd40db6b4df55a4e96b650e16798ef06a0 (patch) | |
tree | 8f82685be24fef97e715c6f5ca4c68d34d5074ee /sys/lib/python/mutex.py | |
parent | 3a742c699f6806c1145aea5149bf15de15a0afd7 (diff) |
add hg and python
Diffstat (limited to 'sys/lib/python/mutex.py')
-rw-r--r-- | sys/lib/python/mutex.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/sys/lib/python/mutex.py b/sys/lib/python/mutex.py new file mode 100644 index 000000000..5d35bdf6a --- /dev/null +++ b/sys/lib/python/mutex.py @@ -0,0 +1,52 @@ +"""Mutual exclusion -- for use with module sched + +A mutex has two pieces of state -- a 'locked' bit and a queue. +When the mutex is not locked, the queue is empty. +Otherwise, the queue contains 0 or more (function, argument) pairs +representing functions (or methods) waiting to acquire the lock. +When the mutex is unlocked while the queue is not empty, +the first queue entry is removed and its function(argument) pair called, +implying it now has the lock. + +Of course, no multi-threading is implied -- hence the funny interface +for lock, where a function is called once the lock is aquired. +""" + +from collections import deque + +class mutex: + def __init__(self): + """Create a new mutex -- initially unlocked.""" + self.locked = 0 + self.queue = deque() + + def test(self): + """Test the locked bit of the mutex.""" + return self.locked + + def testandset(self): + """Atomic test-and-set -- grab the lock if it is not set, + return True if it succeeded.""" + if not self.locked: + self.locked = 1 + return True + else: + return False + + def lock(self, function, argument): + """Lock a mutex, call the function with supplied argument + when it is acquired. If the mutex is already locked, place + function and argument in the queue.""" + if self.testandset(): + function(argument) + else: + self.queue.append((function, argument)) + + def unlock(self): + """Unlock a mutex. If the queue is not empty, call the next + function with its argument.""" + if self.queue: + function, argument = self.queue.popleft() + function(argument) + else: + self.locked = 0 |