From 6151593c084d035c27c2b623fdf1de860efcca03 Mon Sep 17 00:00:00 2001 From: Alexey Khit Date: Fri, 13 Jan 2023 17:28:01 +0300 Subject: [PATCH] Fix ws lock on write and close --- cmd/api/ws.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cmd/api/ws.go b/cmd/api/ws.go index e9f67483..bf4f1329 100644 --- a/cmd/api/ws.go +++ b/cmd/api/ws.go @@ -106,6 +106,7 @@ type Transport struct { closed bool mx sync.Mutex + wrmx sync.Mutex onChange func() onWrite func(msg interface{}) @@ -122,9 +123,9 @@ func (t *Transport) OnWrite(f func(msg interface{})) { } func (t *Transport) Write(msg interface{}) { - t.mx.Lock() + t.wrmx.Lock() t.onWrite(msg) - t.mx.Unlock() + t.wrmx.Unlock() } func (t *Transport) Close() {