blob: bd0fc2d8f3c735d6b82e444b1d25fdc171972725 [file] [log] [blame]
Alexandre Vassalottif260e442008-05-11 19:59:59 +00001:mod:`queue` --- A synchronized queue class
Georg Brandl116aa622007-08-15 14:28:22 +00002===========================================
3
Alexandre Vassalottif260e442008-05-11 19:59:59 +00004.. module:: queue
Georg Brandl116aa622007-08-15 14:28:22 +00005 :synopsis: A synchronized queue class.
6
Raymond Hettinger10480942011-01-10 03:26:08 +00007**Source code:** :source:`Lib/queue.py`
Georg Brandl116aa622007-08-15 14:28:22 +00008
Raymond Hettinger4f707fd2011-01-10 19:54:11 +00009--------------
10
Alexandre Vassalottif260e442008-05-11 19:59:59 +000011The :mod:`queue` module implements multi-producer, multi-consumer queues.
Thomas Wouters89d996e2007-09-08 17:39:28 +000012It is especially useful in threaded programming when information must be
Georg Brandl116aa622007-08-15 14:28:22 +000013exchanged safely between multiple threads. The :class:`Queue` class in this
14module implements all the required locking semantics. It depends on the
Thomas Wouters89d996e2007-09-08 17:39:28 +000015availability of thread support in Python; see the :mod:`threading`
16module.
Georg Brandl116aa622007-08-15 14:28:22 +000017
R David Murrayb98b37f2012-05-08 21:28:24 -040018The module implements three types of queue, which differ only in the order in
Serhiy Storchaka4ecfa452016-05-16 09:31:54 +030019which the entries are retrieved. In a :abbr:`FIFO (first-in, first-out)`
20queue, the first tasks added are the first retrieved. In a
21:abbr:`LIFO (last-in, first-out)` queue, the most recently added entry is
Raymond Hettinger35641462008-01-17 00:13:27 +000022the first retrieved (operating like a stack). With a priority queue,
23the entries are kept sorted (using the :mod:`heapq` module) and the
24lowest valued entry is retrieved first.
Georg Brandl116aa622007-08-15 14:28:22 +000025
Raymond Hettinger87dc4d62016-09-09 15:57:13 -070026Internally, the module uses locks to temporarily block competing threads;
27however, it is not designed to handle reentrancy within a thread.
Éric Araujo6e6cb8e2010-11-16 19:13:50 +000028
Alexandre Vassalottif260e442008-05-11 19:59:59 +000029The :mod:`queue` module defines the following classes and exceptions:
Georg Brandl116aa622007-08-15 14:28:22 +000030
Andrew M. Kuchling2b600e52010-02-26 13:35:56 +000031.. class:: Queue(maxsize=0)
Georg Brandl116aa622007-08-15 14:28:22 +000032
Serhiy Storchaka4ecfa452016-05-16 09:31:54 +030033 Constructor for a :abbr:`FIFO (first-in, first-out)` queue. *maxsize* is
34 an integer that sets the upperbound
Georg Brandl116aa622007-08-15 14:28:22 +000035 limit on the number of items that can be placed in the queue. Insertion will
36 block once this size has been reached, until queue items are consumed. If
37 *maxsize* is less than or equal to zero, the queue size is infinite.
38
Andrew M. Kuchling2b600e52010-02-26 13:35:56 +000039.. class:: LifoQueue(maxsize=0)
Raymond Hettinger35641462008-01-17 00:13:27 +000040
Serhiy Storchaka4ecfa452016-05-16 09:31:54 +030041 Constructor for a :abbr:`LIFO (last-in, first-out)` queue. *maxsize* is
42 an integer that sets the upperbound
Raymond Hettinger35641462008-01-17 00:13:27 +000043 limit on the number of items that can be placed in the queue. Insertion will
44 block once this size has been reached, until queue items are consumed. If
45 *maxsize* is less than or equal to zero, the queue size is infinite.
46
Christian Heimes679db4a2008-01-18 09:56:22 +000047
Andrew M. Kuchling2b600e52010-02-26 13:35:56 +000048.. class:: PriorityQueue(maxsize=0)
Raymond Hettinger35641462008-01-17 00:13:27 +000049
50 Constructor for a priority queue. *maxsize* is an integer that sets the upperbound
51 limit on the number of items that can be placed in the queue. Insertion will
52 block once this size has been reached, until queue items are consumed. If
53 *maxsize* is less than or equal to zero, the queue size is infinite.
54
55 The lowest valued entries are retrieved first (the lowest valued entry is the
56 one returned by ``sorted(list(entries))[0]``). A typical pattern for entries
57 is a tuple in the form: ``(priority_number, data)``.
Georg Brandl116aa622007-08-15 14:28:22 +000058
Christian Heimes679db4a2008-01-18 09:56:22 +000059
Georg Brandl116aa622007-08-15 14:28:22 +000060.. exception:: Empty
61
Serhiy Storchaka9e0ae532013-08-24 00:23:38 +030062 Exception raised when non-blocking :meth:`~Queue.get` (or
63 :meth:`~Queue.get_nowait`) is called
Georg Brandl116aa622007-08-15 14:28:22 +000064 on a :class:`Queue` object which is empty.
65
66
67.. exception:: Full
68
Serhiy Storchaka9e0ae532013-08-24 00:23:38 +030069 Exception raised when non-blocking :meth:`~Queue.put` (or
70 :meth:`~Queue.put_nowait`) is called
Georg Brandl116aa622007-08-15 14:28:22 +000071 on a :class:`Queue` object which is full.
72
73
74.. _queueobjects:
75
76Queue Objects
77-------------
78
Christian Heimes292d3512008-02-03 16:51:08 +000079Queue objects (:class:`Queue`, :class:`LifoQueue`, or :class:`PriorityQueue`)
Georg Brandl48310cd2009-01-03 21:18:54 +000080provide the public methods described below.
Georg Brandl116aa622007-08-15 14:28:22 +000081
82
83.. method:: Queue.qsize()
84
Guido van Rossum7736b5b2008-01-15 21:44:53 +000085 Return the approximate size of the queue. Note, qsize() > 0 doesn't
86 guarantee that a subsequent get() will not block, nor will qsize() < maxsize
87 guarantee that put() will not block.
Georg Brandl116aa622007-08-15 14:28:22 +000088
89
Raymond Hettinger47aa9892009-03-07 14:07:37 +000090.. method:: Queue.empty()
91
92 Return ``True`` if the queue is empty, ``False`` otherwise. If empty()
93 returns ``True`` it doesn't guarantee that a subsequent call to put()
94 will not block. Similarly, if empty() returns ``False`` it doesn't
95 guarantee that a subsequent call to get() will not block.
96
97
98.. method:: Queue.full()
99
100 Return ``True`` if the queue is full, ``False`` otherwise. If full()
101 returns ``True`` it doesn't guarantee that a subsequent call to get()
102 will not block. Similarly, if full() returns ``False`` it doesn't
103 guarantee that a subsequent call to put() will not block.
104
105
Georg Brandl18244152009-09-02 20:34:52 +0000106.. method:: Queue.put(item, block=True, timeout=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000107
108 Put *item* into the queue. If optional args *block* is true and *timeout* is
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300109 ``None`` (the default), block if necessary until a free slot is available. If
Georg Brandl116aa622007-08-15 14:28:22 +0000110 *timeout* is a positive number, it blocks at most *timeout* seconds and raises
111 the :exc:`Full` exception if no free slot was available within that time.
112 Otherwise (*block* is false), put an item on the queue if a free slot is
113 immediately available, else raise the :exc:`Full` exception (*timeout* is
114 ignored in that case).
115
Georg Brandl116aa622007-08-15 14:28:22 +0000116
117.. method:: Queue.put_nowait(item)
118
119 Equivalent to ``put(item, False)``.
120
121
Georg Brandl18244152009-09-02 20:34:52 +0000122.. method:: Queue.get(block=True, timeout=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000123
124 Remove and return an item from the queue. If optional args *block* is true and
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300125 *timeout* is ``None`` (the default), block if necessary until an item is available.
Georg Brandl116aa622007-08-15 14:28:22 +0000126 If *timeout* is a positive number, it blocks at most *timeout* seconds and
127 raises the :exc:`Empty` exception if no item was available within that time.
128 Otherwise (*block* is false), return an item if one is immediately available,
129 else raise the :exc:`Empty` exception (*timeout* is ignored in that case).
130
Georg Brandl116aa622007-08-15 14:28:22 +0000131
132.. method:: Queue.get_nowait()
133
134 Equivalent to ``get(False)``.
135
136Two methods are offered to support tracking whether enqueued tasks have been
137fully processed by daemon consumer threads.
138
139
140.. method:: Queue.task_done()
141
142 Indicate that a formerly enqueued task is complete. Used by queue consumer
143 threads. For each :meth:`get` used to fetch a task, a subsequent call to
144 :meth:`task_done` tells the queue that the processing on the task is complete.
145
146 If a :meth:`join` is currently blocking, it will resume when all items have been
147 processed (meaning that a :meth:`task_done` call was received for every item
148 that had been :meth:`put` into the queue).
149
150 Raises a :exc:`ValueError` if called more times than there were items placed in
151 the queue.
152
Georg Brandl116aa622007-08-15 14:28:22 +0000153
154.. method:: Queue.join()
155
156 Blocks until all items in the queue have been gotten and processed.
157
158 The count of unfinished tasks goes up whenever an item is added to the queue.
159 The count goes down whenever a consumer thread calls :meth:`task_done` to
160 indicate that the item was retrieved and all work on it is complete. When the
Raymond Hettinger28c013d2009-03-10 00:07:25 +0000161 count of unfinished tasks drops to zero, :meth:`join` unblocks.
Georg Brandl116aa622007-08-15 14:28:22 +0000162
Georg Brandl116aa622007-08-15 14:28:22 +0000163
164Example of how to wait for enqueued tasks to be completed::
165
Victor Stinnerde311342015-03-18 14:05:43 +0100166 def worker():
167 while True:
168 item = q.get()
169 if item is None:
170 break
171 do_work(item)
172 q.task_done()
Georg Brandl116aa622007-08-15 14:28:22 +0000173
Victor Stinnerde311342015-03-18 14:05:43 +0100174 q = queue.Queue()
175 threads = []
176 for i in range(num_worker_threads):
177 t = threading.Thread(target=worker)
Georg Brandl48310cd2009-01-03 21:18:54 +0000178 t.start()
Victor Stinnerde311342015-03-18 14:05:43 +0100179 threads.append(t)
Georg Brandl116aa622007-08-15 14:28:22 +0000180
Victor Stinnerde311342015-03-18 14:05:43 +0100181 for item in source():
182 q.put(item)
Georg Brandl116aa622007-08-15 14:28:22 +0000183
Victor Stinnerde311342015-03-18 14:05:43 +0100184 # block until all tasks are done
185 q.join()
186
187 # stop workers
188 for i in range(num_worker_threads):
189 q.put(None)
190 for t in threads:
191 t.join()
Georg Brandl116aa622007-08-15 14:28:22 +0000192
Antoine Pitrou696efdd2011-01-07 19:16:12 +0000193
194.. seealso::
195
196 Class :class:`multiprocessing.Queue`
197 A queue class for use in a multi-processing (rather than multi-threading)
198 context.
199
Georg Brandl2f2a9f72011-01-07 20:58:25 +0000200 :class:`collections.deque` is an alternative implementation of unbounded
Serhiy Storchaka9e0ae532013-08-24 00:23:38 +0300201 queues with fast atomic :meth:`~collections.deque.append` and
202 :meth:`~collections.deque.popleft` operations that do not require locking.
Raymond Hettingerfc902132011-01-07 20:33:09 +0000203