checkIsWebSocketRequest update

zhoulindhq · May 10, 2017 · 9dd2ca9 · 9dd2ca9
1 parent d704a64
commit 9dd2ca9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server.go b/server.go
@@ -467,8 +467,9 @@ func logRequest(req *http.Request, timetaken int64) string {
 }
 
 //check request is the websocket request
+//check Connection contains upgrade
 func checkIsWebSocketRequest(req *http.Request) bool {
-	if req.Header.Get("Connection") == "Upgrade" {
+	if strings.Index(strings.ToLower(req.Header.Get("Connection")), "upgrade") >= 0 {
 		return true
 	}
 	return false