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