Prevent multiple close of channel
Pyxis/orion/pipeline/head There was a failure building this commit Details

This commit is contained in:
wpetit 2021-01-26 18:01:06 +01:00
parent e7bec24f0f
commit c659a98587
1 changed files with 9 additions and 2 deletions

View File

@ -2,6 +2,7 @@ package reachview
import (
"context"
"sync"
gosocketio "forge.cadoles.com/Pyxis/golang-socketio"
"github.com/mitchellh/mapstructure"
@ -31,6 +32,8 @@ type Broadcast struct {
func (c *Client) Broadcast(ctx context.Context) (chan Broadcast, error) {
out := make(chan Broadcast)
closer := new(sync.Once)
handler := func(_ *gosocketio.Channel, data interface{}) {
res := Broadcast{}
if err := mapstructure.WeakDecode(data, &res); err != nil {
@ -38,8 +41,12 @@ func (c *Client) Broadcast(ctx context.Context) (chan Broadcast, error) {
}
select {
case <-ctx.Done():
closer.Do(func() {
c.Off(eventBroadcast)
close(out)
})
return
default:
out <- res
}