Merge pull request #20 from xqdoo00o/master

add timeout
bi1101 · May 3, 2024 · e2a8152 · e2a8152
2 parents a5bfcf4 + 2b6ea6e
commit e2a8152
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/chatgpt/request.go b/internal/chatgpt/request.go
@@ -112,6 +112,7 @@ type rawDialer interface {
 
 func createWSConn(addr string, connInfo *connInfo, proxyURLString string, retry int) error {
 	dialer := websocket.Dialer{
+		HandshakeTimeout: 8 * time.Second,
 		NetDialTLSContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
 			host, _, _ := net.SplitHostPort(addr)
 			config := &tls.Config{ServerName: host, OmitEmptyPsk: true}
@@ -203,7 +204,7 @@ func InitWSConn(secret *tokens.Secret, deviceId string, uuid string, proxy strin
 		if err != nil {
 			return err
 		}
-		createWSConn(wssURL, connInfo, proxy, 0)
+		err = createWSConn(wssURL, connInfo, proxy, 0)
 		if err != nil {
 			return err
 		}