From ee2960e51b724ff231bdce874c690872310a05ca Mon Sep 17 00:00:00 2001
From: lifeengines <lifeengines@gmail.com>
Date: Sat, 27 Apr 2024 17:49:36 +0800
Subject: [PATCH] Update WebSocket.cs

made response header ended up before OnWsConnecting so we can append extra handshake header to response
---
 source/NetCoreServer/WebSocket.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/source/NetCoreServer/WebSocket.cs b/source/NetCoreServer/WebSocket.cs
index ab375f0..c19327b 100644
--- a/source/NetCoreServer/WebSocket.cs
+++ b/source/NetCoreServer/WebSocket.cs
@@ -232,12 +232,13 @@ public bool PerformServerUpgrade(HttpRequest request, HttpResponse response)
             response.SetHeader("Connection", "Upgrade");
             response.SetHeader("Upgrade", "websocket");
             response.SetHeader("Sec-WebSocket-Accept", accept);
-            response.SetBody();
 
             // Validate WebSocket upgrade request and response
             if (!_wsHandler.OnWsConnecting(request, response))
                 return false;
-
+            
+            response.SetBody();
+            
             // Send WebSocket upgrade response
             _wsHandler.SendUpgrade(response);