Add QueueAsync.

Author: tosh-coding

src/AsyncFiberWorks/Threading/ThreadPoolExtensions.cs

@@ -0,0 +1,49 @@
+using AsyncFiberWorks.Core;
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace AsyncFiberWorks.Threading
+{
+    /// <summary>
+    /// Extension of thread pool.
+    /// </summary>
+    public static class ThreadPoolExtensions
+    {
+        /// <summary>
+        /// Enqueue an action. Then returns a task to wait for the completion of the action.
+        /// </summary>
+        /// <param name="threadPool">A thread pool.</param>
+        /// <param name="callback"></param>
+        /// <returns>A task that waits until a given action is finished.</returns>
+        public static async Task QueueAsync(this IThreadPool threadPool, WaitCallback callback)
+        {
+            var tcs = new TaskCompletionSource<byte>(TaskCreationOptions.RunContinuationsAsynchronously);
+            threadPool.Queue((_) =>
+            {
+                try
+                {
+                    callback(null);
+                }
+                catch (Exception ex)
+                {
+                    tcs.SetException(ex);
+                    return;
+                }
+                tcs.SetResult(0);
+            });
+            await tcs.Task.ConfigureAwait(false);
+        }
+
+        /// <summary>
+        /// Enqueue an action. Then returns a task to wait for the completion of the action.
+        /// </summary>
+        /// <param name="threadPool">A thread pool.</param>
+        /// <param name="action">An action.</param>
+        /// <returns>A task that waits until a given action is finished.</returns>
+        public static async Task QueueAsync(this IThreadPool threadPool, Action action)
+        {
+            await threadPool.QueueAsync((_) => action()).ConfigureAwait(false);
+        }
+    }
+}

src/AsyncFiberWorksTests/ThreadPoolTests.cs

@@ -7,6 +7,7 @@
 using AsyncFiberWorks.Core;
 using AsyncFiberWorks.Fibers;
 using AsyncFiberWorks.Threading;
+using System;
 
 namespace AsyncFiberWorksTests
 {
@@ -198,5 +199,39 @@ public void PoolFiberWithUserThreadPool()
                 }
             }
         }
+
+        [Test, TestCaseSource(nameof(ThreadPoolCreators))]
+        public async Task ThreadPoolQueueAsync(Func<IThreadPool> poolCreator)
+        {
+            long value = 0;
+            var threadPool = poolCreator();
+            var t1 = threadPool.QueueAsync(() =>
+            {
+                value = 1;
+                Thread.Sleep(150);
+                value += 1;
+            });
+            var t2 = threadPool.QueueAsync(() =>
+            {
+                Thread.Sleep(50);
+                value *= 2;
+            });
+            var t3 = threadPool.QueueAsync(() =>
+            {
+                Thread.Sleep(100);
+                value *= 100;
+            });
+
+            await Task.WhenAll(t1, t2, t3).ConfigureAwait(false);
+            Assert.AreEqual(201, value);
+        }
+
+
+        static object[] ThreadPoolCreators =
+        {
+            new object[] { (Func<IThreadPool>)(() => DefaultThreadPool.Instance) },
+            new object[] { (Func<IThreadPool>)(() => UserThreadPool.StartNew(3)) },
+        };
+
     }
 }