TPL FromAsync与TaskScheduler和TaskFactory

时间:2012-12-06 15:04:13

标签: c# .net parallel-processing task-parallel-library

我正在尝试使用TaskFactory.FromAsync创建一个Task管道/有序调度程序。

我希望能够触发Web服务请求(使用FromAsync来使用I / O完成端口),但是维护它们的顺序,并且每次只执行一次。

目前我不使用FromAsync,因此我可以TaskFactory.StartNew(()=>api.DoSyncWebServiceCall())依赖OrderedTaskScheduler使用的TaskFactory来确保只有一个请求未完成。

我认为在使用FromAsync方法时这种行为会保留,但它不会:

TaskFactory<Stuff> taskFactory = new TaskFactory<Stuff>(new OrderedTaskScheduler());
var t1 = taskFactory.FromAsync((a, s) => api.beginGetStuff(a, s), a => api.endGetStuff(a));
var t2 = taskFactory.FromAsync((a, s) => api.beginGetStuff(a, s), a => api.endGetStuff(a));
var t3 = taskFactory.FromAsync((a, s) => api.beginGetStuff(a, s), a => api.endGetStuff(a));

所有这些beginGetStuff方法都会在FromAsync调用中被调用(因为虽然它们按顺序分派,但同时发生了n个api调用。)

FromAsync的重载需要一个TaskScheduler:

public Task FromAsync(
    IAsyncResult asyncResult,
    Action<IAsyncResult> endMethod,
    TaskCreationOptions creationOptions,
    TaskScheduler scheduler
)

但文档说:

  

TaskScheduler,用于计划执行结束方法的任务。

正如您所看到的,它需要已经构建的IAsyncResult,而不是Func<IAsyncResult>

这是否需要自定义FromAsync方法,还是我遗漏了某些内容?任何人都可以建议从这个实现开始?

干杯,

编辑:

我想将此行为从调用者中抽象出来,因此,根据TaskFactory的行为(使用专门的TaskScheduler),我需要立即返回任务 - 此任务不仅会封装FromAsync任务,但在等待轮到执行时也排队该任务。

一种可能的解决方案:

class TaskExecutionQueue
{
    private readonly OrderedTaskScheduler _orderedTaskScheduler;
    private readonly TaskFactory _taskFactory;
    public TaskExecutionQueue(OrderedTaskScheduler orderedTaskScheduler)
    {
        _orderedTaskScheduler = orderedTaskScheduler;
        _taskFactory = new TaskFactory(orderedTaskScheduler);

    }

    public Task<TResult> QueueTask<TResult>(Func<Task<TResult>> taskGenerator)
    {
        return _taskFactory.StartNew(taskGenerator).Unwrap();
    }
}

但是,这会在FromAsync呼叫发生时使用线程。理想情况下,我不必这样做。

3 个答案:

答案 0 :(得分:2)

您无法安排IO任务,因为他们没有与之关联的线程。 Windows内核提供无线程IO操作。启动这些IO不涉及托管代码,TaskScheduler类不起作用。

所以你必须延迟启动IO,直到你确定你真的希望网络被击中。您可以使用SemaphoreSlim.WaitAsync来限制当前运行的任务量。在启动单个IO之前等待该方法的结果并等待它。

答案 1 :(得分:2)

最简单的方法是使用TPL Dataflow

您可以定义一个接收异步委托流的“块”,并一次执行一个(等到每个委托完成后再开始下一个):

var block = new ActionBlock<Func<Task>>(func => func());

然后,发起一个Web服务请求:

block.Post(() => Task.Factory.FromAsync(...));

或(我更喜欢):

block.Post(() => client.GetStuffAsync(a, b, c));

如果您只想执行任务,ActionBlock方法就可以了。如果要生成输出流,请查看TransformBlock

var block = new TransformBlock<Func<Task<Stuff>>, Stuff>(func => func());

您以相同的方式触发请求,并且可以通过致电ReceiveReceiveAsync来获得结果。

答案 2 :(得分:0)

我已经决定在这里使用自定义解决方案......锁是杂乱无章的,但此刻,这完成了我想要的工作。

public interface ITaskExecutionQueue
{
    Task<TResult> QueueTask<TResult>(Func<Task<TResult>> taskGenerator);
    Task<TResult> QueueTask<TResult>(Task<Task<TResult>> taskGenerator);
    int OutstandingTaskCount { get; }
    event EventHandler OutstandingTaskCountChanged;
}

/// This class ensures that only a single Task is executed at any one time.  They are executed sequentially in order being queued.
/// The advantages of this class over OrderedTaskScheduler is that you can use any type of Task such as FromAsync (I/O Completion ports) 
/// which are not able to be scheduled using a traditional TaskScheduler.
/// Ensure that the `outer` tasks you queue are unstarted.  E.g. <![CDATA[
/// _taskExeQueue.QueueTask(new Task<Task<TResult>>(() => StartMyRealTask()));
/// ]]>
class OrderedTaskExecutionQueue : ITaskExecutionQueue
{
    private readonly Queue<Task> _queuedTasks = new Queue<Task>();
    private Task _currentTask;
    private readonly object _lockSync = new object();

    /// <summary>
    /// Queues a task for execution
    /// </summary>
    /// <typeparam name="TResult"></typeparam>
    /// <param name="taskGenerator">An unstarted Task that creates your started real-work task</param>
    /// <returns></returns>
    public Task<TResult> QueueTask<TResult>(Func<Task<TResult>> taskGenerator)
    {
        return QueueTask(new Task<Task<TResult>>(taskGenerator));
    }

    public Task<TResult> QueueTask<TResult>(Task<Task<TResult>> taskGenerator)
    {
        Task<TResult> unwrapped = taskGenerator.Unwrap();
        unwrapped.ContinueWith(_ =>
                               {
                                   EndTask();
                                   StartNextTaskIfQueued();
                               }, TaskContinuationOptions.ExecuteSynchronously);

        lock (_lockSync)
        {
            _queuedTasks.Enqueue(taskGenerator);

            if (_currentTask == null)
            {
                StartNextTaskIfQueued();
            }
        }

        TaskCompletionSource<TResult> tcs = new TaskCompletionSource<TResult>();
        tcs.TrySetFromTaskIncomplete(unwrapped);

        OutstandingTaskCountChanged.Raise(this);

        return tcs.Task;
    }

    private void EndTask()
    {
        lock (_lockSync)
        {
            _currentTask = null;
            _queuedTasks.Dequeue();
        }

        OutstandingTaskCountChanged.Raise(this);
    }

    private void StartNextTaskIfQueued()
    {
        lock (_lockSync)
        {
            if (_queuedTasks.Count > 0)
            {
                _currentTask = _queuedTasks.Peek();

                _currentTask.RunSynchronously();
            }
        }
    }

    /// <summary>
    /// Includes the currently executing task.
    /// </summary>
    public int OutstandingTaskCount
    {
        get
        {
            lock (_lockSync)
            {
                return _queuedTasks.Count;
            }
        }
    }

    public event EventHandler OutstandingTaskCountChanged;
}

接受一个未启动的Task<Task<TResult>> - 这允许队列决定何时执行它并开始FromAsync调用(这是内部任务)。用法:

Task<Task<TResult>> queueTask = new Task<Task<TResult>>(() => Task.Factory.FromAsync(beginAction, endAction));
Task<TResult> asyncCallTask = _taskExecutionQueue.QueueTask(queueTask);