refactor: refactor relay part (#935)

This commit is contained in:
JustSong
2024-01-14 19:21:03 +08:00
committed by GitHub
parent f2c51a494c
commit eed9f5fdf0
37 changed files with 1527 additions and 1437 deletions

24
relay/util/init.go Normal file
View File

@@ -0,0 +1,24 @@
package util
import (
"net/http"
"one-api/common"
"time"
)
var HTTPClient *http.Client
var ImpatientHTTPClient *http.Client
func init() {
if common.RelayTimeout == 0 {
HTTPClient = &http.Client{}
} else {
HTTPClient = &http.Client{
Timeout: time.Duration(common.RelayTimeout) * time.Second,
}
}
ImpatientHTTPClient = &http.Client{
Timeout: 5 * time.Second,
}
}