Processing stats size limited, convenient methods

Author: MirekVales

WhatNow/WhatNow.Contracts/Statistics/FixedSizeQueue.cs

@@ -0,0 +1,20 @@
+namespace WhatNow.Contracts.Statistics
+{
+    using System.Collections.Generic;
+
+    public class FixedSizedQueue<T> : Queue<T>
+    {
+        public int MaxCapacity { get; }
+
+        public FixedSizedQueue(int capacity) : base(capacity)
+            => MaxCapacity = capacity;
+
+        public new void Enqueue(T item)
+        {
+            if (MaxCapacity != 0 && MaxCapacity < Count + 1)
+                Dequeue();
+
+            base.Enqueue(item);
+        }
+    }
+}

WhatNow/WhatNow.Essentials/ActionPipe.cs

@@ -31,13 +31,15 @@ public class ActionPipe : IActionPipe
         public bool BreakRequested => Current.Any(a => a.BreakRequested);
 
         readonly object executionsLock = new object();
-        readonly Dictionary<Type, Queue<TimeSpan>> executions;
+        readonly Dictionary<Type, FixedSizedQueue<TimeSpan>> executions;
 
         public IEnumerable<BreakRequestReason> BreakReasons => actions
             .Select(a => a.Value.BreakRequestReason)
             .Where(br => br != null)
             .ToArray();
 
+        public const int MAX_QUEUE_SIZE = 100;
+
         public ActionPipe(IActionPipeMap map, ActionToken actionToken, IDependencyResolver dependencyResolver)
         {
             ActionToken = actionToken;
@@ -50,7 +52,7 @@ public ActionPipe(IActionPipeMap map, ActionToken actionToken, IDependencyResolv
 
             Current = new IAction[0];
 
-            executions = actions.Keys.ToDictionary(k => k, _ => new Queue<TimeSpan>());
+            executions = actions.Keys.ToDictionary(k => k, _ => new FixedSizedQueue<TimeSpan>(MAX_QUEUE_SIZE));
         }
 
         public void Restart()
@@ -63,6 +65,10 @@ public void Restart()
                 .ToDictionary(k => k, t => (IAction)dependencyResolver.Resolve(t));
 
             Current = new IAction[0];
+
+            lock (executionsLock)
+                foreach (var type in executions.Values)
+                    type.Clear();
         }
 
         public bool TryGetNextTask(IResourceManager resourceManager, TaskFactory taskFactory, out Task task)

WhatNow/WhatNow.Essentials/TaskList.cs

@@ -1,12 +1,12 @@
-using System;
-using System.Threading;
-using System.Threading.Tasks;
-using System.Collections.Generic;
-using Helios.Concurrency;
-using WhatNow.Contracts.ThreadPool;
-
-namespace WhatNow.Essentials
+namespace WhatNow.Essentials
 {
+    using System;
+    using System.Threading;
+    using Helios.Concurrency;
+    using System.Threading.Tasks;
+    using System.Collections.Generic;
+    using WhatNow.Contracts.ThreadPool;
+
     public class TaskList : ITaskList
     {
         readonly CancellationTokenSource cancellationTokenSource;
@@ -59,7 +59,7 @@ public ITaskList With(IEnumerable<Action> actions)
 
             return this;
         }
-        
+
         public ITaskList With(Action<CancellationToken> action)
         {
             ThrowIfDisposed();
@@ -82,7 +82,16 @@ public void WaitAllFinished()
 
             Task.WaitAll(tasks.ToArray());
 
-            ClearTasks();
+            ClearCompletedTasks();
+        }
+
+        public async Task WaitAllFinishedAsync()
+        {
+            ThrowIfDisposed();
+
+            await Task.WhenAll(tasks.ToArray());
+
+            ClearCompletedTasks();
         }
 
         void ThrowIfDisposed()
@@ -102,13 +111,13 @@ public void Dispose()
                 cancellationTokenSource?.Dispose();
                 pool?.Dispose();
 
-                ClearTasks();
+                ClearCompletedTasks();
 
                 disposed = true;
             }
         }
 
-        void ClearTasks()
+        void ClearCompletedTasks()
         {
             foreach (var task in tasks)
                 task.Dispose();