using BenchmarkDotNet.Attributes;
using BenchmarkDotNet.Running;
using System.IO;
using System.Net;
using System.Net.Security;
using System.Net.Sockets;
using System.Threading.Tasks;

[MemoryDiagnoser]
public class Program
{
    static void Main(string[] args) => BenchmarkSwitcher.FromAssemblies(new[] { typeof(Program).Assembly }).Run(args);

    private Stream _client, _server;

    [GlobalSetup]
    public void ReadWriteSetup()
    {
        using (var listener = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp))
        {
            listener.Bind(new IPEndPoint(IPAddress.Loopback, 0));
            listener.Listen(1);

            var client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
            client.Connect(listener.LocalEndPoint);
            var server = listener.Accept();
            var clientStream = new NegotiateStream(new NetworkStream(client, ownsSocket: true), leaveInnerStreamOpen: true);
            var serverStream = new NegotiateStream(new NetworkStream(server, ownsSocket: true), leaveInnerStreamOpen: true);

            Task ca = clientStream.AuthenticateAsClientAsync();
            Task sa = serverStream.AuthenticateAsServerAsync();
            Task.WaitAll(ca, sa);

            _client = clientStream;
            _server = serverStream;
        }
    }

    private byte[] _oneByteBuffer = new byte[1];

    [Benchmark]
    public async Task WriteRead()
    {
        for (int i = 0; i < 1_000; i++)
        {
            await _server.WriteAsync(_oneByteBuffer);
            await _client.ReadAsync(_oneByteBuffer);
        }
    }

    [Benchmark]
    public async Task ReadWrite()
    {
        for (int i = 0; i < 1_000; i++)
        {
            var r = _client.ReadAsync(_oneByteBuffer);
            await _server.WriteAsync(_oneByteBuffer);
            await r;
        }
    }
}