fix panic

gagliardetto · Jun 6, 2024 · ed51979 · ed51979
1 parent a5e17d6
commit ed51979
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/rpc/ws/client.go b/rpc/ws/client.go
@@ -246,7 +246,9 @@ func (c *Client) handleSubscriptionMessage(subID uint64, message []byte) {
 		return
 	}
 
-	sub.stream <- result
+	if !sub.closed {
+		sub.stream <- result
+	}
 	return
 }
 

diff --git a/rpc/ws/subscription.go b/rpc/ws/subscription.go
@@ -23,6 +23,7 @@ type Subscription struct {
 	stream            chan result
 	err               chan error
 	closeFunc         func(err error)
+	closed            bool
 	unsubscribeMethod string
 	decoderFunc       decoderFunc
 }
@@ -61,6 +62,7 @@ func (s *Subscription) Unsubscribe() {
 
 func (s *Subscription) unsubscribe(err error) {
 	s.closeFunc(err)
+	s.closed = true
 	close(s.stream)
 	close(s.err)
 }