mirror of
https://github.com/eosswedenorg/thalos
synced 2026-06-16 04:24:56 +02:00
api/client.go: make Subscribe support a list of channels as argument.
This commit is contained in:
parent
133af980a3
commit
816d405d31
1 changed files with 10 additions and 1 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue