102 lines
4.4 KiB
C#
102 lines
4.4 KiB
C#
// 大名科技（天津）有限公司版权所有 电话：18020030720 QQ：515096995
|
||
//
|
||
// 此源代码遵循位于源代码树根目录中的 LICENSE 文件的许可证
|
||
|
||
using Furion.Logging.Extensions;
|
||
using Microsoft.AspNetCore.DataProtection;
|
||
using Newtonsoft.Json;
|
||
using StackExchange.Redis;
|
||
|
||
namespace Admin.NET.Core;
|
||
|
||
public static class SignalRSetup
|
||
{
|
||
/// <summary>
|
||
/// 即时消息SignalR注册
|
||
/// </summary>
|
||
/// <param name="services"></param>
|
||
/// <param name="SetNewtonsoftJsonSetting"></param>
|
||
public static void AddSignalR(this IServiceCollection services, Action<JsonSerializerSettings> SetNewtonsoftJsonSetting)
|
||
{
|
||
var signalRBuilder = services.AddSignalR(options =>
|
||
{
|
||
options.EnableDetailedErrors = true;
|
||
options.ClientTimeoutInterval = TimeSpan.FromMinutes(2);
|
||
options.KeepAliveInterval = TimeSpan.FromMinutes(1);
|
||
options.MaximumReceiveMessageSize = 1024 * 1024 * 10; // 数据包大小10M，默认最大为32K
|
||
}).AddNewtonsoftJsonProtocol(options => SetNewtonsoftJsonSetting(options.PayloadSerializerSettings));
|
||
|
||
// 若未启用Redis缓存，直接返回
|
||
var cacheOptions = App.GetConfig<CacheOptions>("Cache", true);
|
||
if (cacheOptions.CacheType != CacheTypeEnum.Redis.ToString())
|
||
return;
|
||
|
||
// 若已开启集群配置，则把SignalR配置为支持集群模式
|
||
var clusterOpt = App.GetConfig<ClusterOptions>("Cluster", true);
|
||
if (!clusterOpt.Enabled)
|
||
return;
|
||
|
||
var redisOptions = clusterOpt.SentinelConfig;
|
||
ConnectionMultiplexer connection1;
|
||
if (clusterOpt.IsSentinel) // 哨兵模式
|
||
{
|
||
var redisConfig = new ConfigurationOptions
|
||
{
|
||
AbortOnConnectFail = false,
|
||
ServiceName = redisOptions.ServiceName,
|
||
AllowAdmin = true,
|
||
DefaultDatabase = redisOptions.DefaultDb,
|
||
Password = redisOptions.Password
|
||
};
|
||
redisOptions.EndPoints.ForEach(u => redisConfig.EndPoints.Add(u));
|
||
connection1 = ConnectionMultiplexer.Connect(redisConfig);
|
||
}
|
||
else
|
||
{
|
||
connection1 = ConnectionMultiplexer.Connect(clusterOpt.SignalR.RedisConfiguration);
|
||
}
|
||
// 密钥存储（数据保护）
|
||
services.AddDataProtection().PersistKeysToStackExchangeRedis(connection1, clusterOpt.DataProtecteKey);
|
||
|
||
signalRBuilder.AddStackExchangeRedis(options =>
|
||
{
|
||
// 此处设置的ChannelPrefix并不会生效，如果两个不同的项目，且[程序集名+类名]一样，使用同一个redis服务，请注意修改 Hub/OnlineUserHub 的类名。
|
||
// 原因请参考下边链接：
|
||
// https://github.com/dotnet/aspnetcore/blob/f9121bc3e976ec40a959818451d126d5126ce868/src/SignalR/server/StackExchangeRedis/src/RedisHubLifetimeManager.cs#L74
|
||
// https://github.com/dotnet/aspnetcore/blob/f9121bc3e976ec40a959818451d126d5126ce868/src/SignalR/server/StackExchangeRedis/src/Internal/RedisChannels.cs#L33
|
||
options.Configuration.ChannelPrefix = new RedisChannel(clusterOpt.SignalR.ChannelPrefix, RedisChannel.PatternMode.Auto);
|
||
options.ConnectionFactory = async writer =>
|
||
{
|
||
ConnectionMultiplexer connection;
|
||
if (clusterOpt.IsSentinel)
|
||
{
|
||
var config = new ConfigurationOptions
|
||
{
|
||
AbortOnConnectFail = false,
|
||
ServiceName = redisOptions.ServiceName,
|
||
AllowAdmin = true,
|
||
DefaultDatabase = redisOptions.DefaultDb,
|
||
Password = redisOptions.Password
|
||
};
|
||
redisOptions.EndPoints.ForEach(u => config.EndPoints.Add(u));
|
||
connection = await ConnectionMultiplexer.ConnectAsync(config, writer);
|
||
}
|
||
else
|
||
{
|
||
connection = await ConnectionMultiplexer.ConnectAsync(clusterOpt.SignalR.RedisConfiguration);
|
||
}
|
||
|
||
connection.ConnectionFailed += (_, e) =>
|
||
{
|
||
"连接 Redis 失败".LogError();
|
||
};
|
||
|
||
if (!connection.IsConnected)
|
||
{
|
||
"无法连接 Redis".LogError();
|
||
}
|
||
return connection;
|
||
};
|
||
});
|
||
}
|
||
} |