go-aah/aah#126 added isdisconnected method

engine.go

@@ -46,6 +46,8 @@ var (
 	ErrAuthenticationFailed   = errors.New("aahws: authentication failed")
 	ErrWebSocketNotFound      = errors.New("aahws: not found")
 	ErrWebSocketConnectFailed = errors.New("aahws: connect failed")
+	ErrConnectionClosed       = errors.New("aahws: connection closed")
+	ErrUseOfClosedConnection  = errors.New("aahws: use of closed ws connection")
 )
 
 // EventCallbackFunc func type used for all WebSocket event callback.
@@ -121,7 +123,7 @@ func (e *Engine) SetAuthCallback(ac AuthCallbackFunc) {
 	e.doAuth = ac
 }
 
-// Connect method primarly does upgrades HTTP connection into WebSocket
+// Connect method primarily does upgrades HTTP connection into WebSocket
 // connection.
 //
 // Along with Check Origin, Authentication Callback and aah WebSocket events

util.go

@@ -5,8 +5,8 @@
 package ws
 
 import (
-	"errors"
 	"fmt"
+	"io"
 	"net/http"
 	"strconv"
 	"strings"
@@ -18,6 +18,18 @@ import (
 // Package methods
 //______________________________________________________________________________
 
+func IsDisconnected(err error) bool {
+	switch err {
+	case ErrConnectionClosed, ErrUseOfClosedConnection:
+		return true
+	}
+	return false
+}
+
+//‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾
+// Package Unexported methods
+//______________________________________________________________________________
+
 // WriteHTTPError is to write WebSocket context error.
 func writeHTTPError(w http.ResponseWriter, code int, body string) {
 	w.Header().Set(ahttp.HeaderContentType, ahttp.ContentTypePlainText.String())
@@ -38,9 +50,9 @@ func createError(err error) error {
 
 	msg := err.Error()
 	if strings.HasPrefix(msg, "ws closed") {
-		return errors.New("aahws: connection closed")
-	} else if strings.HasSuffix(msg, "use of closed network connection") {
-		return errors.New("aahws: use of closed ws connection")
+		return ErrConnectionClosed
+	} else if err == io.EOF || strings.HasSuffix(msg, "use of closed network connection") {
+		return ErrUseOfClosedConnection
 	}
 	return fmt.Errorf("aah%s", msg)
 }