-
Notifications
You must be signed in to change notification settings - Fork 269
/
AddRemoveFromSameThreads.cs
110 lines (91 loc) · 3.66 KB
/
AddRemoveFromSameThreads.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using BenchmarkDotNet.Attributes;
using MicroBenchmarks;
namespace System.Collections.Concurrent
{
[BenchmarkCategory(Categories.Libraries, Categories.Collections, Categories.GenericCollections, Categories.NoWASM)]
[GenericTypeArguments(typeof(int))] // value type
[GenericTypeArguments(typeof(string))] // reference type
[MinWarmupCount(6, forceAutoWarmup: true)]
[MaxWarmupCount(10, forceAutoWarmup: true)]
public class AddRemoveFromSameThreads<T>
{
const int NumThreads = 2;
[Params(2_000_000)]
public int Size;
private Barrier _barrier;
private Task[] _tasks;
[IterationCleanup]
public void IterationCleanup() => _barrier.Dispose();
[IterationSetup(Target = nameof(ConcurrentBag))]
public void SetupConcurrentBagIteration()
{
var bag = new ConcurrentBag<T>();
_barrier = new Barrier(NumThreads + 1);
_tasks = Enumerable.Range(0, NumThreads)
.Select(_ =>
Task.Factory.StartNew(() =>
{
_barrier.SignalAndWait();
for (int i = 0; i < Size; i++)
{
bag.Add(default);
bag.TryTake(out T _);
}
}, CancellationToken.None, TaskCreationOptions.LongRunning, TaskScheduler.Default))
.ToArray();
}
[Benchmark]
public void ConcurrentBag() => SignalAndWaitForAllTasks();
[IterationSetup(Target = nameof(ConcurrentStack))]
public void SetupConcurrentStackIteration()
{
var stack = new ConcurrentStack<T>();
_barrier = new Barrier(NumThreads + 1);
_tasks = Enumerable.Range(0, NumThreads)
.Select(_ =>
Task.Factory.StartNew(() =>
{
_barrier.SignalAndWait();
for (int i = 0; i < Size; i++)
{
stack.Push(default);
stack.TryPop(out T _);
}
}, CancellationToken.None, TaskCreationOptions.LongRunning, TaskScheduler.Default))
.ToArray();
}
[Benchmark]
public void ConcurrentStack() => SignalAndWaitForAllTasks();
[IterationSetup(Target = nameof(ConcurrentQueue))]
public void SetupConcurrentQueueIteration()
{
var queue = new ConcurrentQueue<T>();
_barrier = new Barrier(NumThreads + 1);
_tasks = Enumerable.Range(0, NumThreads)
.Select(_ =>
Task.Factory.StartNew(() =>
{
_barrier.SignalAndWait();
for (int i = 0; i < Size; i++)
{
queue.Enqueue(default);
queue.TryDequeue(out T _);
}
}, CancellationToken.None, TaskCreationOptions.LongRunning, TaskScheduler.Default))
.ToArray();
}
[Benchmark]
public void ConcurrentQueue() => SignalAndWaitForAllTasks();
private void SignalAndWaitForAllTasks()
{
_barrier.SignalAndWait();
Task.WaitAll(_tasks);
}
}
}