| | | 1 | | using Cysharp.Threading.Tasks; |
| | | 2 | | using System; |
| | | 3 | | using System.Threading; |
| | | 4 | | |
| | | 5 | | namespace DCL.Helpers |
| | | 6 | | { |
| | | 7 | | public interface IAsyncEnumerableWithEvent<T> : IUniTaskAsyncEnumerable<T> |
| | | 8 | | { |
| | | 9 | | void AddListener(Action<T> callback); |
| | | 10 | | void RemoveListener(Action<T> callback); |
| | | 11 | | } |
| | | 12 | | |
| | | 13 | | public class AsyncEnumerableWithEvent<T> : IAsyncEnumerableWithEvent<T>, IDisposable |
| | | 14 | | { |
| | 0 | 15 | | private readonly AsyncReactiveProperty<T> activeProperty = new (default); |
| | | 16 | | private Action<T> callback; |
| | | 17 | | |
| | | 18 | | public void AddListener(Action<T> callback) |
| | | 19 | | { |
| | 0 | 20 | | this.callback += callback; |
| | 0 | 21 | | } |
| | | 22 | | |
| | | 23 | | public void RemoveListener(Action<T> callback) |
| | | 24 | | { |
| | 0 | 25 | | this.callback -= callback; |
| | 0 | 26 | | } |
| | | 27 | | |
| | | 28 | | public void Write(T item) |
| | | 29 | | { |
| | 0 | 30 | | callback?.Invoke(item); |
| | 0 | 31 | | activeProperty.Value = item; |
| | 0 | 32 | | } |
| | | 33 | | |
| | | 34 | | public IUniTaskAsyncEnumerator<T> GetAsyncEnumerator(CancellationToken cancellationToken = new ()) => |
| | 0 | 35 | | activeProperty.WithoutCurrent().GetAsyncEnumerator(cancellationToken); |
| | | 36 | | |
| | | 37 | | public void Dispose() |
| | | 38 | | { |
| | 0 | 39 | | activeProperty?.Dispose(); |
| | 0 | 40 | | callback = null; |
| | 0 | 41 | | } |
| | | 42 | | } |
| | | 43 | | } |