diff options
| -rw-r--r-- | Lib/test/test_thread.py | 69 |
1 files changed, 67 insertions, 2 deletions
diff --git a/Lib/test/test_thread.py b/Lib/test/test_thread.py index 5bc3076..4e0eb70 100644 --- a/Lib/test/test_thread.py +++ b/Lib/test/test_thread.py @@ -7,13 +7,18 @@ import thread import time mutex = thread.allocate_lock() +whmutex = thread.allocate_lock() # for calls to whrandom running = 0 done = thread.allocate_lock() done.acquire() +numtasks = 10 + def task(ident): global running - delay = whrandom.random() * 10 + whmutex.acquire() + delay = whrandom.random() * numtasks + whmutex.release() print 'task', ident, 'will run for', delay, 'sec' time.sleep(delay) print 'task', ident, 'done' @@ -33,9 +38,69 @@ def newtask(): running = running + 1 mutex.release() -for i in range(10): +for i in range(numtasks): newtask() print 'waiting for all tasks to complete' done.acquire() print 'all tasks done' + +class barrier: + def __init__(self, n): + self.n = n + self.waiting = 0 + self.checkin = thread.allocate_lock() + self.checkout = thread.allocate_lock() + self.checkout.acquire() + + def enter(self): + checkin, checkout = self.checkin, self.checkout + + checkin.acquire() + self.waiting = self.waiting + 1 + if self.waiting == self.n: + self.waiting = self.n - 1 + checkout.release() + return + checkin.release() + + checkout.acquire() + self.waiting = self.waiting - 1 + if self.waiting == 0: + checkin.release() + return + checkout.release() + +numtrips = 3 +def task2(ident): + global running + for i in range(numtrips): + if ident == 0: + # give it a good chance to enter the next + # barrier before the others are all out + # of the current one + delay = 0.001 + else: + whmutex.acquire() + delay = whrandom.random() * numtasks + whmutex.release() + print 'task', ident, 'will run for', delay, 'sec' + time.sleep(delay) + print 'task', ident, 'entering barrier', i + bar.enter() + print 'task', ident, 'leaving barrier', i + mutex.acquire() + running = running - 1 + if running == 0: + done.release() + mutex.release() + +print '\n*** Barrier Test ***' +if done.acquire(0): + raise ValueError, "'done' should have remained acquired" +bar = barrier(numtasks) +running = numtasks +for i in range(numtasks): + thread.start_new_thread(task2, (i,)) +done.acquire() +print 'all tasks done' |