From fd409f6c3527bc4b320b48c95becca04742f3168 Mon Sep 17 00:00:00 2001
From: fiatjaf <fiatjaf@gmail.com>
Date: Thu, 22 Jun 2023 09:32:12 -0300
Subject: [PATCH] add closure queue channel that was missing.

---
 relay.go | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/relay.go b/relay.go
index c4f88b8..3e1f147 100644
--- a/relay.go
+++ b/relay.go
@@ -68,12 +68,13 @@ type writeRequest struct {
 func NewRelay(ctx context.Context, url string, opts ...RelayOption) *Relay {
 	ctx, cancel := context.WithCancel(ctx)
 	r := &Relay{
-		URL:                     NormalizeURL(url),
-		connectionContext:       ctx,
-		connectionContextCancel: cancel,
-		Subscriptions:           xsync.NewMapOf[*Subscription](),
-		okCallbacks:             xsync.NewMapOf[func(bool, string)](),
-		writeQueue:              make(chan writeRequest),
+		URL:                           NormalizeURL(url),
+		connectionContext:             ctx,
+		connectionContextCancel:       cancel,
+		Subscriptions:                 xsync.NewMapOf[*Subscription](),
+		okCallbacks:                   xsync.NewMapOf[func(bool, string)](),
+		writeQueue:                    make(chan writeRequest),
+		subscriptionChannelCloseQueue: make(chan *Subscription),
 	}
 
 	for _, opt := range opts {
@@ -141,7 +142,7 @@ func (_ WithAuthHandler) IsRelayOption() {}
 
 var _ RelayOption = (WithAuthHandler)(nil)
 
-// String() just prints the relay URL.
+// String just returns the relay URL.
 func (r *Relay) String() string {
 	return r.URL
 }
@@ -480,6 +481,7 @@ func (r *Relay) PrepareSubscription(ctx context.Context, filters Filters, opts .
 	}
 
 	id := sub.GetID()
+
 	r.Subscriptions.Store(id, sub)
 
 	// the subscription ends once the context is canceled