From e5311338d136bb514b0b9449cb054aa1055340d5 Mon Sep 17 00:00:00 2001
From: daheige <zhuwei313@hotmail.com>
Date: Sat, 5 Jun 2021 10:48:13 +0800
Subject: [PATCH] IsWebsocket func return

---
 context.go | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/context.go b/context.go
index 0c1fb07f..a6775abb 100644
--- a/context.go
+++ b/context.go
@@ -814,11 +814,8 @@ func (c *Context) ContentType() string {
 // IsWebsocket returns true if the request headers indicate that a websocket
 // handshake is being initiated by the client.
 func (c *Context) IsWebsocket() bool {
-	if strings.Contains(strings.ToLower(c.requestHeader("Connection")), "upgrade") &&
-		strings.EqualFold(c.requestHeader("Upgrade"), "websocket") {
-		return true
-	}
-	return false
+	return strings.Contains(strings.ToLower(c.requestHeader("Connection")), "upgrade") &&
+		strings.EqualFold(c.requestHeader("Upgrade"), "websocket")
 }
 
 func (c *Context) requestHeader(key string) string {