添加项目文件。

2026-04-30 21:08:28 +08:00
parent c60f5fe117
commit cc017b6495
327 changed files with 12860 additions and 0 deletions
@@ -0,0 +1,53 @@
+using ConnectorService.Services;
+using IM.Commons;
+using Microsoft.AspNetCore.SignalR;
+using StackExchange.Redis;
+using System.Security.Claims;
+
+namespace ConnectorService.Hubs
+{
+    public class ChatHub : Hub
+    {
+        private readonly IConversationIntergrationService conService;
+        private readonly StackExchange.Redis.IDatabase redis;
+
+        public ChatHub(IConversationIntergrationService conService, IConnectionMultiplexer multiplexer)
+        {
+            this.conService = conService;
+            this.redis = multiplexer.GetDatabase();
+        }
+
+        public async override Task OnConnectedAsync()
+        {
+            if (!Context.User.Identity.IsAuthenticated)
+            {
+                Context.Abort();
+                return;
+            }
+
+            var userId = Context.User.FindFirstValue(ClaimTypes.NameIdentifier);
+
+            var res = await conService.GetUserStreamKeysAsync(Guid.Parse(userId));
+            foreach (var streamkey in res)
+            {
+                await Groups.AddToGroupAsync(Context.ConnectionId, streamkey);
+            }
+
+            await redis.SetAddAsync(RedisHelper.GetConnectionIdKey(userId), Context.ConnectionId);
+
+
+            await base.OnConnectedAsync();
+        }
+
+        public async override Task OnDisconnectedAsync(Exception? exception)
+        {
+            if (Context.User.Identity.IsAuthenticated)
+            {
+                var userId = Context.User.FindFirstValue(ClaimTypes.NameIdentifier);
+
+                await redis.SetRemoveAsync(RedisHelper.GetConnectionIdKey(userId), Context.ConnectionId);
+            }
+            await base.OnDisconnectedAsync(exception);
+        }
+    }
+}