1
0
Fork 0
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:
Henrik Hautakoski 2024-02-04 22:11:37 +01:00
parent 133af980a3
commit 816d405d31

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
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()