github/wails
5
0
Fork 0
mirror of https://github.com/wailsapp/wails.git synced 2025-05-02 19:50:15 +08:00
Code Projects Activity

fix: use mutex to serialise websocket writes

Browse Source
This commit is contained in:
Lea Anthony 2019-03-19 08:35:09 +11:00
parent 9f5e2c7dd4
commit fec3e7eac5
No known key found for this signature in database
GPG Key ID: 33DAF7BB90A58405
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
renderer_headless.go
View File

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"net/http" "net/http"
"strings" "strings"
"sync"
"github.com/dchest/htmlmin" "github.com/dchest/htmlmin"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
@ -45,6 +46,9 @@ type Headless struct {
initialisationJS []string initialisationJS []string
server *http.Server server *http.Server
theConnection *websocket.Conn theConnection *websocket.Conn
// Mutex for writing to the socket
lock sync.Mutex
} }
// Initialise the Headless Renderer // Initialise the Headless Renderer
@ -103,6 +107,10 @@ func (h *Headless) wsBridgeHandler(w http.ResponseWriter, r *http.Request) {
} }
func (h *Headless) sendMessage(conn *websocket.Conn, msg string) { func (h *Headless) sendMessage(conn *websocket.Conn, msg string) {
h.lock.Lock()
defer h.lock.Unlock()
if err := conn.WriteMessage(websocket.TextMessage, []byte(msg)); err != nil { if err := conn.WriteMessage(websocket.TextMessage, []byte(msg)); err != nil {
h.log.Error(err.Error()) h.log.Error(err.Error())
} }