Add ReceiverUpradeAvailable() API
This commit is contained in:
parent
1bf205347e
commit
055ee0000d
|
@ -0,0 +1,47 @@
|
||||||
|
package reach
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"forge.cadoles.com/Pyxis/golang-socketio"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
eventIsReceiverUpgradeAvailable = "is receiver upgrade available"
|
||||||
|
eventReceiverUpgradeAvailable = "receiver upgrade available"
|
||||||
|
)
|
||||||
|
|
||||||
|
type receiverUpgreAvailable struct {
|
||||||
|
Available bool `json:"available"`
|
||||||
|
Running bool `json:"running"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReceiverUpgradeAvailable checks if an upgrade is avaialable/running for the ReachRS module
|
||||||
|
func (c *Client) ReceiverUpgradeAvailable() (available bool, running bool, err error) {
|
||||||
|
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
|
||||||
|
wg.Add(1)
|
||||||
|
|
||||||
|
err = c.conn.On(eventReceiverUpgradeAvailable, func(h *gosocketio.Channel, up *receiverUpgreAvailable) {
|
||||||
|
available = up.Available
|
||||||
|
running = up.Running
|
||||||
|
c.conn.Off(eventReceiverUpgradeAvailable)
|
||||||
|
wg.Done()
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return false, false, errors.Wrapf(err, "error while binding to '%s' event", eventReceiverUpgradeAvailable)
|
||||||
|
}
|
||||||
|
|
||||||
|
c.logf("sending '%s' event", eventIsReceiverUpgradeAvailable)
|
||||||
|
if err = c.conn.Emit(eventIsReceiverUpgradeAvailable, nil); err != nil {
|
||||||
|
return false, false, errors.Wrapf(err, "error while emitting '%s' event", eventIsReceiverUpgradeAvailable)
|
||||||
|
}
|
||||||
|
c.logf("'%s' event sent", eventIsReceiverUpgradeAvailable)
|
||||||
|
|
||||||
|
wg.Wait()
|
||||||
|
|
||||||
|
return available, running, err
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,31 @@
|
||||||
|
package reach
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestClientReceiverUpgradeAvailable(t *testing.T) {
|
||||||
|
|
||||||
|
if !*runIntegrationTests {
|
||||||
|
t.Skip("To run this test, use: go test -integration")
|
||||||
|
}
|
||||||
|
|
||||||
|
client := NewClient(
|
||||||
|
WithStandardLogger(),
|
||||||
|
WithEndpoint(*reachHost, 80),
|
||||||
|
)
|
||||||
|
if err := client.Connect(); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
available, running, err := client.ReceiverUpgradeAvailable()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("receiver upgrade result: available: %v, running: %v", available, running)
|
||||||
|
|
||||||
|
defer client.Close()
|
||||||
|
|
||||||
|
}
|
Reference in New Issue