若干修改,方便其他程序调用

https://github.com/nxtrace/NTrace-V1/issues/84
 要提交的变更:
	修改:     cmd/cmd.go
	修改:     go.mod
	修改:     go.sum
	修改:     wshandle/client.go
This commit is contained in:
tsosunchia
2025-04-13 12:07:52 +08:00
parent 62ab23bdeb
commit 608a2904d4
2 changed files with 13 additions and 8 deletions

View File

@@ -226,8 +226,9 @@ func Excute() {
}
}
if err != nil {
fmt.Println(err)
os.Exit(1)
//fmt.Println(err)
//os.Exit(1)
panic(err)
}
//}()
//

View File

@@ -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}
}