fix: set timeout for WDA request

This commit is contained in:
lilong.129
2024-12-04 17:37:24 +08:00
parent 590e6110a1
commit a59ac47eca
2 changed files with 6 additions and 3 deletions
+1 -1
View File
@@ -1 +1 @@
v5.0.0+2412021645 v5.0.0+2412041737
+5 -2
View File
@@ -108,8 +108,11 @@ func (wd *Driver) httpDELETE(pathElem ...string) (rawResp rawResponse, err error
func (wd *Driver) httpRequest(method string, rawURL string, rawBody []byte) (rawResp rawResponse, err error) { func (wd *Driver) httpRequest(method string, rawURL string, rawBody []byte) (rawResp rawResponse, err error) {
log.Debug().Str("method", method).Str("url", rawURL).Str("body", string(rawBody)).Msg("request driver agent") log.Debug().Str("method", method).Str("url", rawURL).Str("body", string(rawBody)).Msg("request driver agent")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
var req *http.Request var req *http.Request
if req, err = http.NewRequest(method, rawURL, bytes.NewBuffer(rawBody)); err != nil { if req, err = http.NewRequestWithContext(ctx, method, rawURL, bytes.NewBuffer(rawBody)); err != nil {
return nil, err return nil, err
} }
req.Header.Set("Content-Type", "application/json;charset=UTF-8") req.Header.Set("Content-Type", "application/json;charset=UTF-8")
@@ -162,6 +165,6 @@ func convertToHTTPClient(conn net.Conn) *http.Client {
return conn, nil return conn, nil
}, },
}, },
Timeout: 0, Timeout: 30 * time.Second,
} }
} }