From 0c230e12ec38f267aa2cfeaf0634e934b102f3ed Mon Sep 17 00:00:00 2001
From: Azareal <azareal@users.noreply.github.com>
Date: Thu, 16 Jul 2020 20:51:34 +1000
Subject: [PATCH] add WSUser.WriteToPageBytesMulti

---
 common/ws_user.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/common/ws_user.go b/common/ws_user.go
index 5ac9fe20..11cc0df1 100644
--- a/common/ws_user.go
+++ b/common/ws_user.go
@@ -80,6 +80,32 @@ func (u *WSUser) WriteToPageBytes(msg []byte, page string) error {
 	return nil
 }
 
+// Inefficient as it looks for sockets for a page even if there are none
+func (u *WSUser) WriteToPageBytesMulti(msgs [][]byte, page string) error {
+	var success bool
+	for _, socket := range u.Sockets {
+		if socket == nil {
+			continue
+		}
+		if socket.Page != page {
+			continue
+		}
+		w, err := socket.conn.NextWriter(websocket.TextMessage)
+		if err != nil {
+			continue // Skip dead sockets, a dedicated goroutine handles those
+		}
+		for _, msg := range msgs {
+			_, _ = w.Write(msg)
+		}
+		w.Close()
+		success = true
+	}
+	if !success {
+		return ErrNoneOnPage
+	}
+	return nil
+}
+
 func (u *WSUser) AddSocket(conn *websocket.Conn, page string) {
 	u.Lock()
 	// If the number of the sockets is small, then we can keep the size of the slice mostly static and just walk through it looking for empty slots