From 14473b545889539151d7cf5d8c2a58ade870af2d Mon Sep 17 00:00:00 2001
From: jacalvar <jacalvar@microsoft.com>
Date: Mon, 28 Mar 2016 15:46:40 -0700
Subject: [PATCH] Return IServiceCollection from AddSession extension methods

---
 .../SessionServiceCollectionExtensions.cs                | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/Microsoft.AspNetCore.Session/SessionServiceCollectionExtensions.cs b/src/Microsoft.AspNetCore.Session/SessionServiceCollectionExtensions.cs
index af5c9c4344..c078a12fac 100644
--- a/src/Microsoft.AspNetCore.Session/SessionServiceCollectionExtensions.cs
+++ b/src/Microsoft.AspNetCore.Session/SessionServiceCollectionExtensions.cs
@@ -16,7 +16,8 @@ namespace Microsoft.Extensions.DependencyInjection
         /// Adds services required for application session state.
         /// </summary>
         /// <param name="services">The <see cref="IServiceCollection"/> to add the services to.</param>
-        public static void AddSession(this IServiceCollection services)
+        /// <returns>The <see cref="IServiceCollection"/> so that additional calls can be chained.</returns>
+        public static IServiceCollection AddSession(this IServiceCollection services)
         {
             if (services == null)
             {
@@ -24,6 +25,7 @@ namespace Microsoft.Extensions.DependencyInjection
             }
 
             services.AddTransient<ISessionStore, DistributedSessionStore>();
+            return services;
         }
 
         /// <summary>
@@ -31,7 +33,8 @@ namespace Microsoft.Extensions.DependencyInjection
         /// </summary>
         /// <param name="services">The <see cref="IServiceCollection"/> to add the services to.</param>
         /// <param name="configure">The session options to configure the middleware with.</param>
-        public static void AddSession(this IServiceCollection services, Action<SessionOptions> configure)
+        /// <returns>The <see cref="IServiceCollection"/> so that additional calls can be chained.</returns>
+        public static IServiceCollection AddSession(this IServiceCollection services, Action<SessionOptions> configure)
         {
             if (services == null)
             {
@@ -45,6 +48,8 @@ namespace Microsoft.Extensions.DependencyInjection
 
             services.Configure(configure);
             services.AddSession();
+
+            return services;
         }
     }
 }
\ No newline at end of file