Expose WebHostBuilderContext in UseKestrel #1334

Author: Tratcher

samples/SampleApp/Startup.cs

@@ -57,8 +57,8 @@ public static Task Main(string[] args)
                 {
                     factory.AddConsole();
                 })
-                .UseKestrel(options =>
                 {
+                .UseKestrel((context, options) =>
                     // Run callbacks on the transport thread
                     options.ApplicationSchedulingMode = SchedulingMode.Inline;
 

src/Kestrel/WebHostBuilderKestrelExtensions.cs

@@ -57,5 +57,31 @@ public static IWebHostBuilder UseKestrel(this IWebHostBuilder hostBuilder, Actio
                 services.Configure(options);
             });
         }
+
+        /// <summary>
+        /// Specify Kestrel as the server to be used by the web host.
+        /// </summary>
+        /// <param name="hostBuilder">
+        /// The Microsoft.AspNetCore.Hosting.IWebHostBuilder to configure.
+        /// </param>
+        /// <param name="configureOptions">A callback to configure Kestrel options.</param>
+        /// <returns>
+        /// The Microsoft.AspNetCore.Hosting.IWebHostBuilder.
+        /// </returns>
+        public static IWebHostBuilder UseKestrel(this IWebHostBuilder hostBuilder, Action<WebHostBuilderContext, KestrelServerOptions> configureOptions)
+        {
+            if (configureOptions == null)
+            {
+                throw new ArgumentNullException(nameof(configureOptions));
+            }
+
+            return hostBuilder.UseKestrel().ConfigureServices((context, services) =>
+            {
+                services.Configure<KestrelServerOptions>(options =>
+                {
+                    configureOptions(context, options);
+                });
+            });
+        }
     }
 }