fdcbf7aacf
提取 proxyurl.Parse() 公共包，将分散在 6 处的代理 URL 验证逻辑 统一收敛，确保无效代理配置在创建时立即失败，永不静默回退直连。 主要变更： - 新增 proxyurl 包：统一 TrimSpace → url.Parse → Host 校验 → Scheme 白名单 - socks5:// 自动升级为 socks5h://，防止 DNS 泄漏（大小写不敏感） - antigravity: http.ProxyURL → proxyutil.ConfigureTransportProxy 支持 SOCKS5 - openai_oauth: 删除 newOpenAIOAuthHTTPClient，收编至 httpclient.GetClient - 移除未使用的 ProxyStrict 字段（fail-fast 已是全局默认行为） - 补充 15 个 proxyurl 测试 + pricing/usage fail-fast 测试
76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package repository
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/pkg/proxyurl"
|
||
|
||
"github.com/imroc/req/v3"
|
||
)
|
||
|
||
// reqClientOptions 定义 req 客户端的构建参数
|
||
type reqClientOptions struct {
|
||
ProxyURL string // 代理 URL（支持 http/https/socks5）
|
||
Timeout time.Duration // 请求超时时间
|
||
Impersonate bool // 是否模拟 Chrome 浏览器指纹
|
||
ForceHTTP2 bool // 是否强制使用 HTTP/2
|
||
}
|
||
|
||
// sharedReqClients 存储按配置参数缓存的 req 客户端实例
|
||
//
|
||
// 性能优化说明：
|
||
// 原实现在每次 OAuth 刷新时都创建新的 req.Client：
|
||
// 1. claude_oauth_service.go: 每次刷新创建新客户端
|
||
// 2. openai_oauth_service.go: 每次刷新创建新客户端
|
||
// 3. gemini_oauth_client.go: 每次刷新创建新客户端
|
||
//
|
||
// 新实现使用 sync.Map 缓存客户端：
|
||
// 1. 相同配置（代理+超时+模拟设置）复用同一客户端
|
||
// 2. 复用底层连接池，减少 TLS 握手开销
|
||
// 3. LoadOrStore 保证并发安全，避免重复创建
|
||
var sharedReqClients sync.Map
|
||
|
||
// getSharedReqClient 获取共享的 req 客户端实例
|
||
// 性能优化：相同配置复用同一客户端，避免重复创建
|
||
func getSharedReqClient(opts reqClientOptions) (*req.Client, error) {
|
||
key := buildReqClientKey(opts)
|
||
if cached, ok := sharedReqClients.Load(key); ok {
|
||
if c, ok := cached.(*req.Client); ok {
|
||
return c, nil
|
||
}
|
||
}
|
||
|
||
client := req.C().SetTimeout(opts.Timeout)
|
||
if opts.ForceHTTP2 {
|
||
client = client.EnableForceHTTP2()
|
||
}
|
||
if opts.Impersonate {
|
||
client = client.ImpersonateChrome()
|
||
}
|
||
trimmed, _, err := proxyurl.Parse(opts.ProxyURL)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if trimmed != "" {
|
||
client.SetProxyURL(trimmed)
|
||
}
|
||
|
||
actual, _ := sharedReqClients.LoadOrStore(key, client)
|
||
if c, ok := actual.(*req.Client); ok {
|
||
return c, nil
|
||
}
|
||
return client, nil
|
||
}
|
||
|
||
func buildReqClientKey(opts reqClientOptions) string {
|
||
return fmt.Sprintf("%s|%s|%t|%t",
|
||
strings.TrimSpace(opts.ProxyURL),
|
||
opts.Timeout.String(),
|
||
opts.Impersonate,
|
||
opts.ForceHTTP2,
|
||
)
|
||
}
|