blob: 18c28b7ffa42e3f95194f94e8390f597682b5571 [file] [log] [blame]
Jeremy Hylton92bb6e72002-08-14 19:25:42 +00001"""Thread module emulating a subset of Java's threading model."""
Guido van Rossum7f5013a1998-04-09 22:01:42 +00002
Fred Drakea8725952002-12-30 23:32:50 +00003import sys as _sys
4
5try:
6 import thread
7except ImportError:
8 del _sys.modules[__name__]
9 raise
10
Jeffrey Yasskin105f3d42008-03-31 00:35:53 +000011import warnings
Benjamin Petersonf4395602008-06-11 17:50:00 +000012
13from functools import wraps
Fred Drakea8725952002-12-30 23:32:50 +000014from time import time as _time, sleep as _sleep
Neil Schemenauerf607fc52003-11-05 23:03:00 +000015from traceback import format_exc as _format_exc
Raymond Hettinger756b3f32004-01-29 06:37:52 +000016from collections import deque
Guido van Rossum7f5013a1998-04-09 22:01:42 +000017
Benjamin Peterson973e6c22008-09-01 23:12:58 +000018# Note regarding PEP 8 compliant aliases
19# This threading model was originally inspired by Java, and inherited
20# the convention of camelCase function and method names from that
21# language. While those names are not in any imminent danger of being
22# deprecated, starting with Python 2.6, the module now provides a
23# PEP 8 compliant alias for any such method name.
24# Using the new PEP 8 compliant names also facilitates substitution
25# with the multiprocessing module, which doesn't provide the old
26# Java inspired names.
27
28
Guido van Rossum7f5013a1998-04-09 22:01:42 +000029# Rename some stuff so "from threading import *" is safe
Benjamin Peterson13f73822008-06-11 18:02:31 +000030__all__ = ['activeCount', 'active_count', 'Condition', 'currentThread',
31 'current_thread', 'enumerate', 'Event',
Tim Peters685e6972003-06-29 16:50:06 +000032 'Lock', 'RLock', 'Semaphore', 'BoundedSemaphore', 'Thread',
Andrew MacIntyre92913322006-06-13 15:04:24 +000033 'Timer', 'setprofile', 'settrace', 'local', 'stack_size']
Guido van Rossum7f5013a1998-04-09 22:01:42 +000034
Guido van Rossum7f5013a1998-04-09 22:01:42 +000035_start_new_thread = thread.start_new_thread
36_allocate_lock = thread.allocate_lock
37_get_ident = thread.get_ident
Jeremy Hyltonb5fc7492000-06-01 01:17:17 +000038ThreadError = thread.error
Guido van Rossum7f5013a1998-04-09 22:01:42 +000039del thread
40
Guido van Rossum7f5013a1998-04-09 22:01:42 +000041
Jeffrey Yasskin105f3d42008-03-31 00:35:53 +000042# sys.exc_clear is used to work around the fact that except blocks
43# don't fully clear the exception until 3.0.
44warnings.filterwarnings('ignore', category=DeprecationWarning,
45 module='threading', message='sys.exc_clear')
46
Tim Peters59aba122003-07-01 20:01:55 +000047# Debug support (adapted from ihooks.py).
48# All the major classes here derive from _Verbose. We force that to
49# be a new-style class so that all the major classes here are new-style.
50# This helps debugging (type(instance) is more revealing for instances
51# of new-style classes).
Guido van Rossum7f5013a1998-04-09 22:01:42 +000052
Tim Peters0939fac2003-07-01 19:28:44 +000053_VERBOSE = False
Guido van Rossum7f5013a1998-04-09 22:01:42 +000054
55if __debug__:
56
Tim Peters59aba122003-07-01 20:01:55 +000057 class _Verbose(object):
Guido van Rossum7f5013a1998-04-09 22:01:42 +000058
59 def __init__(self, verbose=None):
60 if verbose is None:
61 verbose = _VERBOSE
62 self.__verbose = verbose
63
64 def _note(self, format, *args):
65 if self.__verbose:
66 format = format % args
67 format = "%s: %s\n" % (
Benjamin Petersonb6a95562008-08-22 20:43:48 +000068 current_thread().name, format)
Guido van Rossum7f5013a1998-04-09 22:01:42 +000069 _sys.stderr.write(format)
70
71else:
72 # Disable this when using "python -O"
Tim Peters59aba122003-07-01 20:01:55 +000073 class _Verbose(object):
Guido van Rossum7f5013a1998-04-09 22:01:42 +000074 def __init__(self, verbose=None):
75 pass
76 def _note(self, *args):
77 pass
78
Jeremy Hyltonbfccb352003-06-29 16:58:41 +000079# Support for profile and trace hooks
80
81_profile_hook = None
82_trace_hook = None
83
84def setprofile(func):
85 global _profile_hook
86 _profile_hook = func
Tim Petersd1b108b2003-06-29 17:24:17 +000087
Jeremy Hyltonbfccb352003-06-29 16:58:41 +000088def settrace(func):
89 global _trace_hook
90 _trace_hook = func
Guido van Rossum7f5013a1998-04-09 22:01:42 +000091
92# Synchronization classes
93
94Lock = _allocate_lock
95
96def RLock(*args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +000097 return _RLock(*args, **kwargs)
Guido van Rossum7f5013a1998-04-09 22:01:42 +000098
99class _RLock(_Verbose):
Tim Petersb90f89a2001-01-15 03:26:36 +0000100
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000101 def __init__(self, verbose=None):
102 _Verbose.__init__(self, verbose)
103 self.__block = _allocate_lock()
104 self.__owner = None
105 self.__count = 0
106
107 def __repr__(self):
Nick Coghlanf8bbaa92007-07-31 13:38:01 +0000108 owner = self.__owner
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000109 return "<%s(%s, %d)>" % (
110 self.__class__.__name__,
Benjamin Petersonb6a95562008-08-22 20:43:48 +0000111 owner and owner.name,
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000112 self.__count)
113
114 def acquire(self, blocking=1):
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000115 me = current_thread()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000116 if self.__owner is me:
117 self.__count = self.__count + 1
118 if __debug__:
119 self._note("%s.acquire(%s): recursive success", self, blocking)
120 return 1
121 rc = self.__block.acquire(blocking)
122 if rc:
123 self.__owner = me
124 self.__count = 1
125 if __debug__:
Brett Cannon90cece72005-01-27 22:48:30 +0000126 self._note("%s.acquire(%s): initial success", self, blocking)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000127 else:
128 if __debug__:
129 self._note("%s.acquire(%s): failure", self, blocking)
130 return rc
131
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000132 __enter__ = acquire
133
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000134 def release(self):
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000135 if self.__owner is not current_thread():
Georg Brandlf004d9d2009-10-27 15:39:53 +0000136 raise RuntimeError("cannot release un-acquired lock")
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000137 self.__count = count = self.__count - 1
138 if not count:
139 self.__owner = None
140 self.__block.release()
141 if __debug__:
142 self._note("%s.release(): final release", self)
143 else:
144 if __debug__:
145 self._note("%s.release(): non-final release", self)
146
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000147 def __exit__(self, t, v, tb):
148 self.release()
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000149
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000150 # Internal methods used by condition variables
151
Brett Cannon20050502008-08-02 03:13:46 +0000152 def _acquire_restore(self, count_owner):
153 count, owner = count_owner
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000154 self.__block.acquire()
155 self.__count = count
156 self.__owner = owner
157 if __debug__:
158 self._note("%s._acquire_restore()", self)
159
160 def _release_save(self):
161 if __debug__:
162 self._note("%s._release_save()", self)
163 count = self.__count
164 self.__count = 0
165 owner = self.__owner
166 self.__owner = None
167 self.__block.release()
168 return (count, owner)
169
170 def _is_owned(self):
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000171 return self.__owner is current_thread()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000172
173
174def Condition(*args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +0000175 return _Condition(*args, **kwargs)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000176
177class _Condition(_Verbose):
178
179 def __init__(self, lock=None, verbose=None):
180 _Verbose.__init__(self, verbose)
181 if lock is None:
182 lock = RLock()
183 self.__lock = lock
184 # Export the lock's acquire() and release() methods
185 self.acquire = lock.acquire
186 self.release = lock.release
187 # If the lock defines _release_save() and/or _acquire_restore(),
188 # these override the default implementations (which just call
189 # release() and acquire() on the lock). Ditto for _is_owned().
190 try:
191 self._release_save = lock._release_save
192 except AttributeError:
193 pass
194 try:
195 self._acquire_restore = lock._acquire_restore
196 except AttributeError:
197 pass
198 try:
199 self._is_owned = lock._is_owned
200 except AttributeError:
201 pass
202 self.__waiters = []
203
Guido van Rossumda5b7012006-05-02 19:47:52 +0000204 def __enter__(self):
205 return self.__lock.__enter__()
206
207 def __exit__(self, *args):
208 return self.__lock.__exit__(*args)
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000209
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000210 def __repr__(self):
211 return "<Condition(%s, %d)>" % (self.__lock, len(self.__waiters))
212
213 def _release_save(self):
214 self.__lock.release() # No state to save
215
216 def _acquire_restore(self, x):
217 self.__lock.acquire() # Ignore saved state
218
219 def _is_owned(self):
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000220 # Return True if lock is owned by current_thread.
Jeremy Hyltonaf7fde72002-08-14 17:43:59 +0000221 # This method is called only if __lock doesn't have _is_owned().
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000222 if self.__lock.acquire(0):
223 self.__lock.release()
Tim Petersbc0e9102002-04-04 22:55:58 +0000224 return False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000225 else:
Tim Petersbc0e9102002-04-04 22:55:58 +0000226 return True
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000227
228 def wait(self, timeout=None):
Collin Winter50b79ce2007-06-06 00:17:35 +0000229 if not self._is_owned():
Georg Brandlf004d9d2009-10-27 15:39:53 +0000230 raise RuntimeError("cannot wait on un-acquired lock")
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000231 waiter = _allocate_lock()
232 waiter.acquire()
233 self.__waiters.append(waiter)
234 saved_state = self._release_save()
Tim Petersc951bf92001-04-02 20:15:57 +0000235 try: # restore state no matter what (e.g., KeyboardInterrupt)
236 if timeout is None:
237 waiter.acquire()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000238 if __debug__:
Tim Petersc951bf92001-04-02 20:15:57 +0000239 self._note("%s.wait(): got it", self)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000240 else:
Tim Petersa6a4f272001-08-12 00:41:33 +0000241 # Balancing act: We can't afford a pure busy loop, so we
242 # have to sleep; but if we sleep the whole timeout time,
243 # we'll be unresponsive. The scheme here sleeps very
244 # little at first, longer as time goes on, but never longer
245 # than 20 times per second (or the timeout time remaining).
Tim Petersc951bf92001-04-02 20:15:57 +0000246 endtime = _time() + timeout
Tim Petersa6a4f272001-08-12 00:41:33 +0000247 delay = 0.0005 # 500 us -> initial delay of 1 ms
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000248 while True:
Tim Petersc951bf92001-04-02 20:15:57 +0000249 gotit = waiter.acquire(0)
Tim Petersa6a4f272001-08-12 00:41:33 +0000250 if gotit:
Tim Petersc951bf92001-04-02 20:15:57 +0000251 break
Tim Petersa6a4f272001-08-12 00:41:33 +0000252 remaining = endtime - _time()
253 if remaining <= 0:
254 break
255 delay = min(delay * 2, remaining, .05)
Tim Petersc951bf92001-04-02 20:15:57 +0000256 _sleep(delay)
Tim Petersc951bf92001-04-02 20:15:57 +0000257 if not gotit:
258 if __debug__:
259 self._note("%s.wait(%s): timed out", self, timeout)
260 try:
261 self.__waiters.remove(waiter)
262 except ValueError:
263 pass
264 else:
265 if __debug__:
266 self._note("%s.wait(%s): got it", self, timeout)
267 finally:
268 self._acquire_restore(saved_state)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000269
270 def notify(self, n=1):
Collin Winter50b79ce2007-06-06 00:17:35 +0000271 if not self._is_owned():
Georg Brandlf004d9d2009-10-27 15:39:53 +0000272 raise RuntimeError("cannot notify on un-acquired lock")
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000273 __waiters = self.__waiters
274 waiters = __waiters[:n]
275 if not waiters:
276 if __debug__:
277 self._note("%s.notify(): no waiters", self)
278 return
279 self._note("%s.notify(): notifying %d waiter%s", self, n,
280 n!=1 and "s" or "")
281 for waiter in waiters:
282 waiter.release()
283 try:
284 __waiters.remove(waiter)
285 except ValueError:
286 pass
287
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000288 def notifyAll(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000289 self.notify(len(self.__waiters))
290
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000291 notify_all = notifyAll
Benjamin Petersonf4395602008-06-11 17:50:00 +0000292
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000293
294def Semaphore(*args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +0000295 return _Semaphore(*args, **kwargs)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000296
297class _Semaphore(_Verbose):
298
Andrew M. Kuchling39d3bfc2000-02-29 00:10:24 +0000299 # After Tim Peters' semaphore class, but not quite the same (no maximum)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000300
301 def __init__(self, value=1, verbose=None):
Collin Winter50b79ce2007-06-06 00:17:35 +0000302 if value < 0:
303 raise ValueError("semaphore initial value must be >= 0")
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000304 _Verbose.__init__(self, verbose)
305 self.__cond = Condition(Lock())
306 self.__value = value
307
308 def acquire(self, blocking=1):
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000309 rc = False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000310 self.__cond.acquire()
311 while self.__value == 0:
312 if not blocking:
313 break
Skip Montanarob446fc72001-08-19 04:25:24 +0000314 if __debug__:
315 self._note("%s.acquire(%s): blocked waiting, value=%s",
316 self, blocking, self.__value)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000317 self.__cond.wait()
318 else:
319 self.__value = self.__value - 1
Skip Montanarob446fc72001-08-19 04:25:24 +0000320 if __debug__:
Skip Montanaroae8454a2001-08-19 05:53:47 +0000321 self._note("%s.acquire: success, value=%s",
322 self, self.__value)
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000323 rc = True
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000324 self.__cond.release()
325 return rc
326
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000327 __enter__ = acquire
328
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000329 def release(self):
330 self.__cond.acquire()
331 self.__value = self.__value + 1
Skip Montanarob446fc72001-08-19 04:25:24 +0000332 if __debug__:
Skip Montanaroae8454a2001-08-19 05:53:47 +0000333 self._note("%s.release: success, value=%s",
334 self, self.__value)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000335 self.__cond.notify()
336 self.__cond.release()
337
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000338 def __exit__(self, t, v, tb):
339 self.release()
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000340
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000341
Skip Montanaroe428bb72001-08-20 20:27:58 +0000342def BoundedSemaphore(*args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +0000343 return _BoundedSemaphore(*args, **kwargs)
Skip Montanaroe428bb72001-08-20 20:27:58 +0000344
345class _BoundedSemaphore(_Semaphore):
346 """Semaphore that checks that # releases is <= # acquires"""
347 def __init__(self, value=1, verbose=None):
348 _Semaphore.__init__(self, value, verbose)
349 self._initial_value = value
350
351 def release(self):
352 if self._Semaphore__value >= self._initial_value:
353 raise ValueError, "Semaphore released too many times"
354 return _Semaphore.release(self)
355
356
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000357def Event(*args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +0000358 return _Event(*args, **kwargs)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000359
360class _Event(_Verbose):
361
362 # After Tim Peters' event class (without is_posted())
363
364 def __init__(self, verbose=None):
365 _Verbose.__init__(self, verbose)
366 self.__cond = Condition(Lock())
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000367 self.__flag = False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000368
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000369 def isSet(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000370 return self.__flag
371
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000372 is_set = isSet
Benjamin Petersonf4395602008-06-11 17:50:00 +0000373
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000374 def set(self):
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000375 self.__cond.acquire()
376 try:
Guido van Rossum21b60142002-11-21 21:08:39 +0000377 self.__flag = True
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000378 self.__cond.notify_all()
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000379 finally:
380 self.__cond.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000381
382 def clear(self):
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000383 self.__cond.acquire()
384 try:
Guido van Rossum21b60142002-11-21 21:08:39 +0000385 self.__flag = False
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000386 finally:
387 self.__cond.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000388
389 def wait(self, timeout=None):
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000390 self.__cond.acquire()
391 try:
Guido van Rossum21b60142002-11-21 21:08:39 +0000392 if not self.__flag:
393 self.__cond.wait(timeout)
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000394 finally:
395 self.__cond.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000396
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000397# Helper to generate new thread names
398_counter = 0
399def _newname(template="Thread-%d"):
400 global _counter
401 _counter = _counter + 1
402 return template % _counter
403
404# Active thread administration
405_active_limbo_lock = _allocate_lock()
Tim Peters711906e2005-01-08 07:30:42 +0000406_active = {} # maps thread id to Thread object
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000407_limbo = {}
408
409
410# Main class for threads
411
412class Thread(_Verbose):
413
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000414 __initialized = False
Brett Cannoncc4e9352004-07-03 03:52:35 +0000415 # Need to store a reference to sys.exc_info for printing
416 # out exceptions when a thread tries to use a global var. during interp.
417 # shutdown and thus raises an exception about trying to perform some
418 # operation on/with a NoneType
419 __exc_info = _sys.exc_info
Jeffrey Yasskin8b9091f2008-03-28 04:11:18 +0000420 # Keep sys.exc_clear too to clear the exception just before
421 # allowing .join() to return.
422 __exc_clear = _sys.exc_clear
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000423
424 def __init__(self, group=None, target=None, name=None,
Georg Brandla4a8b822005-07-15 09:13:21 +0000425 args=(), kwargs=None, verbose=None):
Guido van Rossum5a43e1a1998-06-09 19:04:26 +0000426 assert group is None, "group argument must be None for now"
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000427 _Verbose.__init__(self, verbose)
Georg Brandla4a8b822005-07-15 09:13:21 +0000428 if kwargs is None:
429 kwargs = {}
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000430 self.__target = target
431 self.__name = str(name or _newname())
432 self.__args = args
433 self.__kwargs = kwargs
434 self.__daemonic = self._set_daemon()
Gregory P. Smith8856dda2008-06-01 23:48:47 +0000435 self.__ident = None
Jeffrey Yasskin69e13092008-02-28 06:09:19 +0000436 self.__started = Event()
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000437 self.__stopped = False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000438 self.__block = Condition(Lock())
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000439 self.__initialized = True
Brett Cannoncc4e9352004-07-03 03:52:35 +0000440 # sys.stderr is not stored in the class like
441 # sys.exc_info since it can be changed between instances
442 self.__stderr = _sys.stderr
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000443
444 def _set_daemon(self):
445 # Overridden in _MainThread and _DummyThread
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000446 return current_thread().daemon
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000447
448 def __repr__(self):
449 assert self.__initialized, "Thread.__init__() was not called"
450 status = "initial"
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000451 if self.__started.is_set():
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000452 status = "started"
453 if self.__stopped:
454 status = "stopped"
455 if self.__daemonic:
Gregory P. Smith8856dda2008-06-01 23:48:47 +0000456 status += " daemon"
457 if self.__ident is not None:
458 status += " %s" % self.__ident
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000459 return "<%s(%s, %s)>" % (self.__class__.__name__, self.__name, status)
460
461 def start(self):
Collin Winter50b79ce2007-06-06 00:17:35 +0000462 if not self.__initialized:
463 raise RuntimeError("thread.__init__() not called")
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000464 if self.__started.is_set():
Collin Winter50b79ce2007-06-06 00:17:35 +0000465 raise RuntimeError("thread already started")
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000466 if __debug__:
467 self._note("%s.start(): starting thread", self)
468 _active_limbo_lock.acquire()
469 _limbo[self] = self
470 _active_limbo_lock.release()
471 _start_new_thread(self.__bootstrap, ())
Jeffrey Yasskin69e13092008-02-28 06:09:19 +0000472 self.__started.wait()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000473
474 def run(self):
Jeffrey Yasskina885c152008-02-23 20:40:35 +0000475 try:
476 if self.__target:
477 self.__target(*self.__args, **self.__kwargs)
478 finally:
479 # Avoid a refcycle if the thread is running a function with
480 # an argument that has a member that points to the thread.
481 del self.__target, self.__args, self.__kwargs
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000482
483 def __bootstrap(self):
Guido van Rossum54ec61e2007-08-20 15:18:04 +0000484 # Wrapper around the real bootstrap code that ignores
485 # exceptions during interpreter cleanup. Those typically
486 # happen when a daemon thread wakes up at an unfortunate
487 # moment, finds the world around it destroyed, and raises some
488 # random exception *** while trying to report the exception in
489 # __bootstrap_inner() below ***. Those random exceptions
490 # don't help anybody, and they confuse users, so we suppress
491 # them. We suppress them only when it appears that the world
492 # indeed has already been destroyed, so that exceptions in
493 # __bootstrap_inner() during normal business hours are properly
494 # reported. Also, we only suppress them for daemonic threads;
495 # if a non-daemonic encounters this, something else is wrong.
496 try:
497 self.__bootstrap_inner()
498 except:
499 if self.__daemonic and _sys is None:
500 return
501 raise
502
Benjamin Peterson61611f82009-03-31 21:40:18 +0000503 def _set_ident(self):
504 self.__ident = _get_ident()
505
Guido van Rossum54ec61e2007-08-20 15:18:04 +0000506 def __bootstrap_inner(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000507 try:
Benjamin Peterson61611f82009-03-31 21:40:18 +0000508 self._set_ident()
Jeffrey Yasskin69e13092008-02-28 06:09:19 +0000509 self.__started.set()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000510 _active_limbo_lock.acquire()
Gregory P. Smith8856dda2008-06-01 23:48:47 +0000511 _active[self.__ident] = self
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000512 del _limbo[self]
513 _active_limbo_lock.release()
514 if __debug__:
515 self._note("%s.__bootstrap(): thread started", self)
Jeremy Hyltonbfccb352003-06-29 16:58:41 +0000516
517 if _trace_hook:
518 self._note("%s.__bootstrap(): registering trace hook", self)
519 _sys.settrace(_trace_hook)
520 if _profile_hook:
521 self._note("%s.__bootstrap(): registering profile hook", self)
522 _sys.setprofile(_profile_hook)
Tim Petersd1b108b2003-06-29 17:24:17 +0000523
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000524 try:
525 self.run()
526 except SystemExit:
527 if __debug__:
528 self._note("%s.__bootstrap(): raised SystemExit", self)
529 except:
530 if __debug__:
531 self._note("%s.__bootstrap(): unhandled exception", self)
Brett Cannoncc4e9352004-07-03 03:52:35 +0000532 # If sys.stderr is no more (most likely from interpreter
533 # shutdown) use self.__stderr. Otherwise still use sys (as in
534 # _sys) in case sys.stderr was redefined since the creation of
535 # self.
536 if _sys:
537 _sys.stderr.write("Exception in thread %s:\n%s\n" %
Benjamin Petersonb6a95562008-08-22 20:43:48 +0000538 (self.name, _format_exc()))
Brett Cannoncc4e9352004-07-03 03:52:35 +0000539 else:
540 # Do the best job possible w/o a huge amt. of code to
541 # approximate a traceback (code ideas from
542 # Lib/traceback.py)
543 exc_type, exc_value, exc_tb = self.__exc_info()
544 try:
545 print>>self.__stderr, (
Benjamin Petersonb6a95562008-08-22 20:43:48 +0000546 "Exception in thread " + self.name +
Brett Cannoncc4e9352004-07-03 03:52:35 +0000547 " (most likely raised during interpreter shutdown):")
548 print>>self.__stderr, (
549 "Traceback (most recent call last):")
550 while exc_tb:
551 print>>self.__stderr, (
552 ' File "%s", line %s, in %s' %
553 (exc_tb.tb_frame.f_code.co_filename,
554 exc_tb.tb_lineno,
555 exc_tb.tb_frame.f_code.co_name))
556 exc_tb = exc_tb.tb_next
557 print>>self.__stderr, ("%s: %s" % (exc_type, exc_value))
558 # Make sure that exc_tb gets deleted since it is a memory
559 # hog; deleting everything else is just for thoroughness
560 finally:
561 del exc_type, exc_value, exc_tb
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000562 else:
563 if __debug__:
564 self._note("%s.__bootstrap(): normal return", self)
Jeffrey Yasskin8b9091f2008-03-28 04:11:18 +0000565 finally:
566 # Prevent a race in
567 # test_threading.test_no_refcycle_through_target when
568 # the exception keeps the target alive past when we
569 # assert that it's dead.
Amaury Forgeot d'Arc504a48f2008-03-29 01:41:08 +0000570 self.__exc_clear()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000571 finally:
Gregory P. Smith95cd5c02008-01-22 01:20:42 +0000572 with _active_limbo_lock:
573 self.__stop()
574 try:
575 # We don't call self.__delete() because it also
576 # grabs _active_limbo_lock.
577 del _active[_get_ident()]
578 except:
579 pass
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000580
581 def __stop(self):
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000582 self.__block.acquire()
583 self.__stopped = True
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000584 self.__block.notify_all()
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000585 self.__block.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000586
587 def __delete(self):
Tim Peters21429932004-07-21 03:36:52 +0000588 "Remove current thread from the dict of currently running threads."
Brett Cannon8b3d92a2004-07-21 02:21:58 +0000589
Tim Peters21429932004-07-21 03:36:52 +0000590 # Notes about running with dummy_thread:
591 #
592 # Must take care to not raise an exception if dummy_thread is being
593 # used (and thus this module is being used as an instance of
594 # dummy_threading). dummy_thread.get_ident() always returns -1 since
595 # there is only one thread if dummy_thread is being used. Thus
596 # len(_active) is always <= 1 here, and any Thread instance created
597 # overwrites the (if any) thread currently registered in _active.
598 #
599 # An instance of _MainThread is always created by 'threading'. This
600 # gets overwritten the instant an instance of Thread is created; both
601 # threads return -1 from dummy_thread.get_ident() and thus have the
602 # same key in the dict. So when the _MainThread instance created by
603 # 'threading' tries to clean itself up when atexit calls this method
604 # it gets a KeyError if another Thread instance was created.
605 #
606 # This all means that KeyError from trying to delete something from
607 # _active if dummy_threading is being used is a red herring. But
608 # since it isn't if dummy_threading is *not* being used then don't
609 # hide the exception.
Brett Cannon8b3d92a2004-07-21 02:21:58 +0000610
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000611 try:
Amaury Forgeot d'Arcd7a26512008-04-03 23:07:55 +0000612 with _active_limbo_lock:
Brett Cannon8b3d92a2004-07-21 02:21:58 +0000613 del _active[_get_ident()]
Amaury Forgeot d'Arcd7a26512008-04-03 23:07:55 +0000614 # There must not be any python code between the previous line
615 # and after the lock is released. Otherwise a tracing function
616 # could try to acquire the lock again in the same thread, (in
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000617 # current_thread()), and would block.
Amaury Forgeot d'Arcd7a26512008-04-03 23:07:55 +0000618 except KeyError:
619 if 'dummy_threading' not in _sys.modules:
620 raise
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000621
622 def join(self, timeout=None):
Collin Winter50b79ce2007-06-06 00:17:35 +0000623 if not self.__initialized:
624 raise RuntimeError("Thread.__init__() not called")
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000625 if not self.__started.is_set():
Collin Winter50b79ce2007-06-06 00:17:35 +0000626 raise RuntimeError("cannot join thread before it is started")
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000627 if self is current_thread():
Collin Winter50b79ce2007-06-06 00:17:35 +0000628 raise RuntimeError("cannot join current thread")
629
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000630 if __debug__:
631 if not self.__stopped:
632 self._note("%s.join(): waiting until thread stops", self)
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000633 self.__block.acquire()
634 try:
Brett Cannonad07ff22005-11-23 02:15:50 +0000635 if timeout is None:
636 while not self.__stopped:
637 self.__block.wait()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000638 if __debug__:
639 self._note("%s.join(): thread stopped", self)
Brett Cannonad07ff22005-11-23 02:15:50 +0000640 else:
641 deadline = _time() + timeout
642 while not self.__stopped:
643 delay = deadline - _time()
644 if delay <= 0:
645 if __debug__:
646 self._note("%s.join(): timed out", self)
647 break
648 self.__block.wait(delay)
649 else:
650 if __debug__:
651 self._note("%s.join(): thread stopped", self)
Raymond Hettinger70ec29d2008-01-24 18:12:23 +0000652 finally:
653 self.__block.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000654
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000655 @property
656 def name(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000657 assert self.__initialized, "Thread.__init__() not called"
658 return self.__name
659
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000660 @name.setter
661 def name(self, name):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000662 assert self.__initialized, "Thread.__init__() not called"
663 self.__name = str(name)
664
Benjamin Petersond8a89722008-08-18 16:40:03 +0000665 @property
666 def ident(self):
Gregory P. Smith8856dda2008-06-01 23:48:47 +0000667 assert self.__initialized, "Thread.__init__() not called"
668 return self.__ident
669
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000670 def isAlive(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000671 assert self.__initialized, "Thread.__init__() not called"
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000672 return self.__started.is_set() and not self.__stopped
Tim Petersb90f89a2001-01-15 03:26:36 +0000673
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000674 is_alive = isAlive
Benjamin Peterson6ee1a312008-08-18 21:53:29 +0000675
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000676 @property
677 def daemon(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000678 assert self.__initialized, "Thread.__init__() not called"
679 return self.__daemonic
680
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000681 @daemon.setter
682 def daemon(self, daemonic):
Collin Winter50b79ce2007-06-06 00:17:35 +0000683 if not self.__initialized:
684 raise RuntimeError("Thread.__init__() not called")
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000685 if self.__started.is_set():
Collin Winter50b79ce2007-06-06 00:17:35 +0000686 raise RuntimeError("cannot set daemon status of active thread");
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000687 self.__daemonic = daemonic
688
Benjamin Petersond8106262008-08-18 18:13:17 +0000689 def isDaemon(self):
690 return self.daemon
691
692 def setDaemon(self, daemonic):
693 self.daemon = daemonic
694
695 def getName(self):
696 return self.name
697
698 def setName(self, name):
699 self.name = name
700
Martin v. Löwis44f86962001-09-05 13:44:54 +0000701# The timer class was contributed by Itamar Shtull-Trauring
702
703def Timer(*args, **kwargs):
704 return _Timer(*args, **kwargs)
705
706class _Timer(Thread):
707 """Call a function after a specified number of seconds:
Tim Petersb64bec32001-09-18 02:26:39 +0000708
Martin v. Löwis44f86962001-09-05 13:44:54 +0000709 t = Timer(30.0, f, args=[], kwargs={})
710 t.start()
711 t.cancel() # stop the timer's action if it's still waiting
712 """
Tim Petersb64bec32001-09-18 02:26:39 +0000713
Martin v. Löwis44f86962001-09-05 13:44:54 +0000714 def __init__(self, interval, function, args=[], kwargs={}):
715 Thread.__init__(self)
716 self.interval = interval
717 self.function = function
718 self.args = args
719 self.kwargs = kwargs
720 self.finished = Event()
Tim Petersb64bec32001-09-18 02:26:39 +0000721
Martin v. Löwis44f86962001-09-05 13:44:54 +0000722 def cancel(self):
723 """Stop the timer if it hasn't finished yet"""
724 self.finished.set()
Tim Petersb64bec32001-09-18 02:26:39 +0000725
Martin v. Löwis44f86962001-09-05 13:44:54 +0000726 def run(self):
727 self.finished.wait(self.interval)
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000728 if not self.finished.is_set():
Martin v. Löwis44f86962001-09-05 13:44:54 +0000729 self.function(*self.args, **self.kwargs)
730 self.finished.set()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000731
732# Special thread class to represent the main thread
733# This is garbage collected through an exit handler
734
735class _MainThread(Thread):
736
737 def __init__(self):
738 Thread.__init__(self, name="MainThread")
Jeffrey Yasskin69e13092008-02-28 06:09:19 +0000739 self._Thread__started.set()
Benjamin Peterson61611f82009-03-31 21:40:18 +0000740 self._set_ident()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000741 _active_limbo_lock.acquire()
742 _active[_get_ident()] = self
743 _active_limbo_lock.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000744
745 def _set_daemon(self):
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000746 return False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000747
Martin v. Löwis7b7c9d42007-01-04 21:06:12 +0000748 def _exitfunc(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000749 self._Thread__stop()
750 t = _pickSomeNonDaemonThread()
751 if t:
752 if __debug__:
753 self._note("%s: waiting for other threads", self)
754 while t:
755 t.join()
756 t = _pickSomeNonDaemonThread()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000757 if __debug__:
758 self._note("%s: exiting", self)
759 self._Thread__delete()
760
761def _pickSomeNonDaemonThread():
762 for t in enumerate():
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000763 if not t.daemon and t.is_alive():
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000764 return t
765 return None
766
767
768# Dummy thread class to represent threads not started here.
Tim Peters711906e2005-01-08 07:30:42 +0000769# These aren't garbage collected when they die, nor can they be waited for.
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000770# If they invoke anything in threading.py that calls current_thread(), they
Tim Peters711906e2005-01-08 07:30:42 +0000771# leave an entry in the _active dict forever after.
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000772# Their purpose is to return *something* from current_thread().
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000773# They are marked as daemon threads so we won't wait for them
774# when we exit (conform previous semantics).
775
776class _DummyThread(Thread):
Tim Petersb90f89a2001-01-15 03:26:36 +0000777
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000778 def __init__(self):
779 Thread.__init__(self, name=_newname("Dummy-%d"))
Tim Peters711906e2005-01-08 07:30:42 +0000780
781 # Thread.__block consumes an OS-level locking primitive, which
782 # can never be used by a _DummyThread. Since a _DummyThread
783 # instance is immortal, that's bad, so release this resource.
Brett Cannone6539c42005-01-08 02:43:53 +0000784 del self._Thread__block
Tim Peters711906e2005-01-08 07:30:42 +0000785
Jeffrey Yasskin69e13092008-02-28 06:09:19 +0000786 self._Thread__started.set()
Benjamin Peterson61611f82009-03-31 21:40:18 +0000787 self._set_ident()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000788 _active_limbo_lock.acquire()
789 _active[_get_ident()] = self
790 _active_limbo_lock.release()
791
792 def _set_daemon(self):
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000793 return True
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000794
Neal Norwitz45bec8c2002-02-19 03:01:36 +0000795 def join(self, timeout=None):
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000796 assert False, "cannot join a dummy thread"
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000797
798
799# Global API functions
800
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000801def currentThread():
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000802 try:
803 return _active[_get_ident()]
804 except KeyError:
Benjamin Peterson0fbcf692008-06-11 17:27:50 +0000805 ##print "current_thread(): no current thread for", _get_ident()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000806 return _DummyThread()
807
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000808current_thread = currentThread
Benjamin Petersonf4395602008-06-11 17:50:00 +0000809
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000810def activeCount():
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000811 _active_limbo_lock.acquire()
812 count = len(_active) + len(_limbo)
813 _active_limbo_lock.release()
814 return count
815
Benjamin Peterson973e6c22008-09-01 23:12:58 +0000816active_count = activeCount
Benjamin Petersonf4395602008-06-11 17:50:00 +0000817
Antoine Pitrou4fb961c2009-11-05 13:44:28 +0000818def _enumerate():
819 # Same as enumerate(), but without the lock. Internal use only.
820 return _active.values() + _limbo.values()
821
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000822def enumerate():
823 _active_limbo_lock.acquire()
824 active = _active.values() + _limbo.values()
825 _active_limbo_lock.release()
826 return active
827
Andrew MacIntyre92913322006-06-13 15:04:24 +0000828from thread import stack_size
829
Martin v. Löwis7b7c9d42007-01-04 21:06:12 +0000830# Create the main thread object,
831# and make it available for the interpreter
832# (Py_Main) as threading._shutdown.
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000833
Martin v. Löwis7b7c9d42007-01-04 21:06:12 +0000834_shutdown = _MainThread()._exitfunc
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000835
Jim Fultond15dc062004-07-14 19:11:50 +0000836# get thread-local implementation, either from the thread
837# module, or from the python fallback
838
839try:
840 from thread import _local as local
841except ImportError:
842 from _threading_local import local
843
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000844
Jesse Noller5e62ca42008-07-16 20:03:47 +0000845def _after_fork():
846 # This function is called by Python/ceval.c:PyEval_ReInitThreads which
847 # is called from PyOS_AfterFork. Here we cleanup threading module state
848 # that should not exist after a fork.
849
850 # Reset _active_limbo_lock, in case we forked while the lock was held
851 # by another (non-forked) thread. http://bugs.python.org/issue874900
852 global _active_limbo_lock
853 _active_limbo_lock = _allocate_lock()
854
855 # fork() only copied the current thread; clear references to others.
856 new_active = {}
857 current = current_thread()
858 with _active_limbo_lock:
Antoine Pitrou9fb1aca2008-09-06 23:04:32 +0000859 for thread in _active.itervalues():
Jesse Noller5e62ca42008-07-16 20:03:47 +0000860 if thread is current:
Antoine Pitrou9fb1aca2008-09-06 23:04:32 +0000861 # There is only one active thread. We reset the ident to
862 # its new value since it can have changed.
863 ident = _get_ident()
864 thread._Thread__ident = ident
Jesse Noller5e62ca42008-07-16 20:03:47 +0000865 new_active[ident] = thread
866 else:
867 # All the others are already stopped.
868 # We don't call _Thread__stop() because it tries to acquire
869 # thread._Thread__block which could also have been held while
870 # we forked.
871 thread._Thread__stopped = True
872
873 _limbo.clear()
874 _active.clear()
875 _active.update(new_active)
876 assert len(_active) == 1
877
878
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000879# Self-test code
880
881def _test():
882
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000883 class BoundedQueue(_Verbose):
884
885 def __init__(self, limit):
886 _Verbose.__init__(self)
887 self.mon = RLock()
888 self.rc = Condition(self.mon)
889 self.wc = Condition(self.mon)
890 self.limit = limit
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000891 self.queue = deque()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000892
893 def put(self, item):
894 self.mon.acquire()
895 while len(self.queue) >= self.limit:
896 self._note("put(%s): queue full", item)
897 self.wc.wait()
898 self.queue.append(item)
899 self._note("put(%s): appended, length now %d",
900 item, len(self.queue))
901 self.rc.notify()
902 self.mon.release()
903
904 def get(self):
905 self.mon.acquire()
906 while not self.queue:
907 self._note("get(): queue empty")
908 self.rc.wait()
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000909 item = self.queue.popleft()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000910 self._note("get(): got %s, %d left", item, len(self.queue))
911 self.wc.notify()
912 self.mon.release()
913 return item
914
915 class ProducerThread(Thread):
916
917 def __init__(self, queue, quota):
918 Thread.__init__(self, name="Producer")
919 self.queue = queue
920 self.quota = quota
921
922 def run(self):
Guido van Rossumb26a1b41998-05-20 17:05:52 +0000923 from random import random
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000924 counter = 0
925 while counter < self.quota:
926 counter = counter + 1
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000927 self.queue.put("%s.%d" % (self.name, counter))
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000928 _sleep(random() * 0.00001)
929
930
931 class ConsumerThread(Thread):
932
933 def __init__(self, queue, count):
934 Thread.__init__(self, name="Consumer")
935 self.queue = queue
936 self.count = count
937
938 def run(self):
939 while self.count > 0:
940 item = self.queue.get()
941 print item
942 self.count = self.count - 1
943
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000944 NP = 3
945 QL = 4
946 NI = 5
947
948 Q = BoundedQueue(QL)
949 P = []
950 for i in range(NP):
951 t = ProducerThread(Q, NI)
Benjamin Petersoncbae8692008-08-18 17:45:09 +0000952 t.name = ("Producer-%d" % (i+1))
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000953 P.append(t)
954 C = ConsumerThread(Q, NI*NP)
955 for t in P:
956 t.start()
957 _sleep(0.000001)
958 C.start()
959 for t in P:
960 t.join()
961 C.join()
962
963if __name__ == '__main__':
964 _test()