fix: remove listeners when client disconnects

handlers.go

@@ -60,6 +60,7 @@ func (s *Server) handleWebsocket(w http.ResponseWriter, r *http.Request) {
 			if _, ok := s.clients[conn]; ok {
 				conn.Close()
 				delete(s.clients, conn)
+				removeListener(ws)
 			}
 			s.clientsMu.Unlock()
 		}()
@@ -214,7 +215,7 @@ func (s *Server) handleWebsocket(w http.ResponseWriter, r *http.Request) {
 						return
 					}
 
-					removeListener(ws, id)
+					removeListenerId(ws, id)
 					break
 				default:
 					if cwh, ok := s.relay.(CustomWebSocketHandler); ok {

listener.go

@@ -63,7 +63,8 @@ func setListener(id string, ws *WebSocket, filters nostr.Filters) {
 	}
 }
 
-func removeListener(ws *WebSocket, id string) {
+// Remove a specific subscription id from listeners for a given ws client
+func removeListenerId(ws *WebSocket, id string) {
 	listenersMutex.Lock()
 	defer func() {
 		listenersMutex.Unlock()
@@ -78,6 +79,17 @@ func removeListener(ws *WebSocket, id string) {
 	}
 }
 
+// Remove WebSocket conn from listeners
+func removeListener(ws *WebSocket) {
+	listenersMutex.Lock()
+	defer listenersMutex.Unlock()
+
+	_, ok := listeners[ws]
+	if ok {
+		delete(listeners, ws)
+	}
+}
+
 func notifyListeners(event *nostr.Event) {
 	listenersMutex.Lock()
 	defer func() {