/
MessageStream.cs
84 lines (67 loc) · 2.22 KB
/
MessageStream.cs
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
using System;
using System.Collections.Immutable;
namespace HotChocolate.Transport.Sockets.Client.Protocols;
internal sealed class MessageStream : IObservable<IOperationMessage>, IObserver<IOperationMessage>
{
private readonly object _sync = new();
private ImmutableList<Subscription> _subscriptions = ImmutableList<Subscription>.Empty;
public IDisposable Subscribe(IObserver<IOperationMessage> observer)
{
var subscription = new Subscription(observer);
subscription.Disposed += (_, _) => Unsubscribe(subscription);
lock (_sync)
{
_subscriptions = _subscriptions.Add(subscription);
}
return subscription;
}
private void Unsubscribe(Subscription subscription)
{
lock (_sync)
{
_subscriptions = _subscriptions.Remove(subscription);
}
}
public void OnNext(IOperationMessage value) => OnNext(value, _subscriptions);
private void OnNext(IOperationMessage value, ImmutableList<Subscription> subscriptions)
{
foreach (var subscription in subscriptions)
{
subscription.Observer.OnNext(value);
}
}
public void OnError(Exception error) => OnError(error, _subscriptions);
private void OnError(Exception error, ImmutableList<Subscription> subscriptions)
{
foreach (var subscription in subscriptions)
{
subscription.Observer.OnError(error);
}
}
public void OnCompleted() => OnCompleted(_subscriptions);
private void OnCompleted(ImmutableList<Subscription> subscriptions)
{
foreach (var subscription in subscriptions)
{
subscription.Observer.OnCompleted();
}
}
private sealed class Subscription : IDisposable
{
private bool _disposed;
public event EventHandler? Disposed;
public Subscription(IObserver<IOperationMessage> observer)
{
Observer = observer;
}
public IObserver<IOperationMessage> Observer { get; }
public void Dispose()
{
if (!_disposed)
{
Disposed?.Invoke(this, EventArgs.Empty);
_disposed = true;
}
}
}
}