35 lines
1.6 KiB
C#
35 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 Microsoft.AspNetCore.Connections;
|
|
using Microsoft.AspNetCore.SignalR.Internal;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
|
|
namespace Microsoft.AspNetCore.SignalR
|
|
{
|
|
/// <summary>
|
|
/// Extension methods for <see cref="IConnectionBuilder"/>.
|
|
/// </summary>
|
|
public static class SignalRConnectionBuilderExtensions
|
|
{
|
|
/// <summary>
|
|
/// Configure the connection to host the specified <see cref="Hub"/> type.
|
|
/// </summary>
|
|
/// <typeparam name="THub">The <see cref="Hub"/> type to host on the connection.</typeparam>
|
|
/// <param name="connectionBuilder">The connection to configure.</param>
|
|
/// <returns>The same instance of the <see cref="IConnectionBuilder"/> for chaining.</returns>
|
|
public static IConnectionBuilder UseHub<THub>(this IConnectionBuilder connectionBuilder) where THub : Hub
|
|
{
|
|
var marker = connectionBuilder.ApplicationServices.GetService(typeof(SignalRCoreMarkerService));
|
|
if (marker == null)
|
|
{
|
|
throw new InvalidOperationException("Unable to find the required services. Please add all the required services by calling " +
|
|
"'IServiceCollection.AddSignalR' inside the call to 'ConfigureServices(...)' in the application startup code.");
|
|
}
|
|
|
|
return connectionBuilder.UseConnectionHandler<HubConnectionHandler<THub>>();
|
|
}
|
|
}
|
|
}
|