GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-09 17:59:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: set tcp keep alive

Browse Source
This commit is contained in:
debugtalk
2022-10-25 15:13:02 +08:00
parent 2959b94caf
commit 5d4f6610e3
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
hrp/pkg/uixt/client.go
View File

@@ -93,10 +93,18 @@ func (wd *Driver) httpRequest(method string, rawURL string, rawBody []byte) (raw
}
func convertToHTTPClient(conn net.Conn) *http.Client {
// set tcp keep alive
tcpConn := conn.(*net.TCPConn)
if err := tcpConn.SetKeepAlive(true); err != nil {
log.Error().Err(err).Msg("set tcp keep alive failed")
}
if err := tcpConn.SetKeepAlivePeriod(5 * time.Second); err != nil {
log.Error().Err(err).Msg("set tcp keep alive period failed")
}
return &http.Client{
Transport: &http.Transport{
DialContext: func(_ context.Context, _, _ string) (net.Conn, error) {
return conn, nil
return tcpConn, nil
},
},
Timeout: 0,