ValueTaskAwaitAdapter
using System;
using System.Runtime.CompilerServices;
using System.Threading.Tasks;
namespace NUnit.Framework.Internal
{
internal static class ValueTaskAwaitAdapter
{
private sealed class NonGenericAdapter : DefaultBlockingAwaitAdapter
{
private readonly ValueTaskAwaiter _awaiter;
public override bool IsCompleted => _awaiter.IsCompleted;
public NonGenericAdapter(ValueTask task)
{
_awaiter = task.GetAwaiter();
}
[NullableContext(1)]
public override void OnCompleted(Action action)
{
_awaiter.UnsafeOnCompleted(action);
}
[NullableContext(2)]
public override object GetResult()
{
_awaiter.GetResult();
return null;
}
}
[NullableContext(2)]
[Nullable(0)]
private sealed class GenericAdapter<T> : DefaultBlockingAwaitAdapter
{
[Nullable(new byte[] {
0,
1
})]
private readonly ValueTaskAwaiter<T> _awaiter;
public override bool IsCompleted => _awaiter.IsCompleted;
public GenericAdapter([Nullable(new byte[] {
0,
1
})] ValueTask<T> task)
{
_awaiter = task.GetAwaiter();
}
[NullableContext(1)]
public override void OnCompleted(Action action)
{
_awaiter.UnsafeOnCompleted(action);
}
public override object GetResult()
{
return _awaiter.GetResult();
}
}
[NullableContext(1)]
public static AwaitAdapter Create(ValueTask task)
{
return new NonGenericAdapter(task);
}
[NullableContext(1)]
[return: Nullable(2)]
public static AwaitAdapter TryCreate(object task)
{
Type type = task.GetType();
if (type.IsGenericType && type.GetGenericTypeDefinition() == typeof(ValueTask<>))
return (AwaitAdapter)typeof(GenericAdapter<>).MakeGenericType(type.GetGenericArguments()[0]).GetConstructor(new Type[1] {
type
}).Invoke(new object[1] {
task
});
return null;
}
}
}