ThingsGateway/framework/admin/ThingsGateway.Admin.Application/SignalR/Provider/UserIdProvider.cs

#region copyright
//------------------------------------------------------------------------------
//  此代码版权声明为全文件覆盖，如有原作者特别声明，会在下方手动补充
//  此代码版权（除特别声明外的代码）归作者本人Diego所有
//  源代码使用协议遵循本仓库的开源协议及附加协议
//  Gitee源代码仓库：https://gitee.com/diego2098/ThingsGateway
//  Github源代码仓库：https://github.com/kimdiego2098/ThingsGateway
//  使用文档：https://diego2098.gitee.io/thingsgateway-docs/
//  QQ群：605534569
//------------------------------------------------------------------------------
#endregion

using Microsoft.AspNetCore.Http.Connections.Features;
using Microsoft.AspNetCore.SignalR;

using ThingsGateway.Foundation.Extension.String;

using Yitter.IdGenerator;

namespace ThingsGateway.Admin.Application;

/// <summary>
/// 用户ID提供器
/// </summary>
public class UserIdProvider : IUserIdProvider
{
    /// <inheritdoc/>
    public string GetUserId(HubConnectionContext connection)
    {
        var feature = connection.Features.Get<IHttpContextFeature>();
        var UserId = feature.HttpContext.Request.Headers[ClaimConst.UserId].FirstOrDefault()?.ToLong();

        if (UserId > 0)
        {
            return $"{UserId}{SysHub.Separate}{YitIdHelper.NextId()}";//返回用户ID
        }

        return connection.ConnectionId;
    }
}