Skip to content

Commit e361137

Browse files
committed
wsjs: Register OnError
Closes #400
1 parent 10137fa commit e361137

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

ws_js.go

+7
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ type Conn struct {
5555
closeWasClean bool
5656

5757
releaseOnClose func()
58+
releaseOnError func()
5859
releaseOnMessage func()
5960

6061
readSignal chan struct{}
@@ -92,9 +93,15 @@ func (c *Conn) init() {
9293
c.close(err, e.WasClean)
9394

9495
c.releaseOnClose()
96+
c.releaseOnError()
9597
c.releaseOnMessage()
9698
})
9799

100+
c.releaseOnError = c.ws.OnError(func(v js.Value) {
101+
c.setCloseErr(errors.New(v.Get("message").String()))
102+
c.closeWithInternal()
103+
})
104+
98105
c.releaseOnMessage = c.ws.OnMessage(func(e wsjs.MessageEvent) {
99106
c.readBufMu.Lock()
100107
defer c.readBufMu.Unlock()

0 commit comments

Comments
 (0)