From cbbe2ded586cc0ff6409fc4de18789e50dd7d4e4 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 5 Nov 2023 00:48:03 +0800
Subject: [PATCH] feat: optimize error processing

---
 src/main.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 4d9b0af..c025ad9 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -242,7 +242,7 @@ async fn inner_handle_connection(
 ) -> XResult<()> {
     information!("Incoming TCP connection from: {}", addr);
 
-    let ws_stream = tokio_tungstenite::accept_async(raw_stream).await?;
+    let ws_stream = opt_result!(tokio_tungstenite::accept_async(raw_stream).await, "Accept websocket failed: {}");
     information!("WebSocket connection established: {}", addr);
 
     // Insert the write part of this peer to the peer map.