Drop connection to peer if the stream is closed

iotaledger · Sep 14, 2022 · e263216 · e263216
1 parent f6ab41b
commit e263216
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/protocol/gossip/service.go b/pkg/protocol/gossip/service.go
@@ -562,6 +562,12 @@ func (s *Service) deregisterProtocol(peerID peer.ID) error {
 		return fmt.Errorf("unable to cleanly reset stream to %s: %w", peerID, err)
 	}
 
+	// Drop connection to peer since we no longer have a protocol stream to it
+	conn := proto.Stream.Conn()
+	if conn != nil {
+		return conn.Close()
+	}
+
 	return nil
 }