64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package client
|
||
|
||
import (
|
||
"net/http"
|
||
"time"
|
||
)
|
||
|
||
// HTTPClientConfig HTTP客户端配置
|
||
type HTTPClientConfig struct {
|
||
Timeout time.Duration
|
||
MaxIdleConns int
|
||
MaxIdleConnsPerHost int
|
||
MaxConnsPerHost int
|
||
}
|
||
|
||
// DefaultHTTPClientConfig 默认HTTP客户端配置
|
||
var DefaultHTTPClientConfig = HTTPClientConfig{
|
||
Timeout: 30 * time.Second,
|
||
MaxIdleConns: 100,
|
||
MaxIdleConnsPerHost: 50,
|
||
MaxConnsPerHost: 0, // 0表示不限制
|
||
}
|
||
|
||
// NewHTTPClient 创建优化的HTTP客户端
|
||
func NewHTTPClient(config HTTPClientConfig) *http.Client {
|
||
transport := &http.Transport{
|
||
// 连接池配置
|
||
MaxIdleConns: config.MaxIdleConns,
|
||
MaxIdleConnsPerHost: config.MaxIdleConnsPerHost,
|
||
MaxConnsPerHost: config.MaxConnsPerHost,
|
||
IdleConnTimeout: 90 * time.Second,
|
||
|
||
// 性能优化
|
||
DisableCompression: false,
|
||
DisableKeepAlives: false,
|
||
ForceAttemptHTTP2: true, // 启用HTTP/2
|
||
|
||
// 超时配置
|
||
TLSHandshakeTimeout: 10 * time.Second,
|
||
ResponseHeaderTimeout: 30 * time.Second,
|
||
ExpectContinueTimeout: 1 * time.Second,
|
||
|
||
// 缓冲大小
|
||
WriteBufferSize: 32 * 1024, // 32KB写缓冲
|
||
ReadBufferSize: 32 * 1024, // 32KB读缓冲
|
||
}
|
||
|
||
return &http.Client{
|
||
Timeout: config.Timeout,
|
||
Transport: transport,
|
||
}
|
||
}
|
||
|
||
// NewLLMHTTPClient 创建用于LLM的HTTP客户端（连接池更大）
|
||
func NewLLMHTTPClient(timeout time.Duration) *http.Client {
|
||
config := HTTPClientConfig{
|
||
Timeout: timeout,
|
||
MaxIdleConns: 200,
|
||
MaxIdleConnsPerHost: 100,
|
||
MaxConnsPerHost: 0,
|
||
}
|
||
return NewHTTPClient(config)
|
||
}
|