.NET Core 6 Azure微服务：如何让Service Bus仅在生产槽运行

核心思路 #

不要在部署槽中启动Service Bus处理器，从根源避免消息被处理——而非启动后再停止（后者会引发预取消息的异常和数据风险）。Azure App Service会通过环境变量WEBSITE_SLOT_NAME暴露当前运行的槽名，生产槽默认名称为Production，我们可以基于这个变量做判断。

具体实现方案 #

方案1：在消息接收器启动前判断槽名（推荐） #

修改MessageReceiver<T>类，添加槽名检测逻辑，仅在生产槽时才初始化并启动处理器：

1. 添加槽名检测方法

private bool IsProductionSlot()
{
    var slotName = Environment.GetEnvironmentVariable("WEBSITE_SLOT_NAME");
    // 若你的生产槽自定义了名称，请修改此处的匹配值
    return string.Equals(slotName, "Production", StringComparison.OrdinalIgnoreCase);
}

2. 修改ExecuteAsync方法，提前终止非生产槽的启动流程

protected override async Task ExecuteAsync(CancellationToken stoppingToken)
{
    try
    {
        if (!IsProductionSlot())
        {
            _logger.LogInformation("当前为部署槽，不启动Service Bus消息处理器");
            return;
        }

        cancellationToken = stoppingToken;
        await InitializeListener();
        if (processor != null)
        {
            processor.ProcessMessageAsync += MessageHandler;
            processor.ProcessErrorAsync += ErrorHandler;
            await StartReceivingMessagesAsync(processor, stoppingToken);
        }
    }
    catch (Exception ex)
    {
        _logger.LogError(ex.Message);
    }
}

方案2：在服务注册阶段跳过部署槽的处理器注册 #

如果希望部署槽中完全不加载Service Bus相关的HostedService，可以修改AddAzureServiceBusRegistration扩展方法，仅在生产槽时注册处理器：

public static IServiceCollection AddAzureServiceBusRegistration<T>(this IServiceCollection services, ServiceBusConfigurationOptions serviceBusConfigurationOptions)  where T : IIntegrationMessageQueue  
{
    var slotName = Environment.GetEnvironmentVariable("WEBSITE_SLOT_NAME");
    if (string.Equals(slotName, "Production", StringComparison.OrdinalIgnoreCase))
    {
        services.AddHostedService<MessageReceiver<T>>(sp =>
        {
            var messageProcessingService = sp.GetRequiredService<IIntegrationMessageHandler<T>>();
            var logger = sp.GetRequiredService<ILogger<MessageReceiver<T>>>();
            TelemetryClient telemetryClient = sp.GetRequiredService<TelemetryClient>();
            return new MessageReceiver<T>(logger, serviceBusConfigurationOptions, messageProcessingService, telemetryClient);
        });
        services.AddTransient<IIntegrationMessageHandler<T>, MessageHandler>();
    }
    else
    {
        // 部署槽注册空实现，避免依赖注入报错
        services.AddTransient<IIntegrationMessageHandler<T>, NullMessageHandler<T>>();
    }

    return services;
}

// 空消息处理器实现
public class NullMessageHandler<T> : IIntegrationMessageHandler<T> where T : IIntegrationMessageQueue
{
    public Task Message(string body)
    {
        return Task.CompletedTask;
    }
}

关键注意事项 #

• 生产槽名称匹配：如果你的生产槽不是默认的Production，请务必修改判断逻辑中的槽名字符串。
• 避免预取消息风险：之前启动后停止处理器的方式会导致处理器已经预取的消息无法正常处理，出现锁过期、重复消费或数据丢失，直接跳过启动流程可以彻底避免这个问题。
• 依赖注入兼容性：如果采用方案2，要确保部署槽中所有依赖IIntegrationMessageHandler<T>的服务都能正常工作，空实现是最安全的选择。

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