127 lines
3.2 KiB
Go
127 lines
3.2 KiB
Go
package main
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"fmt"
|
||
"net/http"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
// 请求体结构
|
||
type UserRequest struct {
|
||
UserID int `json:"user_id"`
|
||
}
|
||
|
||
// 响应体结构（根据实际接口响应调整）
|
||
type Response struct {
|
||
Success bool `json:"success"`
|
||
Message string `json:"message"`
|
||
}
|
||
|
||
func main() {
|
||
const totalUsers = 1000 // 总用户数
|
||
const batchSize = 130 // 每批发送的请求数
|
||
const interval = 5 * time.Second // 每批请求的时间间隔
|
||
const duration = 20 * time.Second // 总持续时间
|
||
|
||
// 计算总批次数（确保在duration时间内发送完所有请求）
|
||
totalBatches := totalUsers / batchSize
|
||
if totalUsers%batchSize != 0 {
|
||
totalBatches++
|
||
}
|
||
|
||
// 检查总耗时是否符合预期
|
||
expectedDuration := time.Duration(totalBatches-1) * interval
|
||
if expectedDuration > duration {
|
||
fmt.Printf("警告：按当前设置将耗时 %v，超过预期的 %v\n", expectedDuration, duration)
|
||
}
|
||
|
||
var wg sync.WaitGroup
|
||
userID := 1 // 用户ID计数器
|
||
|
||
// 按批次发送请求
|
||
for batch := 1; batch <= totalBatches; batch++ {
|
||
// 计算当前批次的请求数量（最后一批可能不足batchSize）
|
||
currentBatchSize := batchSize
|
||
if userID+currentBatchSize-1 > totalUsers {
|
||
currentBatchSize = totalUsers - userID + 1
|
||
}
|
||
|
||
fmt.Printf("开始第 %d 批请求，共 %d 个，当前时间: %v\n",
|
||
batch, currentBatchSize, time.Now().Format("15:04:05"))
|
||
|
||
// 发送当前批次的请求
|
||
for i := 0; i < currentBatchSize; i++ {
|
||
wg.Add(1)
|
||
go func(uid int) {
|
||
defer wg.Done()
|
||
// 发送POST请求
|
||
result, err := sendPostRequest(uid)
|
||
if err != nil {
|
||
fmt.Printf("用户 %d 请求失败: %v\n", uid, err)
|
||
return
|
||
}
|
||
// 输出结果（可以根据需要调整输出频率，避免日志过多）
|
||
fmt.Printf("用户 %d 响应: %+v\n", uid, result)
|
||
}(userID)
|
||
userID++
|
||
}
|
||
|
||
// 最后一批不需要等待
|
||
if batch < totalBatches {
|
||
// 等待指定间隔后再发送下一批
|
||
time.Sleep(interval)
|
||
}
|
||
}
|
||
|
||
// 等待所有请求完成
|
||
wg.Wait()
|
||
fmt.Println("所有请求已完成")
|
||
}
|
||
|
||
// 发送POST请求
|
||
func sendPostRequest(userID int) (*Response, error) {
|
||
url := "http://localhost:9096/user/kill"
|
||
|
||
// 创建请求体
|
||
requestBody := UserRequest{
|
||
UserID: userID,
|
||
}
|
||
|
||
// 转换为JSON
|
||
jsonBody, err := json.Marshal(requestBody)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("JSON序列化失败: %v", err)
|
||
}
|
||
|
||
// 发送POST请求
|
||
resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonBody))
|
||
if err != nil {
|
||
return nil, fmt.Errorf("请求发送失败: %v", err)
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 解析响应（无论状态码如何都尝试解析，以便获取详细信息）
|
||
var response Response
|
||
decoder := json.NewDecoder(resp.Body)
|
||
if err := decoder.Decode(&response); err != nil {
|
||
return nil, fmt.Errorf("响应解析失败: %v", err)
|
||
}
|
||
|
||
// 检查响应状态码
|
||
// 状态码异常时，返回包含完整响应的错误信息
|
||
if resp.StatusCode != http.StatusOK {
|
||
// 将完整响应转为JSON字符串，便于查看
|
||
responseJSON, _ := json.MarshalIndent(response, "", " ")
|
||
return &response, fmt.Errorf(
|
||
"状态码异常: %d, 响应内容: %s",
|
||
resp.StatusCode,
|
||
responseJSON,
|
||
)
|
||
}
|
||
|
||
return &response, nil
|
||
}
|