53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
package reader
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"github.com/iMDT/bbb-graphql-middleware/internal/common"
|
|
log "github.com/sirupsen/logrus"
|
|
"nhooyr.io/websocket"
|
|
"nhooyr.io/websocket/wsjson"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
func BrowserConnectionReader(browserConnectionId string, ctx context.Context, ctxCancel context.CancelFunc, browserWsConn *websocket.Conn, fromBrowserToHasuraChannel *common.SafeChannel, fromBrowserToHasuraConnectionEstablishingChannel *common.SafeChannel, waitGroups []*sync.WaitGroup) {
|
|
log := log.WithField("_routine", "BrowserConnectionReader").WithField("browserConnectionId", browserConnectionId)
|
|
defer log.Debugf("finished")
|
|
log.Debugf("starting")
|
|
|
|
defer func() {
|
|
fromBrowserToHasuraChannel.Close()
|
|
fromBrowserToHasuraConnectionEstablishingChannel.Close()
|
|
}()
|
|
|
|
defer func() {
|
|
// Let other routines know this is about to die
|
|
for _, wg := range waitGroups {
|
|
wg.Done()
|
|
}
|
|
// Wait a little bit before closing the channels
|
|
time.Sleep(100 * time.Millisecond)
|
|
}()
|
|
|
|
defer ctxCancel()
|
|
|
|
for {
|
|
var v interface{}
|
|
err := wsjson.Read(ctx, browserWsConn, &v)
|
|
if err != nil {
|
|
if errors.Is(err, context.Canceled) {
|
|
log.Debugf("Closing Browser ws connection as Context was cancelled!")
|
|
} else {
|
|
log.Debugf("Hasura is disconnected, skipping reading of ws message: %v", err)
|
|
}
|
|
return
|
|
}
|
|
|
|
log.Tracef("received from browser: %v", v)
|
|
|
|
fromBrowserToHasuraChannel.Send(v)
|
|
fromBrowserToHasuraConnectionEstablishingChannel.Send(v)
|
|
}
|
|
}
|