package reachview import ( "context" "sync" "forge.cadoles.com/Pyxis/golang-socketio" "github.com/pkg/errors" ) const ( eventReboot = "reboot" ) // Reboot asks the ReachRS module to reboot func (c *Client) Reboot(ctx context.Context, waitDisconnect bool) error { var err error var wg sync.WaitGroup if waitDisconnect { var once sync.Once done := func() { c.Off(gosocketio.OnDisconnection) wg.Done() } wg.Add(1) go func() { <-ctx.Done() err = ctx.Err() once.Do(done) }() err = c.On(gosocketio.OnDisconnection, func(h *gosocketio.Channel) { once.Do(done) }) if err != nil { return errors.Wrapf(err, "error while binding to '%s' event", gosocketio.OnDisconnection) } } if err = c.Emit(eventReboot, nil); err != nil { return err } if waitDisconnect { wg.Wait() } return err }