53 lines
1.6 KiB
C#
53 lines
1.6 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Diagnostics;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
|
|
namespace Microsoft.AspNetCore.SignalR
|
|
{
|
|
public class DefaultHubActivator<THub> : IHubActivator<THub> where THub: Hub
|
|
{
|
|
// Object factory for THub instances
|
|
private static readonly Lazy<ObjectFactory> _objectFactory = new Lazy<ObjectFactory>(() => ActivatorUtilities.CreateFactory(typeof(THub), Type.EmptyTypes));
|
|
private readonly IServiceProvider _serviceProvider;
|
|
private bool? _created;
|
|
|
|
public DefaultHubActivator(IServiceProvider serviceProvider)
|
|
{
|
|
_serviceProvider = serviceProvider;
|
|
}
|
|
|
|
public THub Create()
|
|
{
|
|
Debug.Assert(!_created.HasValue, "hub activators must not be reused.");
|
|
|
|
_created = false;
|
|
var hub = _serviceProvider.GetService<THub>();
|
|
if (hub == null)
|
|
{
|
|
hub = (THub)_objectFactory.Value(_serviceProvider, Array.Empty<object>());
|
|
_created = true;
|
|
}
|
|
|
|
return hub;
|
|
}
|
|
|
|
public void Release(THub hub)
|
|
{
|
|
if (hub == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(hub));
|
|
}
|
|
|
|
Debug.Assert(_created.HasValue, "hubs must be released with the hub activator they were created");
|
|
|
|
if (_created.Value)
|
|
{
|
|
hub.Dispose();
|
|
}
|
|
}
|
|
}
|
|
}
|