From 480efadf49419feef8d82c375f6bf1b90a4a153a Mon Sep 17 00:00:00 2001
From: wailovet <19940108>
Date: Mon, 16 Sep 2019 11:25:55 +0800
Subject: [PATCH] fix

---
 src/core/router.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/core/router.go b/src/core/router.go
index a7a69a8..ca805d7 100644
--- a/src/core/router.go
+++ b/src/core/router.go
@@ -78,7 +78,7 @@ func (rm *RouterManage) RouterSend(urlPath string, request Request, response Res
 	vc := reflect.New(rm.RegisteredData[ctr])
 
 	wsinit := vc.MethodByName("WebSocketInit")
-	if wsinit.IsValid() {
+	if wsinit.IsValid() || request.HEADER["Upgrade"] == "websocket" {
 		response.IsWebSocket = true
 		hand := vc.Interface().(interfaces.WebSocketInterface)
 		ws := GetWebSocket(ctr+"-"+fun, hand)