From f174863a17859ea22f9da4bc0b7c102170509394 Mon Sep 17 00:00:00 2001 From: Henrik Hautakoski Date: Wed, 29 Mar 2023 11:57:51 +0200 Subject: [PATCH] transport/redis_pubsub/subscriber.go: implement Close() --- transport/redis_pubsub/subscriber.go | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/transport/redis_pubsub/subscriber.go b/transport/redis_pubsub/subscriber.go index dfdcac3..83e27b1 100644 --- a/transport/redis_pubsub/subscriber.go +++ b/transport/redis_pubsub/subscriber.go @@ -95,3 +95,14 @@ func (s *Subscriber) Read(channel transport.Channel) ([]byte, error) { return <-ch, nil } + +func (s *Subscriber) Close() { + s.sub.Close() + + for _, ch := range s.channels { + close(ch) + } + s.mu.Lock() + s.channels = make(map[string]chan []byte) + s.mu.Unlock() +}