Initial_commit_SecMPS_v2

api_sqlsugar/VolPro.WebApi/Controllers/Hubs/HomePageMessageHub.cs

@@ -0,0 +1,98 @@
+using Microsoft.AspNetCore.Cors;
+using Microsoft.AspNetCore.SignalR;
+using System;
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using VolPro.Core.CacheManager;
+using VolPro.Core.Extensions;
+using VolPro.Core.ManageUser;
+using VolPro.Sys.IServices;
+
+namespace VolPro.WebApi.Controllers.Hubs
+{
+    /// <summary>
+    /// https://docs.microsoft.com/zh-cn/aspnet/core/signalr/introduction?view=aspnetcore-3.1
+    /// https://docs.microsoft.com/zh-cn/aspnet/core/signalr/javascript-client?view=aspnetcore-6.0&tabs=visual-studio
+    /// </summary>
+    public class HomePageMessageHub : Hub
+    {
+        private readonly ICacheService _cacheService;
+
+
+        //public static ConcurrentDictionary<string, string>  UserCache.ConnectionIds = new ConcurrentDictionary<string, string>();
+
+        /// <summary>
+        /// 构造 注入
+        /// </summary>
+        public HomePageMessageHub(ICacheService cacheService)
+        {
+            _cacheService = cacheService;
+        }
+
+        /// <summary>
+        /// 建立连接时异步触发
+        /// </summary>
+        /// <returns></returns>
+        public override async Task OnConnectedAsync()
+        {
+            //Console.WriteLine($"建立连接{Context.ConnectionId}");
+            UserCache.Add(Context);
+            //添加到一个组下
+            //await Groups.AddToGroupAsync(Context.ConnectionId, "SignalR Users");
+            //发送上线消息
+            //await Clients.All.SendAsync("ReceiveHomePageMessage", 1, new { title = "系统消息", content = $"{Context.ConnectionId} 上线" });
+            await base.OnConnectedAsync();
+        }
+
+        /// <summary>
+        /// 离开连接时异步触发
+        /// </summary>
+        /// <param name="ex"></param>
+        /// <returns></returns>
+        public override async Task OnDisconnectedAsync(Exception ex)
+        {
+            //Console.WriteLine($"断开连接{Context.ConnectionId}");
+            //从组中删除
+            // await Groups.RemoveFromGroupAsync(Context.ConnectionId, "SignalR Users");
+            //可自行调用下线业务处理方法...
+
+            await UserOffline();
+            //发送下线消息
+            //   await Clients.All.SendAsync("ReceiveHomePageMessage", 4, new { title = "系统消息", content = $"{Context.ConnectionId} 离线" });
+            await base.OnDisconnectedAsync(ex);
+        }
+
+
+        /// <summary>
+        /// 发送给指定的人
+        /// </summary>
+        /// <param name="username">BW_Core_System_user表的登陆帐号</param>
+        /// <param name="message">发送的消息</param>
+        /// <returns></returns>
+        public async Task<bool> SendHomeMessage(string username, string title, string message)
+        {
+            await Clients.Clients(UserCache.GetCnnectionIds(username)).SendAsync("ReceiveHomePageMessage", new
+            {
+                //   username,
+                title,
+                message,
+                date = DateTime.Now.ToString("yyyy-MM-dd HH:mm:sss")
+            });
+            return true;
+        }
+
+        /// <summary>
+        /// 断开连接
+        /// </summary>
+        /// <returns></returns>
+        public async Task<bool> UserOffline()
+        {
+            UserCache.Remove(Context);
+            await Task.CompletedTask;
+            return true;
+        }
+    }
+}

api_sqlsugar/VolPro.WebApi/Controllers/Hubs/SignalRUserController.cs

@@ -0,0 +1,41 @@
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.AspNetCore.SignalR;
+using System;
+using System.Threading.Tasks;
+using VolPro.Core;
+using VolPro.Core.Controllers.Basic;
+using VolPro.Core.Enums;
+using VolPro.Core.Extensions;
+using VolPro.Core.Filters;
+using VolPro.Entity.DomainModels;
+
+namespace VolPro.WebApi.Controllers.Hubs
+{
+    [Route("api/signalRUser")]
+    public class SignalRUserController : VolController
+    {
+        IHubContext<HomePageMessageHub> _hubClients;
+        public SignalRUserController(IHubContext<HomePageMessageHub> hubClients)
+        {
+            _hubClients = hubClients;
+        }
+        /// <summary>
+        /// 强制下线
+        /// </summary>
+        /// <param name="userName"></param>
+        /// <returns></returns>
+        [HttpGet, Route("loginout")]
+        [ApiActionPermission((nameof(Sys_User)), ActionPermissionOptions.Add | ActionPermissionOptions.Update)]
+        public async Task<IActionResult> Loginout(string userName)
+        {
+            await _hubClients.Clients.Clients(UserCache.GetCnnectionIds(userName)).SendAsync("ReceiveHomePageMessage", new
+            {
+                msg = "您已被强制下线,即将自动退出登录...".Translator(),
+                code = "-1",
+                value="logout",
+                date = DateTime.Now.ToString("yyyy-MM-dd HH:mm:sss")
+            });
+            return Content("操作成功".Translator());
+        }
+    }
+}

api_sqlsugar/VolPro.WebApi/Controllers/Hubs/UserCache.cs

@@ -0,0 +1,63 @@
+using Microsoft.AspNetCore.SignalR;
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+
+namespace VolPro.WebApi.Controllers.Hubs
+{
+    public static class UserCache
+    {
+        public static ConcurrentDictionary<string, string> ConnectionIds = new ConcurrentDictionary<string, string>();
+
+        public static ConcurrentDictionary<string, int> Online = new ConcurrentDictionary<string, int>();
+
+        /// <summary>
+        /// 根据用户名获取所有的客户端
+        /// </summary>
+        /// <param name="username"></param>
+        /// <returns></returns>
+        public static IEnumerable<string> GetCnnectionIds(string username)
+        {
+            foreach (var item in ConnectionIds)
+            {
+                if (item.Value == username)
+                {
+                    yield return item.Key;
+                }
+            }
+        }
+
+        public static int GetOnline(string username)
+        {
+
+            if (Online.TryGetValue(username, out int val))
+            {
+                return val;
+            }
+            return 0;
+        }
+
+
+
+        public static void Add(HubCallerContext context)
+        {
+            string userName = context.GetHttpContext().Request.Query["userName"].ToString();
+            if (string.IsNullOrEmpty(userName))
+            {
+                return;
+            }
+            Online[userName] = 1;
+            ConnectionIds[context.ConnectionId] = userName;
+        }
+
+        public static void Remove(HubCallerContext context)
+        {
+            var cid = context.ConnectionId;
+
+            //移除缓存
+            if (ConnectionIds.TryRemove(cid, out string value))
+            {
+                Online[value] = 0;
+            }
+        }
+    }
+}