Do'h; do it correct

arp242 · Feb 16, 2022 · e59c59b · e59c59b
1 parent 0b6b90a
commit e59c59b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/handlers/loader.go b/handlers/loader.go
@@ -99,14 +99,16 @@ func (l *loaderT) sendJSON(r *http.Request, id zint.Uint128, data interface{}) {
 	defer c.Unlock()
 	w, err := c.conn.NextWriter(websocket.TextMessage)
 	if err != nil {
+		if w != nil {
+			w.Close()
+		}
 		zlog.Fields(zlog.F{
 			"connectID": id,
 			"siteID":    Site(r.Context()).ID,
 			"userID":    User(r.Context()).ID,
 		}).FieldsRequest(r).Errorf("loader.send: NextWriter: %s", err)
 		return
 	}
-	defer w.Close()
 
 	j, err := json.Marshal(data)
 	if err != nil {
@@ -119,6 +121,7 @@ func (l *loaderT) sendJSON(r *http.Request, id zint.Uint128, data interface{}) {
 	}
 
 	_, err = w.Write(j)
+	w.Close()
 	if err != nil {
 		zlog.Fields(zlog.F{
 			"connectID": id,