package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"net/http"
	"sync"
)

// 请求体结构
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 concurrencyLimit = 100 // 并发控制，避免过多同时连接

	// 创建信号量控制并发数量
	semaphore := make(chan struct{}, concurrencyLimit)
	var wg sync.WaitGroup

	for i := 1; i <= totalUsers; i++ {
		wg.Add(1)
		semaphore <- struct{}{} // 获取信号量

		go func(userID int) {
			defer wg.Done()
			defer func() { <-semaphore }() // 释放信号量

			// 发送POST请求
			result, err := sendPostRequest(userID)
			if err != nil {
				fmt.Printf("用户 %d 请求失败: %v\n", userID, err)
				return
			}

			// 输出结果
			fmt.Printf("用户 %d 响应: %+v\n", userID, result)
		}(i)
	}

	// 等待所有请求完成
	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()

	// 检查响应状态码
	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("响应状态码异常: %d", resp.StatusCode)
	}

	// 解析响应
	var response Response
	decoder := json.NewDecoder(resp.Body)
	if err := decoder.Decode(&response); err != nil {
		return nil, fmt.Errorf("响应解析失败: %v", err)
	}

	return &response, nil
}