From 816d405d31e3ef6d104f2facd4351f6db6287404 Mon Sep 17 00:00:00 2001
From: Henrik Hautakoski <henrik.hautakoski@gmail.com>
Date: Sun, 4 Feb 2024 22:11:37 +0100
Subject: [PATCH] api/client.go: make Subscribe support a list of channels as
 argument.

---
 api/client.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

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