From e99576a4294731986a61fdcd2d2b9ccbe3903d44 Mon Sep 17 00:00:00 2001
From: David Fowler <davidfowl@gmail.com>
Date: Thu, 17 Apr 2014 00:12:41 -0700
Subject: [PATCH] Added callback overload to UseContainer - Allows caller to
 pass a delegate to configure add services via the   ServiceCollection.

---
 .../ContainerExtensions.cs                                 | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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<IServiceDescriptor> applicationServices)
+        {
+            return builder.UseContainer(services => services.Add(applicationServices));
+        }
+
+        public static IBuilder UseContainer(this IBuilder builder, Action<ServiceCollection> configureServices)
         {
             var serviceCollection = new ServiceCollection();
-            serviceCollection.Add(applicationServices);
+            configureServices(serviceCollection);
             builder.ServiceProvider = serviceCollection.BuildServiceProvider(builder.ServiceProvider);
 
             return builder.UseMiddleware(typeof(ContainerMiddleware));