diff --git a/main.go b/main.go index 6e653fe..a6cc153 100644 --- a/main.go +++ b/main.go @@ -173,11 +173,11 @@ func (h *Handler) LoadPeers(filename string) error { func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) { wsConn, err := websocket.Accept(w, r, nil) - wsConn.SetReadLimit(-1) if err != nil { log.WithField("client", r.RemoteAddr).Error(err) return } + wsConn.SetReadLimit(-1) go h.ServeWebSocket(wsConn, r.RemoteAddr) }