From 6163d259ed52991e2f95632b5a0516607aa56a5f Mon Sep 17 00:00:00 2001 From: Mel Date: Tue, 12 Apr 2022 17:21:05 +0200 Subject: Handle gateway errors and reconnections --- pkg/libs/cancellablewebsocket/cancellablewebsocket.go | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'pkg/libs/cancellablewebsocket/cancellablewebsocket.go') 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() -- cgit 1.4.1