package main

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

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

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

	// 準(zhǔn)備請(qǐng)求參數(shù)
	url := "https://192.168.24.70:2018/api/zguard/sysmng/syscfg/basecfg/sysname/651d5b9f-225b-4c8c-9f06-80bfad3fa977"
	cookie := "session-id=f416b188c91bc72a06853b362d5cb7b3a6b68a43"

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

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

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

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

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

		fmt.Printf("發(fā)送第 %d 個(gè)請(qǐng)求...\n", i+1)
		// 發(fā)送請(qǐng)求
		resp, err := client.Do(req)
		if err != nil {
			fmt.Printf("請(qǐng)求失敗: %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)建請(qǐng)求失敗: %v\n", err)
					continue
				}

				// 設(shè)置 Content-Length
				reqretry.ContentLength = int64(len(jsonBody))
				// 設(shè)置請(qǐng)求頭
				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
			}
		}

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

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

		fmt.Println("等待10秒后發(fā)送第二個(gè)請(qǐng)求...")
		time.Sleep(10 * time.Second) // 等待10秒，此時(shí)服務(wù)端已經(jīng)關(guān)閉連接（10s）
		time.Sleep(500 * time.Millisecond)
	}
}