diff --git a/Source/MQTTnet.AspnetCore/ServiceCollectionExtensions.cs b/Source/MQTTnet.AspnetCore/ServiceCollectionExtensions.cs index 45a0df5c..d9d9fdb8 100644 --- a/Source/MQTTnet.AspnetCore/ServiceCollectionExtensions.cs +++ b/Source/MQTTnet.AspnetCore/ServiceCollectionExtensions.cs @@ -20,9 +20,22 @@ namespace MQTTnet.AspNetCore return services; } - - public static IServiceCollection AddHostedMqttServer(this IServiceCollection services, Action configure) - { + + public static IServiceCollection AddHostedMqttServer(this IServiceCollection services, Action configure) + { + services.AddSingleton(s => { + var builder = new MqttServerOptionsBuilder(); + configure(builder); + return builder.Build(); + }); + + services.AddHostedMqttServer(); + + return services; + } + + public static IServiceCollection AddHostedMqttServerWithServices(this IServiceCollection services, Action configure) + { services.AddSingleton(s => { var builder = new AspNetMqttServerOptionsBuilder(s); configure(builder);