使用DeviceClient连接IoT Hub时内存占用过高问题求助

问题根源 #

1. 无限重试导致资源堆积：自定义CustomRetryPolicy的ShouldRetry始终返回true，无终止条件，持续发起的连接尝试会不断累积未释放的资源。
2. 异步操作未等待：连接状态回调中直接调用CloseAsync()和DisposeAsync()但未使用await，资源无法及时释放，大量半处理状态的对象占用内存。
3. 重试机制与手动销毁冲突：内置重试逻辑发起重试时，手动关闭并销毁deviceClient会导致内部持有已释放对象的引用，引发内存泄漏。

修复方案 #

1. 给自定义重试策略添加终止条件 #

限制最大重试次数，避免无限重试：

public class CustomRetryPolicy : IRetryPolicy
{
    private const int MaxRetryAttempts = 10; // 可根据业务需求调整

    public bool ShouldRetry(int currentRetryCount, Exception lastException, out TimeSpan retryInterval)
    {
        retryInterval = TimeSpan.FromSeconds(15);
        // 达到最大重试次数后停止重试
        return currentRetryCount < MaxRetryAttempts;
    }
}

2. 正确处理异步资源释放 #

将连接状态回调改为异步方法并等待操作完成，确保资源被及时清理：

static async Task Main(string[] args)
{
    Console.WriteLine("Validating IoT Hub");

    DeviceClient? deviceClient = null;

    try
    {
        var auth = new DeviceAuthenticationWithRegistrySymmetricKey("81ebefd6-f76c-4562-b395-e3247e97d227", "rI3sxcIgDFeMqpRwSutjZ/PoZ1KX5iTUT0OnL7R44Mg=");

        deviceClient = DeviceClient.Create("asdfasdf.azure-devices.net", auth, TransportType.Mqtt);

        deviceClient.OperationTimeoutInMilliseconds = 1000;
        // 包装异步回调适配SDK要求的委托类型
        deviceClient.SetConnectionStatusChangesHandler(async (status, reason) => await HandleConnectionChange(status, reason, ref deviceClient));
        deviceClient.SetRetryPolicy(new CustomRetryPolicy());
        
        await deviceClient.OpenAsync();
        Console.ReadLine();
    }
    catch (Exception ex)
    {
        Console.WriteLine(ex);
        // 异常场景下确保资源被清理
        if (deviceClient != null)
        {
            await deviceClient.CloseAsync();
            await deviceClient.DisposeAsync();
        }
    }
}

static async Task HandleConnectionChange(ConnectionStatus status, ConnectionStatusChangeReason reason, ref DeviceClient? deviceClient)
{
    if (status != ConnectionStatus.Connected && deviceClient != null)
    {
        Console.WriteLine($"Connection to IoT Hub is down! Reason: {reason}");
        // 等待异步操作完成，保证资源释放顺序
        await deviceClient.CloseAsync();
        await deviceClient.DisposeAsync();
        deviceClient = null;
    }
    else
    {
        Console.WriteLine($"Connection to IoT Hub is {status}. Reason: {reason}");
    }
}

3. 替代方案：使用内置重试策略（可选） #

如果不需要自定义逻辑，SDK提供的内置重试策略（如ExponentialBackoffRetry）已做好资源管理优化，可避免手动实现的缺陷：

deviceClient.SetRetryPolicy(new ExponentialBackoffRetry(TimeSpan.FromSeconds(1), TimeSpan.FromSeconds(30), 10));

关键注意事项 #

• 禁止无限重试：无论使用自定义还是内置策略，都必须设置合理的重试终止条件，防止资源无限制堆积。
• 异步操作必须等待：所有返回Task的异步方法都要使用await，保证资源按顺序释放。
• 替换.Wait()为await：同步阻塞方法.Wait()可能导致死锁，且无法保证资源正确清理。

内容的提问来源于stack exchange，提问作者Leonardo Suryana