[Modify] Throw exception

8 years ago · 189d8b38ec
parent 814b64e019
commit 189d8b38ec
1 changed files with 6 additions and 2 deletions
--- a/websocket-sharp/Server/WebSocketSessionManager.cs
+++ b/websocket-sharp/Server/WebSocketSessionManager.cs
@ -880,8 +880,12 @@ namespace WebSocketSharp.Server
    public void CloseSession (string id, ushort code, string reason)
    {
      IWebSocketSession session;
-      if (TryGetSession (id, out session))
-        session.Context.WebSocket.Close (code, reason);
+      if (!TryGetSession (id, out session)) {
+        var msg = "The session could not be found.";
+        throw new ArgumentException (msg, "id");
+      }
+
+      session.Context.WebSocket.Close (code, reason);
    }

    /// <summary>