using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;

// 別スレッドで走る doSomething メソッド中で例外をキャッチせず、ジョインされたスレッド側で
// キャッチを行なっているため、デバッグ実行を行うと例外の発生で一時停止します。その場合は
// [継続] ボタンをクリックして続行させてください。「デバッグ無しで開始」させると、途中での停止は
// 発生しません。

namespace ManagedTasksTest2
{
    class Program
    {
        static void Main(string[] args)
        {
            // 生成されて終了していない Task オブジェクトを保持する
            var tasks = new List<Task>();
            var syncObject = new object();
            var cts = new CancellationTokenSource();

            while (true)
            {
                Console.WriteLine("入力してください(終了: \"]\"キー)。");
                var input = Console.ReadLine();

                if (input == "]") break;

                var task = new Task<string>(() => doSomething(input, cts.Token));
                task.ContinueWith((s) => 
                {
                    // task は正常終了したか?
                    if (task.Status == TaskStatus.RanToCompletion)
                    {
                        Console.WriteLine(s.Result);
                    }
                    else
                    {
                        Console.WriteLine("非同期処理が正常終了しませんでした。");
                    }
                    lock (syncObject)
                    {
                        tasks.Remove(task);
                    }
                });
                lock (syncObject)
                {
                    tasks.Add(task);
                }
                task.Start();
            }

            // キャンセル要求を伝える
            cts.Cancel();

            Console.WriteLine("非同期処理の終了を待ちます。");
            Task[] taskArray;
            lock (syncObject)
            {
                taskArray = tasks.ToArray();
            }
            try
            {
                Task.WaitAll(taskArray);
            }
            // ジョインされたタスクで発生した例外をキャッチ
            catch (AggregateException ae)
            {
                ae.Handle((ex) =>
                {
                    var result = false;

                    // 例外が OperationCanceledException だったら例外を除去する(ほかの例外は再スローされる)
                    // ただし、この例では doSomething の ContinueWith でセットしている後処理内で
                    // 正常終了しなかったタスクも Remove 対象としているため、WaitAll では検出されない
                    if (ex is OperationCanceledException)
                    {
                        Console.WriteLine(ex.Message);
                        result = true;
                    }
                    return result;
                });
            }

            Console.WriteLine("Enter キー押下で終了します。");
            Console.ReadLine();
        }

        static string doSomething(string s, CancellationToken token)
        {
            Thread.Sleep(2500);
            if (token.IsCancellationRequested)
            {
                var message = string.Format("キャンセルしました: 処理中のデータ({0})", s);
                throw new OperationCanceledException(message);
            }
            Thread.Sleep(2500);

            return string.Format("retrun: {0}", s);
        }
    }
}