基于DI的Azure队列触发函数：运行时动态指定数据库

希望单个Azure队列触发函数处理如下格式的消息：

{
  "tenanatDb": "Db_Name_3102",
  "command": "getCountryCurrencies",
  "args": "US"
}

当前使用依赖注入（DI），Startup类配置如下：

internal class Startup : FunctionsStartup
{
    public override void Configure(IFunctionsHostBuilder builder)
    {
        builder.Services.AddDbContext<MyDbContext>(options =>
        options
            .UseSqlServer("<ConnectionString>"));

        builder.Services.AddSingleton<IMyService>((s) => {
            return new MyService();
        });
    }
}

可以提供标准连接字符串，但希望保留连接字符串的其他部分，仅在运行时根据队列消息中的tenantDb指定数据库名称（如连接字符串中的***Db_Name_3102***部分）：

Server=myServerAddress;Database=***Db_Name_3102***;User Id=myUsername;Password=myPassword;

目前已通过非DI方式实现，现寻求基于DI的实现方案。

1. 调整MyDbContext的构造逻辑 #

修改MyDbContext，使其支持通过动态参数生成连接字符串，而非在Startup中固定配置：

public class MyDbContext : DbContext
{
    private readonly string _connectionString;

    // 接受完整连接字符串的构造函数
    public MyDbContext(string connectionString) : base()
    {
        _connectionString = connectionString;
    }

    // 基于基础连接字符串+租户库名生成的构造函数
    public MyDbContext(string baseConnStr, string tenantDbName) : base()
    {
        _connectionString = baseConnStr.Replace("***Db_Name_3102***", tenantDbName);
    }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseSqlServer(_connectionString);
    }

    // 此处定义你的DbSet
}

2. 注册基础依赖到DI容器 #

在Startup中，注册基础连接字符串（带占位符）和服务，不再直接注册固定的DbContext实例：

internal class Startup : FunctionsStartup
{
    public override void Configure(IFunctionsHostBuilder builder)
    {
        // 从应用配置读取基础连接字符串（建议存于Azure应用设置）
        var baseConnectionString = Environment.GetEnvironmentVariable("BaseSqlConnectionString");
        
        // 注册基础连接字符串为单例
        builder.Services.AddSingleton(baseConnectionString);

        // 注册业务服务
        builder.Services.AddSingleton<IMyService, MyService>();
    }
}

3. 在队列函数中动态创建DbContext #

在队列触发函数中，通过DI注入基础连接字符串和业务服务，根据队列消息生成对应DbContext并执行操作：

public class QueueTriggerFunction
{
    private readonly string _baseConnStr;
    private readonly IMyService _myService;

    // 依赖注入构造函数
    public QueueTriggerFunction(string baseConnStr, IMyService myService)
    {
        _baseConnStr = baseConnStr;
        _myService = myService;
    }

    [FunctionName("ProcessTenantQueue")]
    public async Task Run(
        [QueueTrigger("tenant-queue", Connection = "AzureWebJobsStorage")] string queueMessage,
        ILogger log)
    {
        // 反序列化队列消息
        var message = JsonSerializer.Deserialize<QueueMessage>(queueMessage);
        if (message == null || string.IsNullOrWhiteSpace(message.tenanatDb))
        {
            log.LogError("无效消息：缺少tenantDb字段");
            return;
        }

        // 生成动态连接字符串并创建DbContext
        using var dbContext = new MyDbContext(_baseConnStr, message.tenanatDb);

        // 根据command执行对应业务逻辑
        switch (message.command)
        {
            case "getCountryCurrencies":
                var currencies = await _myService.GetCountryCurrencies(dbContext, message.args);
                // 处理返回结果（如写入输出队列、日志等）
                break;
            default:
                log.LogWarning($"不支持的命令：{message.command}");
                break;
        }
    }
}

// 队列消息模型
public class QueueMessage
{
    public string tenanatDb { get; set; }
    public string command { get; set; }
    public string args { get; set; }
}

可选优化：封装DbContext工厂 #

如果希望更符合DI设计规范，可以创建工厂类封装DbContext的创建逻辑：

public interface IDbContextFactory
{
    MyDbContext Create(string tenantDbName);
}

public class DbContextFactory : IDbContextFactory
{
    private readonly string _baseConnStr;

    public DbContextFactory(string baseConnStr)
    {
        _baseConnStr = baseConnStr;
    }

    public MyDbContext Create(string tenantDbName)
    {
        return new MyDbContext(_baseConnStr, tenantDbName);
    }
}

在Startup中注册工厂：

builder.Services.AddSingleton<IDbContextFactory, DbContextFactory>();

函数中注入工厂替代直接注入连接字符串：

public QueueTriggerFunction(IDbContextFactory dbFactory, IMyService myService)
{
    _dbFactory = dbFactory;
    _myService = myService;
}

// Run方法中创建DbContext
using var dbContext = _dbFactory.Create(message.tenanatDb);

注意事项 #

• 必须用using语句包裹DbContext实例，确保使用后及时释放，避免数据库连接泄漏。
• 基础连接字符串不要硬编码，存入Azure Functions应用设置中，通过环境变量读取。
• 若租户数量较多，需关注SQL Server连接池配置，确保连接能被有效复用。

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