diff --git a/src/Microsoft.AspNet.RequestContainer/ContainerExtensions.cs b/src/Microsoft.AspNet.RequestContainer/ContainerExtensions.cs index 05b56f92cd..346eb28680 100644 --- a/src/Microsoft.AspNet.RequestContainer/ContainerExtensions.cs +++ b/src/Microsoft.AspNet.RequestContainer/ContainerExtensions.cs @@ -35,9 +35,14 @@ namespace Microsoft.AspNet.RequestContainer } public static IBuilder UseContainer(this IBuilder builder, IEnumerable applicationServices) + { + return builder.UseContainer(services => services.Add(applicationServices)); + } + + public static IBuilder UseContainer(this IBuilder builder, Action configureServices) { var serviceCollection = new ServiceCollection(); - serviceCollection.Add(applicationServices); + configureServices(serviceCollection); builder.ServiceProvider = serviceCollection.BuildServiceProvider(builder.ServiceProvider); return builder.UseMiddleware(typeof(ContainerMiddleware));