feat: allow user to set request timeout

Author: insthync

SimpleNetworkManager.NET/Network/BaseClientConnection.cs

@@ -90,7 +90,7 @@ public void OnDisconnected()
         /// </summary>
         internal abstract UniTask DisconnectAsync();
 
-        internal async UniTask<TResponse> SendRequestAsync<TResponse>(BaseRequestMessage request)
+        internal async UniTask<TResponse> SendRequestAsync<TResponse>(BaseRequestMessage request, int timeoutMs = 10_000)
             where TResponse : BaseResponseMessage
         {
             uint requestId = GetNewRequestId();
@@ -99,24 +99,22 @@ internal async UniTask<TResponse> SendRequestAsync<TResponse>(BaseRequestMessage
 
             // Waiting for the response
             BaseResponseMessage? response;
-            // 10 seconds timeout
-            int timeoutCountDown = 10_000;
             do
             {
-                if (timeoutCountDown <= 0)
+                if (timeoutMs <= 0)
                 {
                     response = null;
                     break;
                 }
                 await Task.Delay(100);
-                timeoutCountDown -= 100;
+                timeoutMs -= 100;
             }
             while (!_pendingResponses.TryRemove(requestId, out response));
             s_unassignedRequestIds.Enqueue(requestId);
 
             if (response == null)
             {
-                throw new TimeoutException($"Request timed out after 10 seconds (RequestId: {requestId}).");
+                throw new TimeoutException($"Request timed out after {timeoutMs} milliseconds (RequestId: {requestId}).");
             }
 
             if (response is not TResponse castedResponse)

SimpleNetworkManager.NET/Network/BaseNetworkClient.cs

@@ -39,13 +39,13 @@ public async UniTask DisconnectAsync()
             ClientConnection.Dispose();
         }
 
-        public async UniTask<TResponse> SendRequestAsync<TResponse>(BaseRequestMessage request)
+        public async UniTask<TResponse> SendRequestAsync<TResponse>(BaseRequestMessage request, int timeoutMs = 10_000)
             where TResponse : BaseResponseMessage
         {
             if (ClientConnection == null || !ClientConnection.IsConnected)
                 throw new InvalidOperationException("Cannot send request: client is not connected.");
             _messageRouterService.RegisterHandler(new ResponseMessageHandler<TResponse>(), true);
-            return await ClientConnection.SendRequestAsync<TResponse>(request);
+            return await ClientConnection.SendRequestAsync<TResponse>(request, timeoutMs);
         }
 
         public void RegisterHandler<T>(BaseMessageHandler<T> handler)

SimpleNetworkManager.NET/Network/BaseNetworkServer.cs

@@ -50,15 +50,15 @@ public async UniTask DisconnectAsync(uint connectionId)
             await clientConnection.DisconnectAsync();
         }
 
-        public async UniTask<TResponse> SendRequestAsync<TResponse>(uint connectionId, BaseRequestMessage request)
+        public async UniTask<TResponse> SendRequestAsync<TResponse>(uint connectionId, BaseRequestMessage request, int timeoutMs = 10_000)
             where TResponse : BaseResponseMessage
         {
             if (!_connectionManager.TryGetConnection(connectionId, out var clientConnection))
                 throw new KeyNotFoundException($"No connection found with ID {connectionId}.");
             if (clientConnection == null || !clientConnection.IsConnected)
                 throw new InvalidOperationException($"Cannot send request: client {connectionId} is not connected.");
             _messageRouterService.RegisterHandler(new ResponseMessageHandler<TResponse>(), true);
-            return await clientConnection.SendRequestAsync<TResponse>(request);
+            return await clientConnection.SendRequestAsync<TResponse>(request, timeoutMs);
         }
 
         public void RegisterHandler<T>(BaseMessageHandler<T> handler)