pengtaihao
/
UniTask
mirror of https://github.com/Cysharp/UniTask
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

SwitchToMainThread, ReturnToMainThread, SwitchToSynchronizationContext, ReturnToSynchronizationContext supports CancellationToken

pull/115/head
neuecc 2020-06-28 20:57:42 +09:00
parent 529272d11b
commit 23997f0f93
1 changed files with 46 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.Threading.cs
View File

@ -15,33 +15,33 @@ namespace Cysharp.Threading.Tasks
/// <summary> /// <summary>
/// If running on mainthread, do nothing. Otherwise, same as UniTask.Yield(PlayerLoopTiming.Update). /// If running on mainthread, do nothing. Otherwise, same as UniTask.Yield(PlayerLoopTiming.Update).
/// </summary> /// </summary>
public static SwitchToMainThreadAwaitable SwitchToMainThread() public static SwitchToMainThreadAwaitable SwitchToMainThread(CancellationToken cancellationToken = default)
{ {
return new SwitchToMainThreadAwaitable(PlayerLoopTiming.Update); return new SwitchToMainThreadAwaitable(PlayerLoopTiming.Update, cancellationToken);
} }
/// <summary> /// <summary>
/// If running on mainthread, do nothing. Otherwise, same as UniTask.Yield(timing). /// If running on mainthread, do nothing. Otherwise, same as UniTask.Yield(timing).
/// </summary> /// </summary>
public static SwitchToMainThreadAwaitable SwitchToMainThread(PlayerLoopTiming timing) public static SwitchToMainThreadAwaitable SwitchToMainThread(PlayerLoopTiming timing, CancellationToken cancellationToken = default)
{ {
return new SwitchToMainThreadAwaitable(timing); return new SwitchToMainThreadAwaitable(timing, cancellationToken);
} }
/// <summary> /// <summary>
/// Return to mainthread(same as await SwitchToMainThread) after using scope is closed. /// Return to mainthread(same as await SwitchToMainThread) after using scope is closed.
/// </summary> /// </summary>
public static ReturnToMainThread ReturnToMainThread() public static ReturnToMainThread ReturnToMainThread(CancellationToken cancellationToken = default)
{ {
return new ReturnToMainThread(PlayerLoopTiming.Update); return new ReturnToMainThread(PlayerLoopTiming.Update, cancellationToken);
} }
/// <summary> /// <summary>
/// Return to mainthread(same as await SwitchToMainThread) after using scope is closed. /// Return to mainthread(same as await SwitchToMainThread) after using scope is closed.
/// </summary> /// </summary>
public static ReturnToMainThread ReturnToMainThread(PlayerLoopTiming timing) public static ReturnToMainThread ReturnToMainThread(PlayerLoopTiming timing, CancellationToken cancellationToken = default)
{ {
return new ReturnToMainThread(timing); return new ReturnToMainThread(timing, cancellationToken);
} }
/// <summary> /// <summary>
@ -67,20 +67,20 @@ namespace Cysharp.Threading.Tasks
return new SwitchToTaskPoolAwaitable(); return new SwitchToTaskPoolAwaitable();
} }
public static SwitchToSynchronizationContextAwaitable SwitchToSynchronizationContext(SynchronizationContext synchronizationContext) public static SwitchToSynchronizationContextAwaitable SwitchToSynchronizationContext(SynchronizationContext synchronizationContext, CancellationToken cancellationToken = default)
{ {
Error.ThrowArgumentNullException(synchronizationContext, nameof(synchronizationContext)); Error.ThrowArgumentNullException(synchronizationContext, nameof(synchronizationContext));
return new SwitchToSynchronizationContextAwaitable(synchronizationContext); return new SwitchToSynchronizationContextAwaitable(synchronizationContext, cancellationToken);
} }
public static ReturnToSynchronizationContext ReturnToSynchronizationContext(SynchronizationContext synchronizationContext) public static ReturnToSynchronizationContext ReturnToSynchronizationContext(SynchronizationContext synchronizationContext, CancellationToken cancellationToken = default)
{ {
return new ReturnToSynchronizationContext(synchronizationContext, false); return new ReturnToSynchronizationContext(synchronizationContext, false, cancellationToken);
} }
public static ReturnToSynchronizationContext ReturnToCurrentSynchronizationContext(bool dontPostWhenSameContext = true) public static ReturnToSynchronizationContext ReturnToCurrentSynchronizationContext(bool dontPostWhenSameContext = true, CancellationToken cancellationToken = default)
{ {
return new ReturnToSynchronizationContext(SynchronizationContext.Current, dontPostWhenSameContext); return new ReturnToSynchronizationContext(SynchronizationContext.Current, dontPostWhenSameContext, cancellationToken);
} }
} }
@ -89,21 +89,25 @@ namespace Cysharp.Threading.Tasks
public struct SwitchToMainThreadAwaitable public struct SwitchToMainThreadAwaitable
{ {
readonly PlayerLoopTiming playerLoopTiming; readonly PlayerLoopTiming playerLoopTiming;
readonly CancellationToken cancellationToken;
public SwitchToMainThreadAwaitable(PlayerLoopTiming playerLoopTiming) public SwitchToMainThreadAwaitable(PlayerLoopTiming playerLoopTiming, CancellationToken cancellationToken)
{ {
this.playerLoopTiming = playerLoopTiming; this.playerLoopTiming = playerLoopTiming;
this.cancellationToken = cancellationToken;
} }
public Awaiter GetAwaiter() => new Awaiter(playerLoopTiming); public Awaiter GetAwaiter() => new Awaiter(playerLoopTiming, cancellationToken);
public struct Awaiter : ICriticalNotifyCompletion public struct Awaiter : ICriticalNotifyCompletion
{ {
readonly PlayerLoopTiming playerLoopTiming; readonly PlayerLoopTiming playerLoopTiming;
readonly CancellationToken cancellationToken;
public Awaiter(PlayerLoopTiming playerLoopTiming) public Awaiter(PlayerLoopTiming playerLoopTiming, CancellationToken cancellationToken)
{ {
this.playerLoopTiming = playerLoopTiming; this.playerLoopTiming = playerLoopTiming;
this.cancellationToken = cancellationToken;
} }
public bool IsCompleted public bool IsCompleted
@ -122,7 +126,7 @@ namespace Cysharp.Threading.Tasks
} }
} }
public void GetResult() { } public void GetResult() { cancellationToken.ThrowIfCancellationRequested(); }
public void OnCompleted(Action continuation) public void OnCompleted(Action continuation)
{ {
@ -139,31 +143,35 @@ namespace Cysharp.Threading.Tasks
public struct ReturnToMainThread public struct ReturnToMainThread
{ {
readonly PlayerLoopTiming playerLoopTiming; readonly PlayerLoopTiming playerLoopTiming;
readonly CancellationToken cancellationToken;
public ReturnToMainThread(PlayerLoopTiming playerLoopTiming) public ReturnToMainThread(PlayerLoopTiming playerLoopTiming, CancellationToken cancellationToken)
{ {
this.playerLoopTiming = playerLoopTiming; this.playerLoopTiming = playerLoopTiming;
this.cancellationToken = cancellationToken;
} }
public Awaiter DisposeAsync() public Awaiter DisposeAsync()
{ {
return new Awaiter(playerLoopTiming); // run immediate. return new Awaiter(playerLoopTiming, cancellationToken); // run immediate.
} }
public readonly struct Awaiter : ICriticalNotifyCompletion public readonly struct Awaiter : ICriticalNotifyCompletion
{ {
readonly PlayerLoopTiming timing; readonly PlayerLoopTiming timing;
readonly CancellationToken cancellationToken;
public Awaiter(PlayerLoopTiming timing) public Awaiter(PlayerLoopTiming timing, CancellationToken cancellationToken)
{ {
this.timing = timing; this.timing = timing;
this.cancellationToken = cancellationToken;
} }
public Awaiter GetAwaiter() => this; public Awaiter GetAwaiter() => this;
public bool IsCompleted => PlayerLoopHelper.MainThreadId == System.Threading.Thread.CurrentThread.ManagedThreadId; public bool IsCompleted => PlayerLoopHelper.MainThreadId == System.Threading.Thread.CurrentThread.ManagedThreadId;
public void GetResult() { } public void GetResult() { cancellationToken.ThrowIfCancellationRequested(); }
public void OnCompleted(Action continuation) public void OnCompleted(Action continuation)
{ {
@ -285,26 +293,30 @@ namespace Cysharp.Threading.Tasks
public struct SwitchToSynchronizationContextAwaitable public struct SwitchToSynchronizationContextAwaitable
{ {
readonly SynchronizationContext synchronizationContext; readonly SynchronizationContext synchronizationContext;
readonly CancellationToken cancellationToken;
public SwitchToSynchronizationContextAwaitable(SynchronizationContext synchronizationContext) public SwitchToSynchronizationContextAwaitable(SynchronizationContext synchronizationContext, CancellationToken cancellationToken)
{ {
this.synchronizationContext = synchronizationContext; this.synchronizationContext = synchronizationContext;
this.cancellationToken = cancellationToken;
} }
public Awaiter GetAwaiter() => new Awaiter(synchronizationContext); public Awaiter GetAwaiter() => new Awaiter(synchronizationContext, cancellationToken);
public struct Awaiter : ICriticalNotifyCompletion public struct Awaiter : ICriticalNotifyCompletion
{ {
static readonly SendOrPostCallback switchToCallback = Callback; static readonly SendOrPostCallback switchToCallback = Callback;
readonly SynchronizationContext synchronizationContext; readonly SynchronizationContext synchronizationContext;
readonly CancellationToken cancellationToken;
public Awaiter(SynchronizationContext synchronizationContext) public Awaiter(SynchronizationContext synchronizationContext, CancellationToken cancellationToken)
{ {
this.synchronizationContext = synchronizationContext; this.synchronizationContext = synchronizationContext;
this.cancellationToken = cancellationToken;
} }
public bool IsCompleted => false; public bool IsCompleted => false;
public void GetResult() { } public void GetResult() { cancellationToken.ThrowIfCancellationRequested(); }
public void OnCompleted(Action continuation) public void OnCompleted(Action continuation)
{ {
@ -328,16 +340,18 @@ namespace Cysharp.Threading.Tasks
{ {
readonly SynchronizationContext syncContext; readonly SynchronizationContext syncContext;
readonly bool dontPostWhenSameContext; readonly bool dontPostWhenSameContext;
readonly CancellationToken cancellationToken;
public ReturnToSynchronizationContext(SynchronizationContext syncContext, bool dontPostWhenSameContext) public ReturnToSynchronizationContext(SynchronizationContext syncContext, bool dontPostWhenSameContext, CancellationToken cancellationToken)
{ {
this.syncContext = syncContext; this.syncContext = syncContext;
this.dontPostWhenSameContext = dontPostWhenSameContext; this.dontPostWhenSameContext = dontPostWhenSameContext;
this.cancellationToken = cancellationToken;
} }
public Awaiter DisposeAsync() public Awaiter DisposeAsync()
{ {
return new Awaiter(syncContext, dontPostWhenSameContext); return new Awaiter(syncContext, dontPostWhenSameContext, cancellationToken);
} }
public struct Awaiter : ICriticalNotifyCompletion public struct Awaiter : ICriticalNotifyCompletion
@ -346,11 +360,13 @@ namespace Cysharp.Threading.Tasks
readonly SynchronizationContext synchronizationContext; readonly SynchronizationContext synchronizationContext;
readonly bool dontPostWhenSameContext; readonly bool dontPostWhenSameContext;
readonly CancellationToken cancellationToken;
public Awaiter(SynchronizationContext synchronizationContext, bool dontPostWhenSameContext) public Awaiter(SynchronizationContext synchronizationContext, bool dontPostWhenSameContext, CancellationToken cancellationToken)
{ {
this.synchronizationContext = synchronizationContext; this.synchronizationContext = synchronizationContext;
this.dontPostWhenSameContext = dontPostWhenSameContext; this.dontPostWhenSameContext = dontPostWhenSameContext;
this.cancellationToken = cancellationToken;
} }
public Awaiter GetAwaiter() => this; public Awaiter GetAwaiter() => this;
@ -373,7 +389,7 @@ namespace Cysharp.Threading.Tasks
} }
} }
public void GetResult() { } public void GetResult() { cancellationToken.ThrowIfCancellationRequested(); }
public void OnCompleted(Action continuation) public void OnCompleted(Action continuation)
{ {