150 lines
3.8 KiB
Go
150 lines
3.8 KiB
Go
package example
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/yuninks/curlx"
|
|
)
|
|
|
|
func TestConnectionReuse(t *testing.T) {
|
|
// 创建带优化连接池配置的客户端
|
|
client := curlx.NewCurlx(
|
|
curlx.WithMaxIdleConns(50),
|
|
curlx.WithMaxIdleConnsPerHost(10),
|
|
curlx.WithMaxConnsPerHost(20),
|
|
curlx.WithIdleConnTimeout(60*time.Second),
|
|
curlx.WithOptionTimeOut(30*time.Second),
|
|
)
|
|
|
|
// 测试同一个主机的多次请求，观察连接复用效果
|
|
targetURL := "https://httpbin.org/get"
|
|
|
|
fmt.Println("=== 连接复用测试开始 ===")
|
|
|
|
// 预热连接
|
|
fmt.Println("1. 预热连接...")
|
|
ctx1, cancel1 := context.WithTimeout(context.Background(), 10*time.Second)
|
|
_, err := client.Get(ctx1, targetURL)
|
|
cancel1()
|
|
if err != nil {
|
|
t.Logf("预热请求失败: %v", err)
|
|
return
|
|
}
|
|
fmt.Println(" 预热完成")
|
|
|
|
// 连续请求测试
|
|
fmt.Println("\n2. 连续请求测试...")
|
|
for i := 1; i <= 5; i++ {
|
|
start := time.Now()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
|
|
response, err := client.Get(ctx, targetURL)
|
|
duration := time.Since(start)
|
|
|
|
cancel()
|
|
|
|
if err != nil {
|
|
t.Logf("第%d次请求失败: %v (耗时: %v)", i, err, duration)
|
|
} else {
|
|
t.Logf("第%d次请求成功: %d字节 (耗时: %v)", i, len(response), duration)
|
|
}
|
|
|
|
time.Sleep(200 * time.Millisecond) // 短暂间隔
|
|
}
|
|
|
|
fmt.Println("\n=== 测试完成 ===")
|
|
}
|
|
|
|
func TestConcurrentConnectionReuse(t *testing.T) {
|
|
// 创建连接池管理器
|
|
poolManager := NewConnectionPoolManager(
|
|
curlx.WithMaxIdleConns(100),
|
|
curlx.WithMaxIdleConnsPerHost(20),
|
|
curlx.WithMaxConnsPerHost(30),
|
|
curlx.WithIdleConnTimeout(120*time.Second),
|
|
)
|
|
|
|
// 测试并发请求
|
|
urls := []string{
|
|
"https://httpbin.org/get",
|
|
"https://httpbin.org/uuid",
|
|
"https://httpbin.org/user-agent",
|
|
"https://httpbin.org/headers",
|
|
"https://httpbin.org/ip",
|
|
}
|
|
|
|
fmt.Println("=== 并发连接复用测试 ===")
|
|
poolManager.ConcurrentRequests(urls)
|
|
}
|
|
|
|
func TestPersistentConnection(t *testing.T) {
|
|
// 创建优化的客户端
|
|
client := curlx.NewCurlx(
|
|
curlx.WithMaxIdleConns(30),
|
|
curlx.WithMaxIdleConnsPerHost(5),
|
|
curlx.WithMaxConnsPerHost(15),
|
|
curlx.WithIdleConnTimeout(30*time.Second),
|
|
)
|
|
|
|
manager := &ConnectionPoolManager{
|
|
client: client,
|
|
// transport: client.transport,
|
|
}
|
|
|
|
fmt.Println("=== 持久连接测试 ===")
|
|
manager.PersistentConnectionExample("https://httpbin.org/delay/1")
|
|
}
|
|
|
|
func Example_connectionReuse() {
|
|
// 最佳实践示例：如何正确配置连接复用
|
|
|
|
// 1. 创建优化配置的客户端
|
|
client := curlx.NewCurlx(
|
|
// 连接池配置
|
|
curlx.WithMaxIdleConns(100), // 总空闲连接数
|
|
curlx.WithMaxIdleConnsPerHost(10), // 每主机空闲连接数
|
|
curlx.WithMaxConnsPerHost(50), // 每主机最大连接数
|
|
curlx.WithIdleConnTimeout(90*time.Second), // 空闲超时时间
|
|
|
|
// 其他优化配置
|
|
curlx.WithOptionTimeOut(30*time.Second),
|
|
)
|
|
|
|
// 2. 复用同一个客户端实例进行多次请求
|
|
ctx := context.Background()
|
|
|
|
// 第一次请求会建立新连接
|
|
response1, err := client.Get(ctx, "https://httpbin.org/get")
|
|
if err != nil {
|
|
fmt.Printf("首次请求失败: %v\n", err)
|
|
return
|
|
}
|
|
fmt.Printf("首次请求成功: %d字节\n", len(response1))
|
|
|
|
// 后续请求会复用已有连接
|
|
response2, err := client.Get(ctx, "https://httpbin.org/uuid")
|
|
if err != nil {
|
|
fmt.Printf("第二次请求失败: %v\n", err)
|
|
return
|
|
}
|
|
fmt.Printf("第二次请求成功: %d字节\n", len(response2))
|
|
|
|
// 3. 查看连接池状态
|
|
manager := &ConnectionPoolManager{
|
|
client: client,
|
|
// transport: client.transport,
|
|
}
|
|
manager.PrintPoolStats()
|
|
|
|
// Output:
|
|
// 首次请求成功: [字节数]
|
|
// 第二次请求成功: [字节数]
|
|
// === 连接池统计 ===
|
|
// 当前空闲连接数: 1
|
|
// 总连接数: 1
|
|
// 等待队列长度: 0
|
|
}
|