fix race condition in Close

nbd-wtf · Nov 5, 2023 · 20d65d4 · 20d65d4
1 parent 29646d8
commit 20d65d4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/relay.go b/relay.go
@@ -39,6 +39,8 @@ func (s Status) String() string {
 }
 
 type Relay struct {
+	mu sync.Mutex
+
 	URL           string
 	RequestHeader http.Header // e.g. for origin header
 
@@ -553,6 +555,9 @@ func (r *Relay) Count(ctx context.Context, filters Filters, opts ...Subscription
 }
 
 func (r *Relay) Close() error {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+
 	if r.connectionContextCancel == nil {
 		return fmt.Errorf("relay not connected")
 	}