diff --git a/api/client.go b/api/client.go index 146b4aa..db79dc7 100644 --- a/api/client.go +++ b/api/client.go @@ -103,7 +103,7 @@ func (c *Client) hbHandler(payload []byte) { } } -func (c *Client) Subscribe(channel Channel) error { +func (c *Client) sub(channel Channel) error { var handler handler switch channel.Type() { @@ -131,6 +131,15 @@ func (c *Client) Subscribe(channel Channel) error { return nil } +func (c *Client) Subscribe(channels ...Channel) error { + for _, ch := range channels { + if err := c.sub(ch); err != nil { + return err + } + } + return nil +} + func (c *Client) Run() { // Just wait for workers to complete. c.wg.Wait()