c# - 以秒为单位限制 Task.Factory.Start 中的任务数

Question

我看过很多关于一次限制任务数量的帖子（System.Threading.Tasks - 限制并发任务的数量是一个很好的帖子）。

但是，我需要按秒限制任务数量 - 每秒只有 X 个任务？有没有简单的方法来实现这一点？

我考虑过创建一个 ConcurrentDictionary，键是当前的第二个，第二个是到目前为止的计数。检查我们当前是否处于 20 秒，然后停止。这似乎不是最理想的。

我宁愿每 1 秒/20 秒执行一次任务。有什么想法吗？

Answer 1

我认为，这可以作为一个起点。下面的示例创建 50 个任务（运行 5 个任务/秒）。

这不会阻止创建任务的代码。如果您想在所有任务都安排好之前阻止调用者，那么您可以Task.Delay((int)shouldWait).Wait()使用QueueTask

TaskFactory taskFactory = new TaskFactory(new TimeLimitedTaskScheduler(5));

for (int i = 0; i < 50; i++)
{
    var x = taskFactory.StartNew<int>(() => DateTime.Now.Second)
                        .ContinueWith(t => Console.WriteLine(t.Result));
}

Console.WriteLine("End of Loop");

---

public class TimeLimitedTaskScheduler : TaskScheduler
{
    int _TaskCount = 0;
    Stopwatch _Sw = null;
    int _MaxTasksPerSecond;

    public TimeLimitedTaskScheduler(int maxTasksPerSecond)
    {
        _MaxTasksPerSecond = maxTasksPerSecond;
    }

    protected override void QueueTask(Task task)
    {
        if (_TaskCount == 0) _Sw = Stopwatch.StartNew();

        var shouldWait = (1000 / _MaxTasksPerSecond) * _TaskCount - _Sw.ElapsedMilliseconds;

        if (shouldWait < 0)
        {
            shouldWait = _TaskCount = 0;
            _Sw.Restart();
        }

        Task.Delay((int)shouldWait)
            .ContinueWith(t => ThreadPool.QueueUserWorkItem((_) => base.TryExecuteTask(task)));

        _TaskCount++;
    }

    protected override bool TryExecuteTaskInline(Task task, bool taskWasPreviouslyQueued)
    {
        return base.TryExecuteTask(task);
    }

    protected override IEnumerable<Task> GetScheduledTasks()
    {
        throw new NotImplementedException();
    }


}