如何在特定请求中启用CoolFeature？重写中间件ServiceCollection

我在Startup.cs中注册服务的代码如下：

services
    .AddRandomComponent(options =>
            {
                config.DisableCoolFeature();
            }); 

对应的AddRandomComponent方法实现：

public static class ServiceCollectionExtensions
{
    public static RandomComponent AddRandomComponent(this IServiceCollection services,
        Action<Config> setupSettings = null)
    {
        services
            .AddOptions<Config>()
            .Configure<IConfiguration>((settings, configuration) =>
            {
                configuration.BindUISettings(settings);
                setupSettings?.Invoke(settings);
            });

        services.TryAddSingleton<LoggingService>();

        return new RandomComponent (services);
    }
}

现在我需要在特定请求中启用CoolFeature，请问有没有实现方法？

我尝试写了如下中间件，但获取到的RandomComponent实例为null：

public class RandomMiddleware
{
    private readonly RequestDelegate _next;

    public RandomMiddleware(RequestDelegate next)
    {
        _next = next;
    }

    public async Task InvokeAsync(HttpContext context)
    {
        IServiceProvider serviceProvider = context.RequestServices;

        var item = serviceProvider.GetService<RandomComponent>();

        if (context.Request.Query.TryGetValue("coolFeature", out var isCoolFeature))
        {
            // Override service collection 
        }

        await _next(context);
    }
}

1. 为什么RandomComponent实例为null？ #

你在AddRandomComponent方法里只是创建并返回了一个RandomComponent实例，但没有把它注册到DI容器中。DI容器中不存在该类型的服务，所以调用GetService<RandomComponent>()自然获取不到实例。

2. 实现特定请求启用CoolFeature的方法 #

核心是修改请求级别的Config配置，利用IOptionsSnapshot<Config>实现请求范围内的配置覆盖：

步骤1：修改中间件逻辑

public class RandomMiddleware
{
    private readonly RequestDelegate _next;

    public RandomMiddleware(RequestDelegate next)
    {
        _next = next;
    }

    public async Task InvokeAsync(HttpContext context, IOptionsSnapshot<Config> configSnapshot)
    {
        if (context.Request.Query.TryGetValue("coolFeature", out var isCoolFeature) 
            && bool.TryParse(isCoolFeature, out var enableCoolFeature))
        {
            // 根据请求参数切换CoolFeature状态
            if (enableCoolFeature)
            {
                configSnapshot.Value.EnableCoolFeature(); // 假设Config类有对应的启用方法
            }
            else
            {
                configSnapshot.Value.DisableCoolFeature();
            }
        }

        await _next(context);
    }
}

步骤2：业务代码注入IOptionsSnapshot<Config>

确保依赖Config的业务类注入IOptionsSnapshot<Config>，这样就能获取到当前请求的最新配置：

public class SomeBusinessService
{
    private readonly Config _config;

    public SomeBusinessService(IOptionsSnapshot<Config> configSnapshot)
    {
        _config = configSnapshot.Value;
    }

    public void ExecuteLogic()
    {
        if (_config.IsCoolFeatureEnabled) // 假设Config有该状态属性
        {
            // 执行CoolFeature相关逻辑
        }
    }
}

补充说明

• IOptionsSnapshot<Config>是范围服务，每个请求会生成一个实例，修改其Value只会影响当前请求，不会干扰其他请求。
• 如果确实需要使用RandomComponent，需在AddRandomComponent方法中把它注册到DI容器，比如添加：

  services.AddSingleton<RandomComponent>();
  

  但这不是解决当前问题的必要操作，核心还是通过请求级配置覆盖实现需求。

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