|
1 | | -using System.Net.Sockets; |
| 1 | +using System.Net; |
| 2 | +using System.Net.Sockets; |
2 | 3 | using SerializersBenchmark.Base; |
3 | 4 | using SerializersBenchmark.Network.Abstractions; |
4 | 5 |
|
5 | 6 | namespace SerializersBenchmark.Network; |
6 | 7 |
|
7 | | -public sealed class Rabbit(ISerializerTestAsync serializer, int port, bool useBufferedStream = false) : IRabbit, ITcpClient |
| 8 | +public sealed class Rabbit : IRabbit, ITcpClient |
8 | 9 | { |
9 | | - private readonly TcpClient _tcpClient = new(); |
| 10 | + private readonly TcpClient _tcpClient; |
10 | 11 | private NetworkStream _networkStream; |
| 12 | + private readonly ISerializerTestAsync _serializer; |
| 13 | + private readonly int _port; |
| 14 | + private readonly bool _useBufferedStream; |
| 15 | + public Rabbit(ISerializerTestAsync serializer, int port, bool useBufferedStream = false) { |
| 16 | + _serializer = serializer; |
| 17 | + _port = port; |
| 18 | + _useBufferedStream = useBufferedStream; |
| 19 | + _tcpClient = new TcpClient(); |
| 20 | + _tcpClient.Client.NoDelay = true; //disable Nagle's algorithm for low latency |
| 21 | + } |
11 | 22 |
|
12 | | - public async Task ConnectAsync() |
13 | | - { |
14 | | - await _tcpClient.ConnectAsync("127.0.0.1", port); |
| 23 | + public async Task ConnectAsync() { |
| 24 | + await _tcpClient.ConnectAsync(IPAddress.Loopback, _port).ConfigureAwait(false); |
15 | 25 | _networkStream = _tcpClient.GetStream(); |
16 | 26 | } |
17 | 27 |
|
18 | 28 | public async Task SendAsync(object value, int repeatCount) |
19 | 29 | { |
20 | 30 | Stream stream = _networkStream; |
21 | | - if (useBufferedStream) |
| 31 | + if (_useBufferedStream) |
22 | 32 | { |
23 | 33 | stream = new BufferedStream(stream); |
24 | 34 | } |
25 | 35 |
|
26 | 36 | for (var i = 0; i < repeatCount; i++) |
27 | 37 | { |
28 | | - await serializer.SerializeAsync(value, stream).ConfigureAwait(false); |
| 38 | + await _serializer.SerializeAsync(value, stream).ConfigureAwait(false); |
29 | 39 | } |
30 | 40 |
|
31 | | - await _networkStream.FlushAsync().ConfigureAwait(false); |
| 41 | + await stream.FlushAsync().ConfigureAwait(false); |
32 | 42 | } |
33 | 43 |
|
34 | 44 | public async Task<object> ReceiveAsync(int expectedSize) |
35 | 45 | { |
36 | 46 | Stream stream = new LimitedStreamReader(_networkStream, expectedSize); |
37 | | - if (useBufferedStream) |
| 47 | + if (_useBufferedStream) |
38 | 48 | { |
39 | 49 | stream = new BufferedStream(stream); |
40 | 50 | } |
41 | | - return await serializer.DeserializeAsync(stream).ConfigureAwait(false); |
| 51 | + return await _serializer.DeserializeAsync(stream).ConfigureAwait(false); |
42 | 52 | } |
43 | 53 |
|
44 | 54 | public void Dispose() |
|
0 commit comments