mirror of
https://github.com/python/cpython.git
synced 2025-07-30 06:34:15 +00:00
28eeefe566
46640 Patch #1454481: Make thread stack size runtime tunable. 46647 Markup fix The first is causing many buildbots to fail test runs, and there are multiple causes with seemingly no immediate prospects for repairing them. See python-dev discussion. Note that a branch can (and should) be created for resolving these problems, like svn copy svn+ssh://svn.python.org/python/trunk -r46640 svn+ssh://svn.python.org/python/branches/NEW_BRANCH followed by merging rev 46647 to the new branch.
117 lines
3 KiB
Python
117 lines
3 KiB
Python
# Very rudimentary test of thread module
|
|
|
|
# Create a bunch of threads, let each do some work, wait until all are done
|
|
|
|
from test.test_support import verbose
|
|
import random
|
|
import thread
|
|
import time
|
|
|
|
mutex = thread.allocate_lock()
|
|
rmutex = thread.allocate_lock() # for calls to random
|
|
running = 0
|
|
done = thread.allocate_lock()
|
|
done.acquire()
|
|
|
|
numtasks = 10
|
|
|
|
def task(ident):
|
|
global running
|
|
rmutex.acquire()
|
|
delay = random.random() * numtasks
|
|
rmutex.release()
|
|
if verbose:
|
|
print 'task', ident, 'will run for', round(delay, 1), 'sec'
|
|
time.sleep(delay)
|
|
if verbose:
|
|
print 'task', ident, 'done'
|
|
mutex.acquire()
|
|
running = running - 1
|
|
if running == 0:
|
|
done.release()
|
|
mutex.release()
|
|
|
|
next_ident = 0
|
|
def newtask():
|
|
global next_ident, running
|
|
mutex.acquire()
|
|
next_ident = next_ident + 1
|
|
if verbose:
|
|
print 'creating task', next_ident
|
|
thread.start_new_thread(task, (next_ident,))
|
|
running = running + 1
|
|
mutex.release()
|
|
|
|
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:
|
|
rmutex.acquire()
|
|
delay = random.random() * numtasks
|
|
rmutex.release()
|
|
if verbose:
|
|
print 'task', ident, 'will run for', round(delay, 1), 'sec'
|
|
time.sleep(delay)
|
|
if verbose:
|
|
print 'task', ident, 'entering barrier', i
|
|
bar.enter()
|
|
if verbose:
|
|
print 'task', ident, 'leaving barrier', i
|
|
mutex.acquire()
|
|
running -= 1
|
|
# Must release mutex before releasing done, else the main thread can
|
|
# exit and set mutex to None as part of global teardown; then
|
|
# mutex.release() raises AttributeError.
|
|
finished = running == 0
|
|
mutex.release()
|
|
if finished:
|
|
done.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'
|