Update recovery.go

celestiaorg · Jan 25, 2025 · 4a1d93a · 4a1d93a
1 parent 77bcc9d
commit 4a1d93a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/share/shwap/p2p/shrex/recovery.go b/share/shwap/p2p/shrex/recovery.go
@@ -14,6 +14,8 @@ func RecoveryMiddleware(handler network.StreamHandler) network.StreamHandler {
 			if r != nil {
 				err := fmt.Errorf("PANIC while handling request: %s", r)
 				log.Error(err)
+				// Close the stream to prevent resource leaks
+				_ = stream.Close()
 			}
 		}()
 		handler(stream)