using Microsoft.VisualStudio.TestTools.UnitTesting; using System.Collections.Generic; using System.Threading; using System.Threading.Tasks; namespace Ex.RulesTester.Threads { /// <summary> /// 異步等待所有線程結束通知測試 /// </summary> [TestClass] public class TaskAllCompleteTester { [TestMethod] public void TestTaskPoolStatus() { //運行異步多任務 TaskTest.RunTask(); var list = TaskTest.SuccessList; bool isAllComplete = false; //等待異步通知 while (!isAllComplete) { isAllComplete = TaskTest.IsAllComplete; list = TaskTest.SuccessList; } Assert.IsTrue(true); } /// <summary> /// 對於 task 多任務執行的封裝 /// </summary> private class TaskTest { private static object m_lock = new object(); public static List<string> SuccessList = new List<string>(); private static string msg = ""; public static bool IsFinish = false; public static List<Task> TaskList = new List<Task>(); static bool isAllComplete = false; /// <summary> /// 標記所有任務是否全部執行完畢 /// </summary> public static bool IsAllComplete { get { return isAllComplete; } } private TaskTest() { } public static void RunTask() { //運行5個任務 for (int i = 1; i <= 5; i++) { var task = Task.Factory.StartNew(DoTask, i); TaskList.Add(task); } //異步等待所有任務執行完畢 Task.Factory.StartNew(x => { Task.WaitAll(TaskList.ToArray()); //標記所有任務運行完成 isAllComplete = true; }, null); } private static void DoTask(object par) { Thread.Sleep(int.Parse(par.ToString()) * 1000); lock (m_lock) { SuccessList.Add("Has Completed T" + par.ToString()); } } } } }
