clear topic when no client relates with it (#13)

RabbyHub · Dec 5, 2023 · 0d8ecc2 · 0d8ecc2
1 parent 22e0da3
commit 0d8ecc2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/relay/wshandlers.go b/relay/wshandlers.go
@@ -141,12 +141,14 @@ func (ws *WsServer) handleClientDisconnect(client *client) {
 	for topic := range subscribedTopics {
 		ws.subscribers.Unset(topic, client)
 		if ws.subscribers.Len(topic) == 0 {
+			ws.subscribers.Clear(topic)
 			channelsToClear = append(channelsToClear, messageChanKey(topic))
 		}
 	}
 	for topic := range client.pubTopics.Get() {
 		ws.publishers.Unset(topic, client)
 		if ws.publishers.Len(topic) == 0 {
+			ws.publishers.Clear(topic)
 			channelsToClear = append(channelsToClear, messageChanKey(topic))
 			// for dapp, need to further clear notify channels
 			if client.role == Dapp {