eosswedenorg/thalos
1
0
Fork 0
mirror of https://github.com/eosswedenorg/thalos synced 2026-06-20 09:56:47 +02:00
Code Releases Activity

api/client.go: make Subscribe support a list of channels as argument.

Browse source
This commit is contained in:
Henrik Hautakoski 2024-02-04 22:11:37 +01:00
parent 133af980a3
commit 816d405d31
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/client.go
View file

@ -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 var handler handler
switch channel.Type() { switch channel.Type() {
@ -131,6 +131,15 @@ func (c *Client) Subscribe(channel Channel) error {
return nil 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() { func (c *Client) Run() {
// Just wait for workers to complete. // Just wait for workers to complete.
c.wg.Wait() c.wg.Wait()