优化HTTP客户端管理

This commit is contained in:
刘祥超
2024-04-17 18:24:21 +08:00
parent 3ce30e36bb
commit 2b01162ab4
4 changed files with 35 additions and 24 deletions

View File

@@ -23,14 +23,14 @@ func TestHTTPClientPool_Client(t *testing.T) {
t.Fatal(err)
}
{
client, err := pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false)
client, err := pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false, "example.com")
if err != nil {
t.Fatal(err)
}
t.Log("client:", client)
}
for i := 0; i < 10; i++ {
client, err := pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false)
client, err := pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false, "example.com")
if err != nil {
t.Fatal(err)
}
@@ -54,7 +54,7 @@ func TestHTTPClientPool_cleanClients(t *testing.T) {
for i := 0; i < 10; i++ {
t.Log("get", i)
_, _ = pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false)
_, _ = pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false, "example.com")
if testutils.IsSingleTesting() {
time.Sleep(1 * time.Second)
@@ -79,6 +79,6 @@ func BenchmarkHTTPClientPool_Client(b *testing.B) {
var pool = NewHTTPClientPool()
for i := 0; i < b.N; i++ {
_, _ = pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false)
_, _ = pool.Client(nil, origin, origin.Addr.PickAddress(), nil, false, "example.com")
}
}