OpenTween/TaskCollection.cs

   1 // OpenTween - Client of Twitter
   2 // Copyright (c) 2023 kim_upsilon (@kim_upsilon) <https://upsilo.net/~upsilon/>
   3 // All rights reserved.
   4 //
   5 // This file is part of OpenTween.
   6 //
   7 // This program is free software; you can redistribute it and/or modify it
   8 // under the terms of the GNU General Public License as published by the Free
   9 // Software Foundation; either version 3 of the License, or (at your option)
  10 // any later version.
  11 //
  12 // This program is distributed in the hope that it will be useful, but
  13 // WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  14 // or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
  15 // for more details.
  16 //
  17 // You should have received a copy of the GNU General Public License along
  18 // with this program. If not, see <http://www.gnu.org/licenses/>, or write to
  19 // the Free Software Foundation, Inc., 51 Franklin Street - Fifth Floor,
  20 // Boston, MA 02110-1301, USA.
  21
  22 #nullable enable
  23
  24 using System;
  25 using System.Collections.Generic;
  26 using System.Linq;
  27 using System.Threading.Tasks;
  28
  29 namespace OpenTween
  30 {
  31     public class TaskCollection
  32     {
  33         private readonly List<Func<Task>> taskFuncs;
  34         private Func<Exception, bool> ignoreExceptionFunc = _ => true;
  35
  36         public TaskCollection()
  37             => this.taskFuncs = new();
  38
  39         public TaskCollection(int capacity)
  40             => this.taskFuncs = new(capacity);
  41
  42         public void Add(Func<Task> func)
  43             => this.taskFuncs.Add(func);
  44
  45         public void Add(IEnumerable<Func<Task>> tasks)
  46             => this.taskFuncs.AddRange(tasks);
  47
  48         public TaskCollection IgnoreException(Func<Exception, bool> condition)
  49         {
  50             this.ignoreExceptionFunc = condition;
  51             return this;
  52         }
  53
  54         public Task RunAll()
  55             => this.RunAll(runOnThreadPool: false);
  56
  57         public Task RunAll(bool runOnThreadPool)
  58             => Task.WhenAll(this.taskFuncs.Select(x => this.WrapAsyncFunc(x, runOnThreadPool)));
  59
  60         private Task WrapAsyncFunc(Func<Task> func, bool runOnThreadPool)
  61         {
  62             Task WrappedFunc()
  63                 => AsyncExceptionBoundary.Wrap(func, this.ignoreExceptionFunc);
  64
  65             if (runOnThreadPool)
  66                 return Task.Run(WrappedFunc);
  67             else
  68                 return WrappedFunc();
  69         }
  70     }
  71 }