feat: 完善聊天模块

2024-04-06 18:28:32 +08:00
parent 43b4032bbb
commit 6aedff75f1
17 changed files with 507 additions and 110 deletions
--- a/Yi.Abp.Net8/module/chat-hub/Yi.Framework.ChatHub.Domain/Managers/UserMessageManager.cs
+++ b/Yi.Abp.Net8/module/chat-hub/Yi.Framework.ChatHub.Domain/Managers/UserMessageManager.cs
@@ -0,0 +1,66 @@
+using FreeRedis;
+using Mapster;
+using Microsoft.AspNetCore.SignalR;
+using Microsoft.Extensions.Options;
+using Volo.Abp.Caching;
+using Volo.Abp.Domain.Services;
+using Yi.Framework.ChatHub.Domain.Shared.Caches;
+using Yi.Framework.ChatHub.Domain.Shared.Consts;
+using Yi.Framework.ChatHub.Domain.Shared.Model;
+using Yi.Framework.ChatHub.Domain.SignalRHubs;
+
+namespace Yi.Framework.ChatHub.Domain.Managers
+{
+    public class UserMessageManager : DomainService
+    {
+        private IHubContext<ChatCenterHub> _hubContext;
+        public UserMessageManager(IHubContext<ChatCenterHub> hubContext)
+        {
+            _hubContext = hubContext;
+        }
+        /// <summary>
+        /// 使用懒加载防止报错
+        /// </summary>
+        private IRedisClient RedisClient => LazyServiceProvider.LazyGetRequiredService<IRedisClient>();
+        private string CacheKeyPrefix => LazyServiceProvider.LazyGetRequiredService<IOptions<AbpDistributedCacheOptions>>().Value.KeyPrefix;
+
+        public async Task SendMessageAsync(MessageContext context)
+        {
+            switch (context.MessageType)
+            {
+                case MessageType.Personal:
+                    var userModel =await GetUserAsync(context.ReceiveId.Value);
+                    if (userModel is not null)
+                    {
+                        await _hubContext.Clients.Client(userModel.ClientId).SendAsync(ChatConst.ClientActionReceiveMsg, context.MessageType, context);
+                    }
+                    break;
+                case MessageType.Group:
+                    throw new NotImplementedException();
+                    break;
+                case MessageType.All:
+                    await _hubContext.Clients.All.SendAsync(ChatConst.ClientActionReceiveMsg, context.MessageType, context);
+                    break;
+                default:
+                    break;
+            }
+
+        }
+
+        public async Task<List<ChatOnlineUserCacheItem>> GetAllUserAsync()
+        {
+            var key = new ChatOnlineUserCacheKey(CacheKeyPrefix);
+            var cacheUsers = (await RedisClient.HGetAllAsync(key.GetKey())).Select(x => System.Text.Json.JsonSerializer.Deserialize<ChatOnlineUserCacheItem>(x.Value)).ToList();
+            return cacheUsers;
+        }
+
+        public async Task<ChatOnlineUserCacheItem?> GetUserAsync(Guid userId)
+        {
+            var key = new ChatOnlineUserCacheKey(CacheKeyPrefix);
+            var cacheUser = System.Text.Json.JsonSerializer.Deserialize < ChatOnlineUserCacheItem >( await RedisClient.HGetAsync(key.GetKey(), key.GetField(userId)));
+            return cacheUser;
+        }
+    }
+
+
+}