You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
4.4 KiB
102 lines
4.4 KiB
5 months ago
|
// 大名科技(天津)有限公司版权所有 电话: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;
|
||
|
};
|
||
|
});
|
||
|
}
|
||
|
}
|