C# TcpClient遇ConnectionAborted错误后首次读取极慢问题求助

问题场景 #

我正在编写与TCP/IP设备交互的代码，设备会在15秒无活动时发送FIN报文断开连接。我用Polly实现了重试策略，触发重试前会释放当前TcpClient实例、新建实例并重连目标IP/端口。正常读取响应时重试策略能正常工作，但当因ConnectionAborted套接字异常触发重试时，读取字节到缓冲区的操作会耗时约15秒。

相关代码 #

// Retry policy and the TcpClient are private member variables in this case
_readRetryPolicy = Policy.Handle<IOException>()
    .RetryAsync(3, onRetryAsync: async (_, _) =>
    {
        Disconnect();
        await ConnectAsync(_ip!, _port).ConfigureAwait(false);
    });

public async Task<byte[]> ReadResponseAsync()
{
    if (_tcpClient is not TcpClient { Connected: true })
    {
        Disconnect();
        await ConnectAsync(_ip!, _port).ConfigureAwait(false);

        // Possible StackOverflowException but I'm just hacking at this problem right now
        return await ReadResponseAsync().ConfigureAwait(false);
    }

    var buffer = new byte[32];
    var responseBytesStream = Enumerable.Empty<byte>();
    int numBytesRead = 0;
    do
    {
        numBytesRead = await _readRetryPolicy.ExecuteAsync(() => _tcpClient.GetStream().ReadAsync(buffer).AsTask()).ConfigureAwait(false);
        responseBytesStream = responseBytesStream.Concat(buffer);
    }
    while (_tcpClient.GetStream().DataAvailable);

    return responseBytesStream.ToArray();
}

public void Disconnect()
{
    if (_tcpClient is TcpClient c) c.Dispose();
    _tcpClient = null;
}

问题原因 #

1. TcpClient.Connected属性不可靠：这个属性仅反映上一次套接字操作的连接状态，不会实时检测远端是否已经断开连接。当设备发送FIN报文断开后，本地的Connected属性可能仍显示为true，导致代码认为连接有效并尝试读取。
2. ReadAsync的系统默认超时过长：当连接已被远端关闭，调用ReadAsync时，系统会等待TCP协议层的超时（通常就是设备设置的15秒无活动超时）才会抛出ConnectionAborted异常，这就是耗时15秒的核心原因。
3. DataAvailable判断的局限性：在连接半关闭状态下，DataAvailable可能返回false，但ReadAsync仍会阻塞等待，直到系统超时。

解决方案 #

1. 给ReadAsync添加主动超时 #

不要依赖系统默认的长超时，用Task.WhenAny给读取操作设置一个较短的超时（比如1秒），快速检测连接异常并触发重试。

2. 替换不可靠的Connected判断 #

放弃用TcpClient.Connected验证连接状态，改为通过实际读写操作的结果来判断，或者在操作前发送心跳包（如果设备支持）。

3. 修复递归调用的栈溢出风险 #

把当前的递归重连逻辑改成循环，避免多次重连时触发StackOverflowException。

4. 优化字节拼接方式 #

用List<byte>代替Enumerable.Concat，后者每次拼接都会创建新的枚举对象，性能差且会包含未读取的空字节。

修改后的代码示例 #

// 重试策略增加捕获超时取消异常
_readRetryPolicy = Policy.Handle<IOException>()
    .Or<OperationCanceledException>()
    .RetryAsync(3, async (_, _) =>
    {
        Disconnect();
        await ConnectAsync(_ip!, _port).ConfigureAwait(false);
    });

// 配置读取超时时间，根据实际场景调整
private readonly TimeSpan _readTimeout = TimeSpan.FromSeconds(1);

public async Task<byte[]> ReadResponseAsync()
{
    // 用循环代替递归，避免栈溢出
    while (_tcpClient is not { Connected: true })
    {
        Disconnect();
        await ConnectAsync(_ip!, _port).ConfigureAwait(false);
    }

    var responseBytes = new List<byte>();
    var buffer = new byte[32];
    int numBytesRead;

    do
    {
        // 给ReadAsync添加超时控制
        var readTask = _tcpClient.GetStream().ReadAsync(buffer);
        var timeoutTask = Task.Delay(_readTimeout);
        var completedTask = await Task.WhenAny(readTask, timeoutTask).ConfigureAwait(false);

        if (completedTask == timeoutTask)
        {
            // 超时抛出异常，触发重试策略
            throw new IOException("读取操作超时，连接可能已断开");
        }

        numBytesRead = await readTask.ConfigureAwait(false);
        if (numBytesRead > 0)
        {
            // 仅添加实际读取到的字节，避免空字节污染
            responseBytes.AddRange(buffer.AsSpan(0, numBytesRead));
        }
    }
    while (_tcpClient.GetStream().DataAvailable);

    return responseBytes.ToArray();
}

public void Disconnect()
{
    if (_tcpClient is TcpClient c)
    {
        try
        {
            // 主动关闭流，确保资源正确释放
            c.GetStream().Close();
        }
        catch
        {
            // 忽略关闭时的异常，避免影响后续释放
        }
        finally
        {
            c.Dispose();
            _tcpClient = null;
        }
    }
}

关键修改说明 #

• 用Task.WhenAny实现主动超时，快速终止阻塞的读取操作，避免等待15秒系统超时。
• 改用List<byte>存储响应字节，提升性能同时只保留有效数据。
• 递归改循环，彻底消除栈溢出风险。
• 断开连接时主动关闭流，确保套接字资源及时释放。
• 重试策略新增捕获OperationCanceledException，兼容超时取消场景。

内容的提问来源于stack exchange，提问作者Logan K