3 // Copyright (c) 2008 Jérémie "Garuma" Laval
5 // Permission is hereby granted, free of charge, to any person obtaining a copy
6 // of this software and associated documentation files (the "Software"), to deal
7 // in the Software without restriction, including without limitation the rights
8 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9 // copies of the Software, and to permit persons to whom the Software is
10 // furnished to do so, subject to the following conditions:
12 // The above copyright notice and this permission notice shall be included in
13 // all copies or substantial portions of the Software.
15 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
28 using System.Threading;
29 using System.Collections;
30 using System.Collections.Generic;
31 using System.Runtime.Serialization;
33 namespace System.Collections.Concurrent
36 [System.Diagnostics.DebuggerDisplay ("Count={Count}")]
37 [System.Diagnostics.DebuggerTypeProxy (typeof (CollectionDebuggerView<>))]
38 public class ConcurrentQueue<T> : IProducerConsumerCollection<T>, IEnumerable<T>, ICollection,
47 Node head = new Node ();
51 public ConcurrentQueue ()
56 public ConcurrentQueue (IEnumerable<T> collection): this()
58 foreach (T item in collection)
62 public void Enqueue (T item)
64 Node node = new Node ();
73 oldNext = oldTail.Next;
75 Thread.MemoryBarrier ();
77 // Did tail was already updated ?
78 if (tail == oldTail) {
79 if (oldNext == null) {
80 // The place is for us
81 update = Interlocked.CompareExchange (ref tail.Next, node, null) == null;
83 // another Thread already used the place so give him a hand by putting tail where it should be
84 Interlocked.CompareExchange (ref tail, oldNext, oldTail);
88 // At this point we added correctly our node, now we have to update tail. If it fails then it will be done by another thread
89 Interlocked.CompareExchange (ref tail, node, oldTail);
90 Interlocked.Increment (ref count);
93 bool IProducerConsumerCollection<T>.TryAdd (T item)
99 public bool TryDequeue (out T result)
101 result = default (T);
103 bool advanced = false;
108 oldNext = oldHead.Next;
110 Thread.MemoryBarrier ();
112 if (oldHead == head) {
114 if (oldHead == oldTail) {
115 // This should be false then
116 if (oldNext != null) {
117 // If not then the linked list is mal formed, update tail
118 Interlocked.CompareExchange (ref tail, oldNext, oldTail);
121 result = default (T);
124 result = oldNext.Value;
125 advanced = Interlocked.CompareExchange (ref head, oldNext, oldHead) == oldHead;
130 oldNext.Value = default (T);
132 Interlocked.Decrement (ref count);
137 public bool TryPeek (out T result)
139 result = default (T);
145 Node oldNext = oldHead.Next;
147 if (oldNext == null) {
148 result = default (T);
152 result = oldNext.Value;
154 Thread.MemoryBarrier ();
156 //check if head has been updated
157 update = head != oldHead;
162 internal void Clear ()
165 tail = head = new Node ();
168 IEnumerator IEnumerable.GetEnumerator ()
170 return (IEnumerator)InternalGetEnumerator ();
173 public IEnumerator<T> GetEnumerator ()
175 return InternalGetEnumerator ();
178 IEnumerator<T> InternalGetEnumerator ()
181 while ((my_head = my_head.Next) != null) {
182 yield return my_head.Value;
186 void ICollection.CopyTo (Array array, int index)
189 throw new ArgumentNullException ("array");
191 throw new ArgumentException ("The array can't be multidimensional");
192 if (array.GetLowerBound (0) != 0)
193 throw new ArgumentException ("The array needs to be 0-based");
195 T[] dest = array as T[];
197 throw new ArgumentException ("The array cannot be cast to the collection element type", "array");
198 CopyTo (dest, index);
201 public void CopyTo (T[] array, int index)
204 throw new ArgumentNullException ("array");
206 throw new ArgumentOutOfRangeException ("index");
207 if (index >= array.Length)
208 throw new ArgumentException ("index is equals or greather than array length", "index");
210 IEnumerator<T> e = InternalGetEnumerator ();
212 while (e.MoveNext ()) {
213 if (i == array.Length - index)
214 throw new ArgumentException ("The number of elememts in the collection exceeds the capacity of array", "array");
215 array[i++] = e.Current;
219 public T[] ToArray ()
221 return new List<T> (this).ToArray ();
224 bool ICollection.IsSynchronized {
228 bool IProducerConsumerCollection<T>.TryTake (out T item)
230 return TryDequeue (out item);
233 object syncRoot = new object();
234 object ICollection.SyncRoot {
235 get { return syncRoot; }
244 public bool IsEmpty {