|
|
//go:build client
|
|
|
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"log"
|
|
|
"sync"
|
|
|
"time"
|
|
|
|
|
|
"github.com/noahlann/nnet/pkg/nnet"
|
|
|
)
|
|
|
|
|
|
func main() {
|
|
|
// 创建连接池配置
|
|
|
poolConfig := &nnet.ClientPoolConfig{
|
|
|
MaxSize: 10, // 最大连接数
|
|
|
MinSize: 2, // 最小连接数（预创建）
|
|
|
ClientConfig: &nnet.ClientConfig{
|
|
|
Addr: "tcp://127.0.0.1:8888",
|
|
|
ConnectTimeout: 5 * time.Second,
|
|
|
ReadTimeout: 5 * time.Second,
|
|
|
WriteTimeout: 5 * time.Second,
|
|
|
},
|
|
|
AcquireTimeout: 5 * time.Second, // 获取连接超时时间
|
|
|
IdleTimeout: 5 * time.Minute, // 空闲连接超时时间
|
|
|
}
|
|
|
|
|
|
// 创建连接池
|
|
|
pool, err := nnet.NewClientPool(poolConfig)
|
|
|
if err != nil {
|
|
|
log.Fatalf("Failed to create connection pool: %v", err)
|
|
|
}
|
|
|
defer pool.Close()
|
|
|
|
|
|
log.Printf("Connection pool created: Size=%d, Available=%d", pool.Size(), pool.Available())
|
|
|
|
|
|
// 等待连接建立
|
|
|
time.Sleep(500 * time.Millisecond)
|
|
|
|
|
|
log.Printf("Connection pool ready: Size=%d, Available=%d", pool.Size(), pool.Available())
|
|
|
|
|
|
// 示例1：单个请求
|
|
|
fmt.Println("\n=== 示例1：单个请求 ===")
|
|
|
client, err := pool.Acquire()
|
|
|
if err != nil {
|
|
|
log.Fatalf("Failed to acquire connection: %v", err)
|
|
|
}
|
|
|
|
|
|
// 发送请求
|
|
|
request := []byte("echo hello\n")
|
|
|
resp, err := client.Request(request, 5*time.Second)
|
|
|
if err != nil {
|
|
|
log.Printf("Request failed: %v", err)
|
|
|
} else {
|
|
|
fmt.Printf("Response: %s", string(resp))
|
|
|
}
|
|
|
|
|
|
// 释放连接
|
|
|
pool.Release(client)
|
|
|
|
|
|
// 示例2：并发请求（使用连接池）
|
|
|
fmt.Println("\n=== 示例2：并发请求 ===")
|
|
|
const numRequests = 20
|
|
|
var wg sync.WaitGroup
|
|
|
wg.Add(numRequests)
|
|
|
|
|
|
for i := 0; i < numRequests; i++ {
|
|
|
go func(id int) {
|
|
|
defer wg.Done()
|
|
|
|
|
|
// 从连接池获取连接
|
|
|
client, err := pool.Acquire()
|
|
|
if err != nil {
|
|
|
log.Printf("Goroutine %d: Failed to acquire connection: %v", id, err)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 使用完毕后释放连接
|
|
|
defer pool.Release(client)
|
|
|
|
|
|
// 发送请求
|
|
|
request := []byte(fmt.Sprintf("echo request-%d\n", id))
|
|
|
resp, err := client.Request(request, 5*time.Second)
|
|
|
if err != nil {
|
|
|
log.Printf("Goroutine %d: Request failed: %v", id, err)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
fmt.Printf("Goroutine %d: Response: %s", id, string(resp))
|
|
|
}(i)
|
|
|
}
|
|
|
|
|
|
wg.Wait()
|
|
|
|
|
|
// 示例3：获取时间
|
|
|
fmt.Println("\n=== 示例3：获取时间 ===")
|
|
|
client, err = pool.Acquire()
|
|
|
if err != nil {
|
|
|
log.Fatalf("Failed to acquire connection: %v", err)
|
|
|
}
|
|
|
defer pool.Release(client)
|
|
|
|
|
|
request = []byte("time\n")
|
|
|
resp, err = client.Request(request, 5*time.Second)
|
|
|
if err != nil {
|
|
|
log.Printf("Request failed: %v", err)
|
|
|
} else {
|
|
|
fmt.Printf("Response: %s", string(resp))
|
|
|
}
|
|
|
|
|
|
// 显示连接池状态
|
|
|
fmt.Printf("\n=== 连接池状态 ===\n")
|
|
|
fmt.Printf("Size: %d\n", pool.Size())
|
|
|
fmt.Printf("Available: %d\n", pool.Available())
|
|
|
fmt.Printf("IsClosed: %v\n", pool.IsClosed())
|
|
|
}
|