操控平台后端代码
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

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;
};
});
}
}