chore: restore previous state

Signed-off-by: D4ryl00 <[email protected]>
berty · Aug 5, 2024 · d3037f9 · d3037f9
1 parent 7cfbd32
commit d3037f9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/sync_inmem_provider.go b/sync_inmem_provider.go
@@ -117,18 +117,14 @@ func (ps *PubSub) handleStream(s inet.Stream) {
 	for {
 		var req pb.Message
 
-		defer func() {
+		err := r.ReadMsg(&req)
+		if err != nil {
 			for ns := range subscribedTopics {
 				topic := ps.getOrCreateTopic(ns)
 				topic.mu.Lock()
 				delete(topic.subscribers, s.Conn().RemotePeer())
 				topic.mu.Unlock()
 			}
-		}()
-
-		err := r.ReadMsg(&req)
-		if err != nil {
-			log.Errorf("unable to read from stream: %s", err.Error())
 			return
 		}