From df9057a6f7424a61fe51808f9f3e6d844afa07a8 Mon Sep 17 00:00:00 2001
From: David Fowler <davidfowl@gmail.com>
Date: Wed, 14 Dec 2016 00:01:35 -0800
Subject: [PATCH] Revert "Revert "React to hosting changes""

This reverts commit 12649b19bfed2b2d66a4e904de7b3d0ad94ae33c.
---
 ...nts.cs => SocketsApplicationLifetimeService.cs} | 14 ++++----------
 .../SocketsDependencyInjectionExtensions.cs        |  2 +-
 2 files changed, 5 insertions(+), 11 deletions(-)
 rename src/Microsoft.AspNetCore.Sockets/{SocketsApplicationLifetimeEvents.cs => SocketsApplicationLifetimeService.cs} (57%)

diff --git a/src/Microsoft.AspNetCore.Sockets/SocketsApplicationLifetimeEvents.cs b/src/Microsoft.AspNetCore.Sockets/SocketsApplicationLifetimeService.cs
similarity index 57%
rename from src/Microsoft.AspNetCore.Sockets/SocketsApplicationLifetimeEvents.cs
rename to src/Microsoft.AspNetCore.Sockets/SocketsApplicationLifetimeService.cs
index f06f09ff47..619dbf9492 100644
--- a/src/Microsoft.AspNetCore.Sockets/SocketsApplicationLifetimeEvents.cs
+++ b/src/Microsoft.AspNetCore.Sockets/SocketsApplicationLifetimeService.cs
@@ -6,26 +6,20 @@ using Microsoft.AspNetCore.Hosting;
 
 namespace Microsoft.AspNetCore.Sockets
 {
-    public class SocketsApplicationLifetimeEvents : IApplicationLifetimeEvents
+    public class SocketsApplicationLifetimeService : IHostedService
     {
         private readonly ConnectionManager _connectionManager;
 
-        public SocketsApplicationLifetimeEvents(ConnectionManager connectionManager)
+        public SocketsApplicationLifetimeService(ConnectionManager connectionManager)
         {
             _connectionManager = connectionManager;
         }
 
-        public void OnApplicationStarted()
+        public void Start()
         {
-
         }
 
-        public void OnApplicationStopped()
-        {
-
-        }
-
-        public void OnApplicationStopping()
+        public void Stop()
         {
             _connectionManager.CloseConnections();
         }
diff --git a/src/Microsoft.AspNetCore.Sockets/SocketsDependencyInjectionExtensions.cs b/src/Microsoft.AspNetCore.Sockets/SocketsDependencyInjectionExtensions.cs
index fd20047a9a..b87a6d28c2 100644
--- a/src/Microsoft.AspNetCore.Sockets/SocketsDependencyInjectionExtensions.cs
+++ b/src/Microsoft.AspNetCore.Sockets/SocketsDependencyInjectionExtensions.cs
@@ -14,7 +14,7 @@ namespace Microsoft.Extensions.DependencyInjection
         {
             services.AddRouting();
             services.TryAddSingleton<ConnectionManager>();
-            services.TryAddEnumerable(ServiceDescriptor.Singleton<IApplicationLifetimeEvents, SocketsApplicationLifetimeEvents>());
+            services.TryAddEnumerable(ServiceDescriptor.Singleton<IHostedService, SocketsApplicationLifetimeService>());
             services.TryAddSingleton<PipelineFactory>();
             services.TryAddSingleton<HttpConnectionDispatcher>();
             return services;