Clamp cache size

gehongyan · Dec 19, 2024 · 25ff2ef · 25ff2ef
1 parent 7e913e2
commit 25ff2ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Kook.Net.WebSocket/Entities/Messages/MessageCache.cs b/src/Kook.Net.WebSocket/Entities/Messages/MessageCache.cs
@@ -14,7 +14,8 @@ internal class MessageCache
     public MessageCache(KookSocketClient kook)
     {
         _size = kook.MessageCacheSize;
-        _messages = new ConcurrentDictionary<Guid, SocketMessage>(ConcurrentHashSet.DefaultConcurrencyLevel, (int)(_size * 1.05));
+        int dictSize = _size * 1.05 > int.MaxValue ? int.MaxValue : (int)(_size * 1.05);
+        _messages = new ConcurrentDictionary<Guid, SocketMessage>(ConcurrentHashSet.DefaultConcurrencyLevel, dictSize);
         _orderedMessages = new ConcurrentQueue<(Guid MsgId, DateTimeOffset Timestamp)>();
     }