tuiwucarrer/Admin.NET/Admin.NET.Core/Service/OnlineUser/SysOnlineUserService.cs

// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
//
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
//
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！

using Microsoft.AspNetCore.SignalR;

namespace Admin.NET.Core.Service;

/// <summary>
/// 系统在线用户服务 🧩
/// </summary>
[ApiDescriptionSettings(Order = 300)]
public class SysOnlineUserService : IDynamicApiController, ITransient
{
    private readonly SqlSugarRepository<SysOnlineUser> _sysOnlineUerRep;
    private readonly SysConfigService _sysConfigService;
    private readonly IHubContext<OnlineUserHub, IOnlineUserHub> _onlineUserHubContext;
    private readonly UserManager _userManager;
    public SysOnlineUserService(SqlSugarRepository<SysOnlineUser> sysOnlineUerRep,
        SysConfigService sysConfigService,
        IHubContext<OnlineUserHub, IOnlineUserHub> onlineUserHubContext,
        UserManager userManager)
    {
        _sysOnlineUerRep = sysOnlineUerRep;
        _sysConfigService = sysConfigService;
        _onlineUserHubContext = onlineUserHubContext;
        _userManager = userManager;
    }

    /// <summary>
    /// 获取在线用户分页列表 🔖
    /// </summary>
    /// <returns></returns>
    [DisplayName("获取在线用户分页列表")]
    public async Task<SqlSugarPagedList<SysOnlineUser>> Page(PageOnlineUserInput input)
    {
        return await _sysOnlineUerRep.AsQueryable()
            .WhereIF(!string.IsNullOrWhiteSpace(input.UserName), u => u.UserName.Contains(input.UserName))
            .WhereIF(!string.IsNullOrWhiteSpace(input.RealName), u => u.RealName.Contains(input.RealName))
            .ToPagedListAsync(input.Page, input.PageSize);
    }



    /// <summary>
    /// 用户功能是否已到期
    /// </summary>
    /// <returns></returns>
    [DisplayName("用户功能是否已到期"), HttpGet]
    public async Task<OnlineUserHasitexpireDTO> Hasitexpired()
    {
        var userinfo = _userManager;
        var single = await _sysOnlineUerRep.AsQueryable().FirstAsync(e => e.UserId == userinfo.UserId);
        if (single == null)
        {
            var user = new SysOnlineUser
            {
                UserId = userinfo.UserId,
                TenantId = userinfo.TenantId,
                UserName = userinfo.Account,
                RealName = userinfo.RealName,
                Time = DateTime.Now,
                //  ConnectionId = Context.ConnectionId,
                // Ip = context.Connection.RemoteIpAddress?.MapToIPv4().ToString(),
                //      Browser = client.UA.Family + client.UA.Major,
                //   Os = client.OS.Family + client.OS.Major,
                TotalOnlineSeconds = 0
            };
            await _sysOnlineUerRep.InsertAsync(user);
            return new OnlineUserHasitexpireDTO()
            {
                IsExpired = true,
                Msg = "ok"
            };
        }
        else
        {
            if (single.TotalOnlineSeconds / 60 > 5)
            {
                return new OnlineUserHasitexpireDTO()
                {
                    IsExpired = true,
                    Msg = "您的AI功能已到期，请续费"
                };
            }
        }
        return new OnlineUserHasitexpireDTO()
        {
            IsExpired = true,
            Msg = "ok"
        };
    }
    /// <summary>
    /// 强制下线 🔖
    /// </summary>
    /// <param name="user"></param>
    /// <returns></returns>
    [NonValidation]
    [DisplayName("强制下线")]
    public async Task ForceOffline(SysOnlineUser user)
    {
        await _onlineUserHubContext.Clients.Client(user.ConnectionId).ForceOffline("强制下线");
        await _sysOnlineUerRep.DeleteAsync(user);
    }

    /// <summary>
    /// 发布站内消息
    /// </summary>
    /// <param name="notice"></param>
    /// <param name="userIds"></param>
    /// <returns></returns>
    [NonAction]
    public async Task PublicNotice(SysNotice notice, List<long> userIds)
    {
        var userList = await _sysOnlineUerRep.GetListAsync(u => userIds.Contains(u.UserId));
        if (!userList.Any()) return;

        foreach (var item in userList)
        {
            await _onlineUserHubContext.Clients.Client(item.ConnectionId).PublicNotice(notice);
        }
    }

    /// <summary>
    /// 单用户登录
    /// </summary>
    /// <returns></returns>
    [NonAction]
    public async Task SingleLogin(long userId)
    {
        if (await _sysConfigService.GetConfigValue<bool>(CommonConst.SysSingleLogin))
        {
            var users = await _sysOnlineUerRep.GetListAsync(u => u.UserId == userId);
            foreach (var user in users)
            {
                await ForceOffline(user);
            }
        }
    }

    /// <summary>
    /// 通过用户ID踢掉在线用户
    /// </summary>
    /// <param name="userId"></param>
    /// <returns></returns>
    [NonAction]
    public async Task ForceOffline(long userId)
    {
        var users = await _sysOnlineUerRep.GetListAsync(u => u.UserId == userId);
        foreach (var user in users)
        {
            await ForceOffline(user);
        }
    }
}