package main

import (
	"bytes"
	"crypto/tls"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"time"
)

func main() {
	// 創(chuàng)建自定義的 Transport,設置連接池參數
	tr := &http.Transport{
		TLSClientConfig: &tls.Config{
			InsecureSkipVerify: true, // 忽略 TLS 證書驗證
		},
		MaxIdleConns:        1,                // 限制最大空閑連接數為1
		MaxIdleConnsPerHost: 1,                // 限制每個host最大空閑連接數為1
		IdleConnTimeout:     20 * time.Second, // 本地空閑連接超時設置為20s
		DisableKeepAlives:   false,            // 啟用 keep-alive
		MaxConnsPerHost:     1,                // 限制每個host的最大連接數為1，強制復用連接
		ForceAttemptHTTP2:   false,            // 禁用 HTTP/2
	}

	// 創(chuàng)建 HTTP 客戶端
	client := &http.Client{
		Transport: tr,
		Timeout:   5 * time.Second, // 設置請求超時時間
	}

	// 準備請求參數
	url := "https://192.168.24.70:2018/api/zguard/sysmng/syscfg/basecfg/sysname/651d5b9f-225b-4c8c-9f06-80bfad3fa977"
	cookie := "session-id=f416b188c91bc72a06853b362d5cb7b3a6b68a43"

	// 準備請求體數據
	requestBody := map[string]string{
		"sys_name": "N-GUARD",
	}
	// 將 map 轉換為 JSON
	jsonBody, err := json.Marshal(requestBody)
	if err != nil {
		fmt.Printf("JSON 編碼失敗: %v\n", err)
	}

	// 循環(huán)發(fā)送請求,模擬使用已關閉的連接
	for i := 0; i < 5; i++ { // 只測試兩次請求即可

		// 每次請求都創(chuàng)建新的 bytes.Buffer，確保 Body 可以重復讀取
		bodyReader := bytes.NewBuffer(jsonBody)

		req, err := http.NewRequest("PUT", url, bodyReader)
		if err != nil {
			fmt.Printf("創(chuàng)建請求失敗: %v\n", err)
			continue
		}

		// 設置 Content-Length
		req.ContentLength = int64(len(jsonBody))
		// 設置請求頭
		req.Header.Set("Cookie", cookie)
		req.Header.Set("Content-Type", "application/json")

		fmt.Printf("發(fā)送第 %d 個請求...\n", i+1)
		// 發(fā)送請求
		resp, err := client.Do(req)
		if err != nil {
			fmt.Printf("請求失敗: %v\n", err)
			if errors.Is(err, io.EOF) {
				fmt.Printf("連接不再可用: 重試：新的五元重新發(fā)起連接\n")
				bodyReader := bytes.NewBuffer(jsonBody)

				reqretry, err := http.NewRequest("PUT", url, bodyReader)
				if err != nil {
					fmt.Printf("創(chuàng)建請求失敗: %v\n", err)
					continue
				}

				// 設置 Content-Length
				reqretry.ContentLength = int64(len(jsonBody))
				// 設置請求頭
				reqretry.Header.Set("Cookie", cookie)
				reqretry.Header.Set("Content-Type", "application/json")

				resp, err = client.Do(reqretry)
				if err != nil {
					fmt.Printf("err:\n", err)
					continue
				}
			} else {
				continue
			}
		}

		// 讀取響應
		body, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			fmt.Printf("讀取響應失敗: %v\n", err)
		}
		resp.Body.Close()

		fmt.Printf("請求 %d - 狀態(tài)碼: %d, 響應: %s\n", i+1, resp.StatusCode, string(body))

		fmt.Println("等待10秒后發(fā)送第二個請求...")
		time.Sleep(10 * time.Second) // 等待10秒，此時服務端已經關閉連接（10s）
		time.Sleep(500 * time.Millisecond)
	}
}