diff options
| author | Mel <einebeere@gmail.com> | 2022-04-12 17:21:05 +0200 |
|---|---|---|
| committer | Mel <einebeere@gmail.com> | 2022-04-12 17:21:05 +0200 |
| commit | 6163d259ed52991e2f95632b5a0516607aa56a5f (patch) | |
| tree | d87514d024d55f976ec78176fd0b0d8ebe946d7c /pkg/libs/cancellablewebsocket | |
| parent | 6cd2890450aaf71e97004d421237996f0a42d04c (diff) | |
| download | jinx-6163d259ed52991e2f95632b5a0516607aa56a5f.tar.zst jinx-6163d259ed52991e2f95632b5a0516607aa56a5f.zip | |
Handle gateway errors and reconnections
Diffstat (limited to 'pkg/libs/cancellablewebsocket')
| -rw-r--r-- | pkg/libs/cancellablewebsocket/cancellablewebsocket.go | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/pkg/libs/cancellablewebsocket/cancellablewebsocket.go b/pkg/libs/cancellablewebsocket/cancellablewebsocket.go index accc180..eb9a5be 100644 --- a/pkg/libs/cancellablewebsocket/cancellablewebsocket.go +++ b/pkg/libs/cancellablewebsocket/cancellablewebsocket.go @@ -79,6 +79,10 @@ func (cws *CancellableWebSocket) Close(code int) error { return nil } +func (cws *CancellableWebSocket) OnClose(f func(code int, text string) error) { + cws.conn.SetCloseHandler(f) +} + func (cws *CancellableWebSocket) listenForCancel() { <-cws.ctx.Done() |