diff --git a/src/Microsoft.AspNetCore.Http.Connections/ConnectionsAppBuilderExtensions.cs b/src/Microsoft.AspNetCore.Http.Connections/ConnectionsAppBuilderExtensions.cs index 7242afef6b..f18713b37e 100644 --- a/src/Microsoft.AspNetCore.Http.Connections/ConnectionsAppBuilderExtensions.cs +++ b/src/Microsoft.AspNetCore.Http.Connections/ConnectionsAppBuilderExtensions.cs @@ -10,13 +10,18 @@ namespace Microsoft.AspNetCore.Builder { public static class ConnectionsAppBuilderExtensions { - public static IApplicationBuilder UseConnections(this IApplicationBuilder app, Action callback) + public static IApplicationBuilder UseConnections(this IApplicationBuilder app, Action configure) { + if (configure == null) + { + throw new ArgumentNullException(nameof(configure)); + } + var dispatcher = app.ApplicationServices.GetRequiredService(); var routes = new RouteBuilder(app); - callback(new ConnectionsRouteBuilder(routes, dispatcher)); + configure(new ConnectionsRouteBuilder(routes, dispatcher)); app.UseWebSockets(); app.UseRouter(routes.Build());