Remove default close handler which caused abnormal closes

2017-04-12 14:41:55 -07:00
parent 647ca280c6
commit a9e9ebdfdc
2 changed files with 6 additions and 5 deletions
@@ -115,10 +115,11 @@ func (s *Session) readPump() {
 		return nil
 	})

-	s.conn.SetCloseHandler(func(code int, text string) error {
-		s.melody.closeHandler(s, code, text)
-		return nil
-	})
+	if s.melody.closeHandler != nil {
+		s.conn.SetCloseHandler(func(code int, text string) error {
+			return s.melody.closeHandler(s, code, text)
+		})
+	}

 	for {
 		t, message, err := s.conn.ReadMessage()