From 608a2904d44ce6ca906b323293c753dc2f6cdc24 Mon Sep 17 00:00:00 2001
From: tsosunchia <59512455+tsosunchia@users.noreply.github.com>
Date: Sun, 13 Apr 2025 12:07:52 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8B=A5=E5=B9=B2=E4=BF=AE=E6=94=B9=EF=BC=8C?=
 =?UTF-8?q?=E6=96=B9=E4=BE=BF=E5=85=B6=E4=BB=96=E7=A8=8B=E5=BA=8F=E8=B0=83?=
 =?UTF-8?q?=E7=94=A8=20https://github.com/nxtrace/NTrace-V1/issues/84=20?=
 =?UTF-8?q?=20=E8=A6=81=E6=8F=90=E4=BA=A4=E7=9A=84=E5=8F=98=E6=9B=B4?=
 =?UTF-8?q?=EF=BC=9A=20=09=E4=BF=AE=E6=94=B9=EF=BC=9A=20=20=20=20=20cmd/cm?=
 =?UTF-8?q?d.go=20=09=E4=BF=AE=E6=94=B9=EF=BC=9A=20=20=20=20=20go.mod=20?=
 =?UTF-8?q?=09=E4=BF=AE=E6=94=B9=EF=BC=9A=20=20=20=20=20go.sum=20=09?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9=EF=BC=9A=20=20=20=20=20wshandle/client.go?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/cmd.go         |  5 +++--
 wshandle/client.go | 16 ++++++++++------
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/cmd/cmd.go b/cmd/cmd.go
index 3faf70e..ebcf963 100644
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -226,8 +226,9 @@ func Excute() {
 		}
 	}
 	if err != nil {
-		fmt.Println(err)
-		os.Exit(1)
+		//fmt.Println(err)
+		//os.Exit(1)
+		panic(err)
 	}
 	//}()
 	//
diff --git a/wshandle/client.go b/wshandle/client.go
index 1813254..dcef153 100644
--- a/wshandle/client.go
+++ b/wshandle/client.go
@@ -104,7 +104,8 @@ func (c *WsConn) messageSendHandler() {
 			err := c.Conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
 			if err != nil {
 				// log.Println("write close:", err)
-				os.Exit(1)
+				//os.Exit(1)
+				panic(err)
 			}
 			select {
 			// 等到了结果，直接退出
@@ -112,7 +113,8 @@ func (c *WsConn) messageSendHandler() {
 			// 如果等待 1s 还是拿不到结果，不再等待，超时退出
 			case <-time.After(time.Second):
 			}
-			os.Exit(1)
+			//os.Exit(1)
+			panic("请求超时退出")
 			// return
 		}
 	}
@@ -134,8 +136,9 @@ func (c *WsConn) recreateWsConn() {
 				jwtToken, err = pow.GetToken(util.GetPowProvider(), util.GetPowProvider(), port)
 			}
 			if err != nil {
-				log.Println(err)
-				os.Exit(1)
+				//log.Println(err)
+				//os.Exit(1)
+				panic(err)
 			}
 		} else {
 			// 使用 cacheToken
@@ -206,8 +209,9 @@ func createWsConn() *WsConn {
 			jwtToken, err = pow.GetToken(util.GetPowProvider(), util.GetPowProvider(), port)
 		}
 		if err != nil {
-			log.Println(err)
-			os.Exit(1)
+			//log.Println(err)
+			//os.Exit(1)
+			panic(err)
 		}
 		ua = []string{util.UserAgent}
 	}