三国卡牌客户端基础资源仓库
hch
4 天以前 cdac25a8e4f91a4663bf5f80994538dd263b757c
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
138
139
140
141
142
143
144
145
146
147
148
149
150
151
using Cysharp.Threading.Tasks.Internal;
using System;
using System.Threading;
 
namespace Cysharp.Threading.Tasks.Linq
{
    public static partial class UniTaskAsyncEnumerable
    {
        public static IUniTaskAsyncEnumerable<TSource> Append<TSource>(this IUniTaskAsyncEnumerable<TSource> source, TSource element)
        {
            Error.ThrowArgumentNullException(source, nameof(source));
 
            return new AppendPrepend<TSource>(source, element, true);
        }
 
        public static IUniTaskAsyncEnumerable<TSource> Prepend<TSource>(this IUniTaskAsyncEnumerable<TSource> source, TSource element)
        {
            Error.ThrowArgumentNullException(source, nameof(source));
 
            return new AppendPrepend<TSource>(source, element, false);
        }
    }
 
    internal sealed class AppendPrepend<TSource> : IUniTaskAsyncEnumerable<TSource>
    {
        readonly IUniTaskAsyncEnumerable<TSource> source;
        readonly TSource element;
        readonly bool append; // or prepend
 
        public AppendPrepend(IUniTaskAsyncEnumerable<TSource> source, TSource element, bool append)
        {
            this.source = source;
            this.element = element;
            this.append = append;
        }
 
        public IUniTaskAsyncEnumerator<TSource> GetAsyncEnumerator(CancellationToken cancellationToken = default)
        {
            return new _AppendPrepend(source, element, append, cancellationToken);
        }
 
        sealed class _AppendPrepend : MoveNextSource, IUniTaskAsyncEnumerator<TSource>
        {
            enum State : byte
            {
                None,
                RequirePrepend,
                RequireAppend,
                Completed
            }
 
            static readonly Action<object> MoveNextCoreDelegate = MoveNextCore;
 
            readonly IUniTaskAsyncEnumerable<TSource> source;
            readonly TSource element;
            CancellationToken cancellationToken;
 
            State state;
            IUniTaskAsyncEnumerator<TSource> enumerator;
            UniTask<bool>.Awaiter awaiter;
 
            public _AppendPrepend(IUniTaskAsyncEnumerable<TSource> source, TSource element, bool append, CancellationToken cancellationToken)
            {
                this.source = source;
                this.element = element;
                this.state = append ? State.RequireAppend : State.RequirePrepend;
                this.cancellationToken = cancellationToken;
 
                TaskTracker.TrackActiveTask(this, 3);
            }
 
            public TSource Current { get; private set; }
 
 
            public UniTask<bool> MoveNextAsync()
            {
                cancellationToken.ThrowIfCancellationRequested();
                completionSource.Reset();
 
                if (enumerator == null)
                {
                    if (state == State.RequirePrepend)
                    {
                        Current = element;
                        state = State.None;
                        return CompletedTasks.True;
                    }
 
                    enumerator = source.GetAsyncEnumerator(cancellationToken);
                }
 
                if (state == State.Completed)
                {
                    return CompletedTasks.False;
                }
 
                awaiter = enumerator.MoveNextAsync().GetAwaiter();
 
                if (awaiter.IsCompleted)
                {
                    MoveNextCoreDelegate(this);
                }
                else
                {
                    awaiter.SourceOnCompleted(MoveNextCoreDelegate, this);
                }
 
                return new UniTask<bool>(this, completionSource.Version);
            }
 
            static void MoveNextCore(object state)
            {
                var self = (_AppendPrepend)state;
 
                if (self.TryGetResult(self.awaiter, out var result))
                {
                    if (result)
                    {
                        self.Current = self.enumerator.Current;
                        self.completionSource.TrySetResult(true);
                    }
                    else
                    {
                        if (self.state == State.RequireAppend)
                        {
                            self.state = State.Completed;
                            self.Current = self.element;
                            self.completionSource.TrySetResult(true);
                        }
                        else
                        {
                            self.state = State.Completed;
                            self.completionSource.TrySetResult(false);
                        }
                    }
                }
            }
 
            public UniTask DisposeAsync()
            {
                TaskTracker.RemoveTracking(this);
                if (enumerator != null)
                {
                    return enumerator.DisposeAsync();
                }
                return default;
            }
        }
    }
 
}