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

replace all promisepool to new taskpool system

Browse Source
This commit is contained in:
neuecc
2020-05-29 03:43:18 +09:00
parent d5db96b913
commit 21bf08a6b3
19 changed files with 467 additions and 281 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/UniTask.NetCore/NetCore/UniTask.Yield.cs
View File

@@ -59,16 +59,22 @@ namespace Cysharp.Threading.Tasks
#if NETCOREAPP3_1
sealed class ThreadPoolWorkItem : IThreadPoolWorkItem
sealed class ThreadPoolWorkItem : IThreadPoolWorkItem, ITaskPoolNode<ThreadPoolWorkItem>
{
static readonly ConcurrentQueue<ThreadPoolWorkItem> pool = new ConcurrentQueue<ThreadPoolWorkItem>();
static TaskPool<ThreadPoolWorkItem> pool;
public ThreadPoolWorkItem NextNode { get; set; }
static ThreadPoolWorkItem()
{
TaskPoolMonitor.RegisterSizeGetter(typeof(ThreadPoolWorkItem), () => pool.Size);
}
Action continuation;
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static ThreadPoolWorkItem Create(Action continuation)
{
if (!pool.TryDequeue(out var item))
if (!pool.TryPop(out var item))
{
item = new ThreadPoolWorkItem();
}
@@ -82,9 +88,11 @@ namespace Cysharp.Threading.Tasks
{
var call = continuation;
continuation = null;
pool.Enqueue(this);
call.Invoke();
if (call != null)
{
pool.TryPush(this);
call.Invoke();
}
}
}