Threading.cs

using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using MtkClient.Library.Utils;

namespace MtkClient.Library
{
    public class ThreadHandler : LogBase
    {
        private readonly List<Task> _tasks = new List<Task>();
        private readonly CancellationTokenSource _cts = new CancellationTokenSource();

        public CancellationToken CancellationToken => _cts.Token;
        public bool IsCancelled => _cts.IsCancellationRequested;

        public ThreadHandler(LogLevel logLevel = LogLevel.Info)
        {
            InitLogger(logLevel);
        }

        public Task RunAsync(Action action, string name = null)
        {
            var task = Task.Factory.StartNew(() =>
            {
                try
                {
                    Debug($"Thread started: {name ?? "unnamed"}");
                    action();
                    Debug($"Thread completed: {name ?? "unnamed"}");
                }
                catch (OperationCanceledException)
                {
                    Debug($"Thread cancelled: {name ?? "unnamed"}");
                }
                catch (Exception ex)
                {
                    Error($"Thread error ({name ?? "unnamed"}): {ex.Message}");
                }
            }, _cts.Token, TaskCreationOptions.LongRunning, TaskScheduler.Default);

            _tasks.Add(task);
            return task;
        }

        public void Cancel()
        {
            _cts.Cancel();
        }

        public void WaitAll(int timeoutMs = -1)
        {
            try
            {
                if (timeoutMs > 0)
                    Task.WaitAll(_tasks.ToArray(), timeoutMs);
                else
                    Task.WaitAll(_tasks.ToArray());
            }
            catch (AggregateException ae)
            {
                foreach (var ex in ae.InnerExceptions)
                {
                    if (!(ex is OperationCanceledException))
                        Error($"WaitAll error: {ex.Message}");
                }
            }
        }

        public void Dispose()
        {
            Cancel();
            _cts.Dispose();
        }
    }
}