// ========================= private void PeekWakesUpTest(LevelingQueue <int> queue) { Barrier bar = new Barrier(3); int peekResult = 0; int peekResult2 = 0; Task task = Task.Run(() => { bar.SignalAndWait(); int item = 0; AtomicSet(ref peekResult, queue.TryPeek(out item, 60000)); Assert.AreEqual(100, item); }); Task task2 = Task.Run(() => { bar.SignalAndWait(); int item = 0; AtomicSet(ref peekResult2, queue.TryPeek(out item, 60000)); Assert.AreEqual(100, item); }); bar.SignalAndWait(); Thread.Sleep(10); Assert.AreEqual(0, Volatile.Read(ref peekResult)); queue.Add(100); TimingAssert.AreEqual(10000, 1, () => Volatile.Read(ref peekResult)); TimingAssert.AreEqual(10000, 1, () => Volatile.Read(ref peekResult2)); Task.WaitAll(task, task2); }
// ========================= private void TimeoutWorksTest(LevelingQueue <int> queue) { Barrier bar = new Barrier(2); int takeResult = 0; int peekResult = 0; int addResult = 0; Task task = Task.Run(() => { bar.SignalAndWait(); int item = 0; AtomicSet(ref takeResult, queue.TryTake(out item, 100)); AtomicSet(ref peekResult, queue.TryPeek(out item, 100)); while (queue.TryAdd(-1)) { ; } if (queue.IsBackgroundTransferingEnabled) { queue.AddForcedToHighLevelQueue(-1); queue.AddForced(-1); // To prevent background transferer from freeing the space } AtomicSet(ref addResult, queue.TryAdd(100, 100)); }); bar.SignalAndWait(); TimingAssert.AreEqual(10000, 2, () => Volatile.Read(ref takeResult), "take"); TimingAssert.AreEqual(10000, 2, () => Volatile.Read(ref peekResult), "peek"); TimingAssert.AreEqual(10000, 2, () => Volatile.Read(ref addResult), "Add"); task.Wait(); }
// ========================== private void RunComplexTest(LevelingQueue <int> q, int elemCount, int thCount) { int atomicRandom = 0; int trackElemCount = elemCount; int addFinished = 0; Thread[] threadsTake = new Thread[thCount]; Thread[] threadsAdd = new Thread[thCount]; CancellationTokenSource tokSrc = new CancellationTokenSource(); List <int> global = new List <int>(elemCount); Action addAction = () => { Random rnd = new Random(Environment.TickCount + Interlocked.Increment(ref atomicRandom) * thCount * 2); while (true) { int item = Interlocked.Decrement(ref trackElemCount); if (item < 0) { break; } if (rnd.Next(100) == 0) { q.AddForced(item); } else { q.Add(item); } int sleepTime = rnd.Next(12); int tmpItem = 0; if (q.TryPeek(out tmpItem) && tmpItem == item) { sleepTime += 12; } if (sleepTime > 0) { SpinWaitHelper.SpinWait(sleepTime); } } Interlocked.Increment(ref addFinished); }; Action takeAction = () => { Random rnd = new Random(Environment.TickCount + Interlocked.Increment(ref atomicRandom) * thCount * 2); List <int> data = new List <int>(); try { while (Volatile.Read(ref addFinished) < thCount) { int tmp = 0; if (q.TryTake(out tmp, -1, tokSrc.Token)) { data.Add((int)tmp); } int sleepTime = rnd.Next(12); if (sleepTime > 0) { SpinWaitHelper.SpinWait(sleepTime); } } } catch (OperationCanceledException) { } int tmp2; while (q.TryTake(out tmp2)) { data.Add((int)tmp2); } lock (global) global.AddRange(data); }; for (int i = 0; i < threadsTake.Length; i++) { threadsTake[i] = new Thread(new ThreadStart(takeAction)); } for (int i = 0; i < threadsAdd.Length; i++) { threadsAdd[i] = new Thread(new ThreadStart(addAction)); } for (int i = 0; i < threadsTake.Length; i++) { threadsTake[i].Start(); } for (int i = 0; i < threadsAdd.Length; i++) { threadsAdd[i].Start(); } for (int i = 0; i < threadsAdd.Length; i++) { threadsAdd[i].Join(); } tokSrc.Cancel(); for (int i = 0; i < threadsTake.Length; i++) { threadsTake[i].Join(); } Assert.AreEqual(elemCount, global.Count); global.Sort(); for (int i = 0; i < elemCount; i++) { Assert.AreEqual(i, global[i]); } }
// ========================= private void CancellationWorksTest(LevelingQueue <int> queue) { Barrier bar = new Barrier(2); CancellationTokenSource takeSource = new CancellationTokenSource(); int takeResult = 0; CancellationTokenSource peekSource = new CancellationTokenSource(); int peekResult = 0; CancellationTokenSource addSource = new CancellationTokenSource(); int addResult = 0; Task task = Task.Run(() => { bar.SignalAndWait(); int item = 0; try { AtomicSet(ref takeResult, queue.TryTake(out item, 60000, takeSource.Token)); } catch (OperationCanceledException) { Interlocked.Exchange(ref takeResult, 3); } try { AtomicSet(ref peekResult, queue.TryPeek(out item, 60000, peekSource.Token)); } catch (OperationCanceledException) { Interlocked.Exchange(ref peekResult, 3); } while (queue.TryAdd(-1)) { ; } if (queue.IsBackgroundTransferingEnabled) { queue.AddForced(-1); } try { AtomicSet(ref addResult, queue.TryAdd(100, 60000, addSource.Token)); } catch (OperationCanceledException) { Interlocked.Exchange(ref addResult, 3); } }); bar.SignalAndWait(); Thread.Sleep(10); Assert.AreEqual(0, takeResult); takeSource.Cancel(); TimingAssert.AreEqual(10000, 3, () => Volatile.Read(ref takeResult)); Thread.Sleep(10); Assert.AreEqual(0, peekResult); peekSource.Cancel(); TimingAssert.AreEqual(10000, 3, () => Volatile.Read(ref peekResult)); Thread.Sleep(10); Assert.AreEqual(0, addResult); addSource.Cancel(); TimingAssert.AreEqual(10000, 3, () => Volatile.Read(ref addResult)); task.Wait(); }