1 // AsyncExecutingMessageBox.cs
3 // Copyright (c) 2011 Jérémie "garuma" Laval
4 // Copyright (c) 2012 Petr Onderka
6 // Permission is hereby granted, free of charge, to any person obtaining a copy
7 // of this software and associated documentation files (the "Software"), to deal
8 // in the Software without restriction, including without limitation the rights
9 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 // copies of the Software, and to permit persons to whom the Software is
11 // furnished to do so, subject to the following conditions:
13 // The above copyright notice and this permission notice shall be included in
14 // all copies or substantial portions of the Software.
16 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
24 using System.Collections.Concurrent;
26 namespace System.Threading.Tasks.Dataflow {
27 class AsyncExecutingMessageBox<TInput, TTask>
28 : ExecutingMessageBoxBase<TInput>
30 public delegate bool AsyncProcessItem (out TTask task);
32 readonly AsyncProcessItem processItem;
33 readonly Action<TTask> processFinishedTask;
35 public AsyncExecutingMessageBox (
36 ITargetBlock<TInput> target, BlockingCollection<TInput> messageQueue,
37 CompletionHelper compHelper, Func<bool> externalCompleteTester,
38 AsyncProcessItem processItem, Action<TTask> processFinishedTask,
39 Action outgoingQueueComplete, ExecutionDataflowBlockOptions options)
41 target, messageQueue, compHelper, externalCompleteTester,
42 outgoingQueueComplete, options)
44 this.processItem = processItem;
45 this.processFinishedTask = processFinishedTask;
48 protected override void ProcessQueue ()
52 ProcessQueueWithoutStart ();
55 void ProcessQueueWithoutStart ()
57 // catch is needed here, if the Task-returning delegate throws exception itself
62 if (!processItem (out task))
64 if (task == null || task.IsCanceled
65 || (task.IsCompleted && !task.IsFaulted)) {
66 if (processFinishedTask != null)
67 processFinishedTask (task);
68 } else if (task.IsFaulted) {
69 CompHelper.RequestFault (task.Exception, false);
73 t => TaskFinished ((TTask)t), Options.TaskScheduler);
78 } catch (Exception e) {
79 CompHelper.RequestFault (e, false);
82 FinishProcessQueue ();
85 void TaskFinished (TTask task)
88 CompHelper.RequestFault (task.Exception, false);
89 FinishProcessQueue ();
93 if (processFinishedTask != null)
94 processFinishedTask (task);
96 ProcessQueueWithoutStart ();