From f6e1993eb6c0b1c1947cee65e7d5b837376a8d7d Mon Sep 17 00:00:00 2001
From: claudemiro <dimiro1@gmail.com>
Date: Fri, 16 Jan 2015 13:18:50 -0300
Subject: [PATCH] The websocket must be closed only when it is alive

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

diff --git a/websockets.go b/websockets.go
index 4e9ec79..4dfd2d7 100644
--- a/websockets.go
+++ b/websockets.go
@@ -204,11 +204,14 @@ func onMessage(conn *websocket.Conn, w http.ResponseWriter, r *http.Request, ses
 // Websocket GET /app/{key}
 func Websocket(w http.ResponseWriter, r *http.Request) {
 	conn, err := upgrader.Upgrade(w, r, nil)
-	defer conn.Close()
+	defer func() {
+		if conn != nil {
+			conn.Close()
+		}
+	}()
 
 	if err != nil {
 		log.Error(err)
-		emitWSError(NewGenericReconnectImmediatelyError(), conn)
 		return
 	}