forked from ClearFoundry/ClearScript
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathEnumerableHelpers.NetCore.cs
More file actions
137 lines (108 loc) · 4.01 KB
/
Copy pathEnumerableHelpers.NetCore.cs
File metadata and controls
137 lines (108 loc) · 4.01 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license.
using System.Collections;
using System.Collections.Generic;
using System.Threading.Tasks;
using Microsoft.ClearScript.JavaScript;
namespace Microsoft.ClearScript.Util
{
/// <exclude/>
public interface IAsyncEnumeratorPromise<out T> : IAsyncEnumerator<T>
{
/// <exclude/>
object MoveNextPromise();
}
internal static partial class EnumerableHelpers
{
public static IAsyncEnumeratorPromise<T> ToAsyncEnumerator<T>(this IEnumerator<T> enumerator, ScriptEngine engine)
{
return new AsyncEnumeratorPromiseOnEnumerator<T>(engine, enumerator);
}
public static IAsyncEnumeratorPromise<object> ToAsyncEnumerator(this IEnumerator enumerator, ScriptEngine engine)
{
return new AsyncEnumeratorPromiseOnEnumerator(engine, enumerator);
}
public static IAsyncEnumeratorPromise<T> GetAsyncEnumerator<T>(IAsyncEnumerable<T> source, ScriptEngine engine)
{
return new AsyncEnumeratorPromiseOnAsyncEnumerator<T>(engine, source.GetAsyncEnumerator());
}
public static IAsyncEnumeratorPromise<T> GetAsyncEnumerator<T>(IEnumerable<T> source, ScriptEngine engine)
{
return source.GetEnumerator().ToAsyncEnumerator(engine);
}
public static IAsyncEnumeratorPromise<object> GetAsyncEnumerator(IEnumerable source, ScriptEngine engine)
{
return source.GetEnumerator().ToAsyncEnumerator(engine);
}
}
internal abstract class AsyncEnumeratorPromise<T> : IAsyncEnumeratorPromise<T>
{
private readonly ScriptEngine engine;
protected AsyncEnumeratorPromise(ScriptEngine engine)
{
this.engine = engine;
}
public abstract T Current { get; }
public abstract ValueTask<bool> MoveNextAsync();
public object MoveNextPromise()
{
return MoveNextAsync().ToPromise(engine);
}
public abstract ValueTask DisposeAsync();
}
internal sealed class AsyncEnumeratorPromiseOnAsyncEnumerator<T> : AsyncEnumeratorPromise<T>
{
private readonly IAsyncEnumerator<T> enumerator;
public AsyncEnumeratorPromiseOnAsyncEnumerator(ScriptEngine engine, IAsyncEnumerator<T> enumerator)
: base(engine)
{
this.enumerator = enumerator;
}
public override T Current => enumerator.Current;
public override ValueTask<bool> MoveNextAsync()
{
return enumerator.MoveNextAsync();
}
public override ValueTask DisposeAsync()
{
return enumerator.DisposeAsync();
}
}
internal sealed class AsyncEnumeratorPromiseOnEnumerator<T> : AsyncEnumeratorPromise<T>
{
private readonly IEnumerator<T> enumerator;
public AsyncEnumeratorPromiseOnEnumerator(ScriptEngine engine, IEnumerator<T> enumerator)
: base(engine)
{
this.enumerator = enumerator;
}
public override T Current => enumerator.Current;
public override ValueTask<bool> MoveNextAsync()
{
return new ValueTask<bool>(enumerator.MoveNext());
}
public override ValueTask DisposeAsync()
{
enumerator.Dispose();
return default;
}
}
internal sealed class AsyncEnumeratorPromiseOnEnumerator : AsyncEnumeratorPromise<object>
{
private readonly IEnumerator enumerator;
public AsyncEnumeratorPromiseOnEnumerator(ScriptEngine engine, IEnumerator enumerator)
: base(engine)
{
this.enumerator = enumerator;
}
public override object Current => enumerator.Current;
public override ValueTask<bool> MoveNextAsync()
{
return new ValueTask<bool>(enumerator.MoveNext());
}
public override ValueTask DisposeAsync()
{
return default;
}
}
}