| | 1 | | using System; |
| | 2 | | using System.Collections; |
| | 3 | | using System.Threading; |
| | 4 | | using Cysharp.Threading.Tasks; |
| | 5 | |
|
| | 6 | | namespace DCL.Helpers |
| | 7 | | { |
| | 8 | | public static class TaskUtils |
| | 9 | | { |
| 0 | 10 | | private static BaseVariable<bool> multithreading => DataStore.i.performance.multithreading; |
| | 11 | | public static async UniTask Run(Action action, CancellationToken cancellationToken = default, bool returnToMainT |
| | 12 | | { |
| 1233 | 13 | | if (Configuration.EnvironmentSettings.RUNNING_TESTS) |
| | 14 | | { |
| 1233 | 15 | | action(); |
| 1233 | 16 | | return; |
| | 17 | | } |
| | 18 | |
|
| 0 | 19 | | cancellationToken.ThrowIfCancellationRequested(); |
| | 20 | |
|
| 0 | 21 | | if (multithreading.Get()) |
| | 22 | | { |
| 0 | 23 | | await UniTask.RunOnThreadPool(action, returnToMainThread, cancellationToken); |
| 0 | 24 | | } |
| | 25 | | else |
| | 26 | | { |
| 0 | 27 | | action(); |
| | 28 | | } |
| 1233 | 29 | | } |
| | 30 | |
|
| | 31 | | public static async UniTask Run(Func<UniTask> action, CancellationToken cancellationToken = default, bool return |
| | 32 | | { |
| 31 | 33 | | if (Configuration.EnvironmentSettings.RUNNING_TESTS) |
| | 34 | | { |
| 31 | 35 | | action(); |
| 31 | 36 | | return; |
| | 37 | | } |
| | 38 | |
|
| 0 | 39 | | cancellationToken.ThrowIfCancellationRequested(); |
| | 40 | |
|
| 0 | 41 | | if (multithreading.Get()) |
| | 42 | | { |
| 0 | 43 | | await UniTask.RunOnThreadPool(action, returnToMainThread, cancellationToken); |
| 0 | 44 | | } |
| | 45 | | else |
| | 46 | | { |
| 0 | 47 | | await UniTask.Create(action).AttachExternalCancellation(cancellationToken); |
| | 48 | | } |
| 31 | 49 | | } |
| | 50 | |
|
| | 51 | | public static async UniTask<T> Run<T>(Func<UniTask<T>> action, CancellationToken cancellationToken = default, bo |
| | 52 | | { |
| | 53 | | cancellationToken.ThrowIfCancellationRequested(); |
| | 54 | |
|
| | 55 | | if (multithreading.Get()) |
| | 56 | | { |
| | 57 | | return await UniTask.RunOnThreadPool(action, returnToMainThread, cancellationToken); |
| | 58 | | } |
| | 59 | |
|
| | 60 | | return await UniTask.Create(action).AttachExternalCancellation(cancellationToken); |
| | 61 | | } |
| | 62 | |
|
| | 63 | | public static async UniTask<T> Run<T>(Func<T> action, CancellationToken cancellationToken = default, bool return |
| | 64 | | { |
| | 65 | | if (Configuration.EnvironmentSettings.RUNNING_TESTS) |
| | 66 | | { |
| | 67 | | return action(); |
| | 68 | | } |
| | 69 | |
|
| | 70 | | cancellationToken.ThrowIfCancellationRequested(); |
| | 71 | |
|
| | 72 | | if (multithreading.Get()) |
| | 73 | | { |
| | 74 | | return await UniTask.RunOnThreadPool(action, returnToMainThread, cancellationToken); |
| | 75 | | } |
| | 76 | |
|
| | 77 | | return action(); |
| | 78 | | } |
| | 79 | |
|
| | 80 | | public static async UniTask RunThrottledCoroutine(IEnumerator enumerator, Action<Exception> onFail, Func<double, |
| | 81 | | { |
| 468 | 82 | | IEnumerator routine = DCLCoroutineRunner.Run(enumerator, onFail, timeBudget); |
| 1404 | 83 | | await routine.ToUniTask(CoroutineStarter.instance); |
| 468 | 84 | | } |
| | 85 | |
|
| | 86 | | } |
| | 87 | | } |