package main

import (
	"fmt"

	"github.com/go-resty/resty/v2"
)

func main() {
	client := resty.New()

	resp, err := client.R().
		Get("https://www.baidu.com")

	if err != nil {
		fmt.Println("請求出錯:", err)
		return
	}

	fmt.Println("狀態(tài)碼:", resp.StatusCode())
	fmt.Println("響應體:", resp.String())
	fmt.Println("響應頭:", resp.Header())                       // 獲取全部header
	fmt.Println("特定響應頭:", resp.Header().Get("Content-Type")) // 獲取特定的header
	// 狀態(tài)碼: 200
	// 響應體: xxx太多了省略掉...
	// 響應頭: map[Accept-Ranges:[bytes] Cache-Control:[no-cache] Connection:[keep-alive] Content-Length:[227] Content-Security-Policy:[frame-ancestors 'self' https://chat.baidu.com http://mirror-chat.baidu.com https://fj-chat.baidu.com https://hba-chat.baidu.com https://hbe-chat.baidu.com https://njjs-chat.baidu.com https://nj-chat.baidu.com https://hna-chat.baidu.com https://hnb-chat.baidu.com http://debug.baidu-int.com;] Content-Type:[text/html] Date:[Sun, 16 Mar 2025 09:43:18 GMT] P3p:[CP=" OTI DSP COR IVA OUR IND COM " CP=" OTI DSP COR IVA OUR IND COM "] Pragma:[no-cache] Server:[BWS/1.1] Set-Cookie:[BD_NOT_HTTPS=1; path=/; Max-Age=300 BIDUPSID=10846246655E82CCF356A792677D7EA8; expires=Thu, 31-Dec-37 23:55:55 GMT; max-age=2147483647; path=/; domain=.baidu.com PSTM=1742118198; expires=Thu, 31-Dec-37 23:55:55 GMT; max-age=2147483647; path=/; domain=.baidu.com BAIDUID=10846246655E82CC89D4B0052594BBBE:FG=1; max-age=31536000; expires=Mon, 16-Mar-26 09:43:18 GMT; domain=.baidu.com; path=/; version=1; comment=bd] Traceid:[1742118198045022490612843349543995319034] X-Ua-Compatible:[IE=Edge,chrome=1] X-Xss-Protection:[1;mode=block]]
	// 特定響應頭: text/html
}

resp, err := client.R().
	SetQueryParam("postId", "1"). // 設置單個查詢參數(shù) 也是可以的
	Get("https://jsonplaceholder.typicode.com/posts")

// resp, err := client.R().
// 	SetQueryParams(map[string]string{ // 設置多個查詢參數(shù)
// 		"postId": "1",
// 	}).
// 	Get("https://jsonplaceholder.typicode.com/posts")

	resp, err := client.R().
		SetHeader("Content-Type", "application/json"). // 設置單個請求頭
		SetBody(`{"title": "foo", "body": "bar", "userId": 1}`). // 字符串形式
		Post("https://jsonplaceholder.typicode.com/posts")

	resp, err := client.R().
		SetBody(map[string]interface{}{ // 支持 map結構
			"title":  "foo",
			"body":   "bar",
			"userId": 1,
		}).
		SetHeaders(map[string]string{  // 設置多個請求頭
			"Content-Type": "application/json",
		}).
		Post("https://jsonplaceholder.typicode.com/posts")

	resp, err := client.R().
		SetBody(Post{Title: "foo", Body: "bar", UserId: 1}). // 支持struct
		SetHeaders(map[string]string{
			"Content-Type": "application/json",
		}).
		Post("https://jsonplaceholder.typicode.com/posts")

	// 從文件創(chuàng)建 io.Reader
	file, err := os.Open("my_file.txt")
	if err != nil {
			// ... 錯誤處理 ...
	}
	defer file.Close()
	resp, err := client.R().
			// 不設置也可以， resty會根據(jù)reader自動推斷content-type
			SetHeader("Content-Type", "application/octet-stream"). // 或者根據(jù)文件類型設置
			SetBody(file). // 支持 io.Reader方式
			Post("https://example.com/upload")

resp, err := client.R().
		SetFormData(map[string]string{"title": "foo", "body": "bar", "userId": "1"}).
		Post("https://jsonplaceholder.typicode.com/posts")

// 請求
	type postReq struct {
		Title  string `json:"title"`
		Body   string `json:"body"`
		UserId int    `json:"userId"`
	}

	// 響應
	type postRes struct {
		ID     int    `json:"id"`
		Title  string `json:"title"`
		Body   string `json:"body"`
		UserId int    `json:"userId"`
	}

	var pr postRes

	resp, err := client.R().
		SetHeader("Content-Type", "application/json").
		SetBody(postReq{Title: "foo", Body: "bar", UserId: 1}).
		SetResult(&pr). // 設置響應后內(nèi)容
		Post("https://jsonplaceholder.typicode.com/posts")

	if err != nil {
		fmt.Println("請求出錯:", err)
		return
	}

	fmt.Println("請求成功了么？", resp.IsSuccess()) // 是否響應成了
	fmt.Printf("響應結果:%#v\n", pr)
	// 請求成功了么？ true
	// 響應結果:main.postRes{ID:101, Title:"foo", Body:"bar", UserId:1}

	client := resty.New().
		SetTimeout(5 * time.Second).          // 設置超時時間
		SetRetryCount(3).                     // 設置重試次數(shù)為 3
		SetRetryWaitTime(1 * time.Second).    // 設置重試間隔為 1 秒
		SetRetryMaxWaitTime(5 * time.Second). //最大重試間隔
		AddRetryCondition(
			func(r *resty.Response, err error) bool {
				return r.StatusCode() == http.StatusTooManyRequests // 429 錯誤時重試
			},
		)

	resp, err := client.R().
		SetHeader("Content-Type", "application/json").
		SetBody(postReq{Title: "foo", Body: "bar", UserId: 1}).
		SetResult(&pr).
		Post("https://jsonplaceholder.typicode.com/posts")

resp, err := client.R().
		SetHeader("Content-Type", "application/json").
		SetBody(postReq{Title: "foo", Body: "bar", UserId: 1}).
		SetResult(&pr).
		SetDebug(true). // 開啟調(diào)試模式
		Post("https://jsonplaceholder.typicode.com/posts")

client := resty.New()

// 請求中間件
client.OnBeforeRequest(func(c *resty.Client, req *resty.Request) error {
	fmt.Println("發(fā)送請求前:", req.URL)
	// 可以修改請求, 比如 req.SetHeader("New-Header", "value")
	return nil
})

// 響應中間件
client.OnAfterResponse(func(c *resty.Client, resp *resty.Response) error {
	fmt.Println("收到響應后:", resp.Status())
	return nil
})

resp, err := client.R().
	SetHeader("Content-Type", "application/json").
	SetBody(postReq{Title: "foo", Body: "bar", UserId: 1}).
	SetResult(&pr).
	Post("https://jsonplaceholder.typicode.com/posts")