foldcc/UniTask
1
0
Fork 0
mirror of https://github.com/Cysharp/UniTask.git synced 2026-05-14 19:10:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Force complete awaiting JobHandles when quitted in UnityEditor

Browse Source
This commit is contained in:
Ram.Type-0
2020-11-24 23:50:01 +09:00
parent e82353b4d9
commit fd70c031cb
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/UniTask/Assets/Plugins/UniTask/Runtime/PlayerLoopHelper.cs
View File

@@ -101,7 +101,7 @@ namespace Cysharp.Threading.Tasks
static SynchronizationContext unitySynchronizationContetext;
static ContinuationQueue[] yielders;
static PlayerLoopRunner[] runners;
internal static bool IsEditorApplicationQuitting { get; private set; }
static PlayerLoopSystem[] InsertRunner(PlayerLoopSystem loopSystem,
Type loopRunnerYieldType, ContinuationQueue cq, Type lastLoopRunnerYieldType, ContinuationQueue lastCq,
Type loopRunnerType, PlayerLoopRunner runner, Type lastLoopRunnerType, PlayerLoopRunner lastRunner)
@@ -112,6 +112,7 @@ namespace Cysharp.Threading.Tasks
{
if (state == PlayModeStateChange.EnteredEditMode || state == PlayModeStateChange.ExitingEditMode)
{
IsEditorApplicationQuitting = true;
// run rest action before clear.
if (runner != null)
{
@@ -134,6 +135,7 @@ namespace Cysharp.Threading.Tasks
lastCq.Run();
lastCq.Clear();
}
IsEditorApplicationQuitting = false;
}
};
#endif

3
src/UniTask/Assets/Plugins/UniTask/Runtime/UnityAsyncExtensions.Jobs.cs
View File

@@ -4,6 +4,7 @@
using System;
using System.Threading;
using Unity.Jobs;
using UnityEngine;
namespace Cysharp.Threading.Tasks
{
@@ -85,7 +86,7 @@ namespace Cysharp.Threading.Tasks
public bool MoveNext()
{
if (jobHandle.IsCompleted)
if (jobHandle.IsCompleted | PlayerLoopHelper.IsEditorApplicationQuitting)
{
jobHandle.Complete();
core.TrySetResult(AsyncUnit.Default);