Fix ws lock on write and close

This commit is contained in:
Alexey Khit
2023-01-13 17:28:01 +03:00
parent dba0989c54
commit 6151593c08
+3 -2
View File
@@ -106,6 +106,7 @@ type Transport struct {
closed bool closed bool
mx sync.Mutex mx sync.Mutex
wrmx sync.Mutex
onChange func() onChange func()
onWrite func(msg interface{}) onWrite func(msg interface{})
@@ -122,9 +123,9 @@ func (t *Transport) OnWrite(f func(msg interface{})) {
} }
func (t *Transport) Write(msg interface{}) { func (t *Transport) Write(msg interface{}) {
t.mx.Lock() t.wrmx.Lock()
t.onWrite(msg) t.onWrite(msg)
t.mx.Unlock() t.wrmx.Unlock()
} }
func (t *Transport) Close() { func (t *Transport) Close() {