如何在Host Builder扩展方法中获取正确的IServiceProvider实例？

核心逻辑说明 #

你看到的AddDbContext<T>那个带(IServiceProvider, DbContextOptionsBuilder)参数的重载，本质是利用了依赖注入容器构建完成后的回调机制——配置Services阶段容器还没初始化，没法直接拿到IServiceProvider，而是注册一个在容器就绪后执行的逻辑，由.NET Host框架自动把构建好的IServiceProvider传给你定义的委托。

具体实现步骤 #

1. 定义扩展方法 #

以下是模仿该机制的自定义扩展方法示例，假设你要配置自定义服务和选项：

public static IServiceCollection AddCustomService<TService>(
    this IServiceCollection services,
    Action<IServiceProvider, CustomOptionsBuilder> configure)
    where TService : class
{
    // 先把你的服务注册到DI容器
    services.AddSingleton<TService>();

    // 注册一个HostedService，用来触发容器就绪后的配置逻辑
    services.AddSingleton<IHostedService>(provider =>
    {
        var optionsBuilder = new CustomOptionsBuilder();
        // 这里由框架传入已构建好的IServiceProvider
        configure(provider, optionsBuilder);
        
        // 用配置好的选项初始化你的服务
        var targetService = provider.GetRequiredService<TService>();
        targetService.Initialize(optionsBuilder.Options);
        
        return new EmptyHostedService(); // 空实现，仅用来触发执行
    });

    return services;
}

// 空的HostedService实现，只负责触发初始化逻辑
private class EmptyHostedService : IHostedService
{
    public Task StartAsync(CancellationToken cancellationToken) => Task.CompletedTask;
    public Task StopAsync(CancellationToken cancellationToken) => Task.CompletedTask;
}

// 自定义的选项构建器和选项类示例
public class CustomOptionsBuilder
{
    public CustomOptions Options { get; } = new CustomOptions();
}

public class CustomOptions
{
    public string ConnectionString { get; set; }
}

2. 谁来传入IServiceProvider #

答案是**.NET Host框架**。当你调用host.Build()完成容器构建后，Host启动时会自动执行所有注册的IHostedService的StartAsync方法，此时容器已经完全初始化，框架会把现成的IServiceProvider实例注入到我们注册的委托里，也就是你在扩展方法里定义的configure参数。

3. 预期用法示例 #

var builder = Host.CreateDefaultBuilder(args);
builder.ConfigureServices(services =>
{
    services.AddCustomService<MyBusinessService>((sp, optionsBuilder) =>
    {
        // 这里可以通过IServiceProvider获取其他服务，比如配置
        var config = sp.GetRequiredService<IConfiguration>();
        optionsBuilder.Options.ConnectionString = config.GetConnectionString("DefaultDB");
    });
});

var host = builder.Build();
host.Run();

替代方案（可选） #

如果不需要在服务初始化时执行，也可以用PostConfigure结合IServiceProvider，不过这种方式适合纯选项配置：

public static IServiceCollection AddCustomOptions(
    this IServiceCollection services,
    Action<IServiceProvider, CustomOptions> configure)
{
    services.PostConfigure<CustomOptions>(sp =>
    {
        var options = new CustomOptions();
        configure(sp, options);
        // 应用配置
    });
    return services;
}

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