同名扩展方法替换第三方密封API的无侵入解决方案咨询

问题背景 #

需要替换第三方密封API的RetryLogic方法，该方法被数十个消费类调用，直接修改这些消费类不可行，且无法直接通过同名扩展方法覆盖原方法（编译器会优先调用实例方法而非扩展方法）。

原消费类代码示例：

public class RestAppClassThatUsesRestClient
{
    private IRestClient _3rdPartyRestClient;
    private ILogger _logger;
    private IConfiguration _configuration;

    public RestAppClassThatUsesRestClient(IRestClient restClient, ILogger logger, IConfiguration configuration)
    {
        _3rdPartyRestClient = restClient;
        _logger = logger;
        _configuration = configuration;
    }

    public Task<TModel> GetSomething<TModel>(string url)
    {
        return _3rdPartyRestClient.RetryLogic(url);
    }
}

需要替换为的自定义重试逻辑（已修正原代码依赖缺失、类型转换等问题）：

/// <summary>
/// 自定义IRestClient重试逻辑扩展类
/// </summary>
public static class RestClientExtensions
{
    // 调整扩展方法，传入所需依赖
    public static async Task<TModel> CustomRetryLogic<TModel>(this IRestClient restClient, string url, 
        IConfiguration configuration, ILogger logger)
    {
        int retryCount = 0;
        // 配置参数解析+默认值兜底
        if (!int.TryParse(configuration.GetSection("AppSettings:MaxRetryAttempt").Value, out int maxRetryAttempt))
        {
            maxRetryAttempt = 3;
        }
        if (!int.TryParse(configuration.GetSection("AppSettings:Timeout").Value, out int timeoutMs))
        {
            timeoutMs = 1000;
        }

        while (retryCount < maxRetryAttempt)
        {
            try
            {
                return await restClient.Get<TModel>(url);
            }
            catch (System.Net.WebException wex)
            {
                retryCount++;
                if (retryCount >= maxRetryAttempt)
                {
                    logger.LogError(wex, "重试次数耗尽，请求失败");
                    return default(TModel);
                }
                await Task.Delay(timeoutMs); // 用Task.Delay替代Thread.Sleep，避免阻塞线程
            }
            catch (Exception ex)
            {
                logger.LogError(ex, "请求发生非网络异常");
                throw;
            }
        }
        return default(TModel);
    }
}

解决方案：装饰器模式 #

通过装饰器模式包装第三方IRestClient实例，在不修改现有消费类的前提下，替换RetryLogic的实现。

步骤1：创建IRestClient装饰器类 #

实现IRestClient接口，包装原第三方实例，并注入所需依赖：

public class RestClientRetryDecorator : IRestClient
{
    private readonly IRestClient _innerRestClient;
    private readonly IConfiguration _configuration;
    private readonly ILogger<RestClientRetryDecorator> _logger;

    public RestClientRetryDecorator(IRestClient innerRestClient, 
        IConfiguration configuration, 
        ILogger<RestClientRetryDecorator> logger)
    {
        _innerRestClient = innerRestClient;
        _configuration = configuration;
        _logger = logger;
    }

    // 转发IRestClient的其他方法到原实例
    public Task<TModel> Get<TModel>(string url)
    {
        return _innerRestClient.Get<TModel>(url);
    }

    // 重写RetryLogic，调用自定义重试逻辑
    public Task<TModel> RetryLogic<TModel>(string url)
    {
        return _innerRestClient.CustomRetryLogic(url, _configuration, _logger);
    }

    // 若IRestClient有其他方法（如Post、Put），需同样转发到原实例
}

步骤2：在依赖注入容器中注册装饰器 #

以ASP.NET Core为例，在Program.cs中替换原IRestClient的注册：

// 先注册第三方IRestClient的具体实现
builder.Services.AddScoped<IRestClient, ThirdPartyRestClient>();

// 用装饰器包裹原实例（需.NET Core 3.0+，内置Decorate方法）
builder.Services.Decorate<IRestClient, RestClientRetryDecorator>();

若使用旧版本DI容器，可手动注册：

builder.Services.AddScoped<IRestClient>(sp => 
{
    var innerClient = sp.GetRequiredService<ThirdPartyRestClient>();
    var config = sp.GetRequiredService<IConfiguration>();
    var logger = sp.GetRequiredService<ILogger<RestClientRetryDecorator>>();
    return new RestClientRetryDecorator(innerClient, config, logger);
});

原理说明 #

• 装饰器类与第三方IRestClient实现相同接口，可无缝替换原实例
• 现有消费类通过依赖注入获取的是装饰器实例，调用RetryLogic时会执行自定义逻辑
• 其他方法直接转发给原第三方实例，不影响原有功能

内容的提问来源于stack exchange，提问作者Keshav Prasad