快捷導(dǎo)航

Go語(yǔ)言封裝HTTP請(qǐng)求的Curl工具包詳解

更新時(shí)間：2025年03月11日 08:32:21 作者：熬了夜的程序員

在 Go 語(yǔ)言開發(fā)中,與 HTTP 服務(wù)進(jìn)行交互是非常常見的需求,本文將分享一個(gè)用 Go 語(yǔ)言封裝的 Curl 工具包,它提供了簡(jiǎn)潔易用的接口來(lái)進(jìn)行 HTTP 請(qǐng)求,需要的可以了解下

在 Go 語(yǔ)言開發(fā)中，與 HTTP 服務(wù)進(jìn)行交互是非常常見的需求。本文將分享一個(gè)用 Go 語(yǔ)言封裝的 Curl 工具包，它提供了簡(jiǎn)潔易用的接口來(lái)進(jìn)行 HTTP 請(qǐng)求，包括 GET、POST、PUT 和 DELETE 等常見操作。通過(guò)這個(gè)工具包，可以輕松管理 HTTP 請(qǐng)求頭、查詢參數(shù)和請(qǐng)求體，并處理響應(yīng)。

工具包結(jié)構(gòu)簡(jiǎn)介

在這個(gè) curl 工具包中，主要定義了一個(gè) Curl 結(jié)構(gòu)體，封裝了 HTTP 客戶端的常見操作。

核心結(jié)構(gòu)體定義

type Curl struct {
    client  *http.Client      // http client
    baseURL string            // base url
    headers map[string]string // headers
}

client：使用 http.Client 發(fā)送 HTTP 請(qǐng)求。

baseURL：基礎(chǔ) URL，便于在請(qǐng)求時(shí)自動(dòng)拼接路徑。

headers：一個(gè)存儲(chǔ) HTTP 請(qǐng)求頭的 map，支持動(dòng)態(tài)設(shè)置。

初始化函數(shù)

func NewCurl(baseURL string, timeout time.Duration) *Curl {
    return &Curl{
        client: &http.Client{
            Timeout: timeout,
        },
        baseURL: strings.TrimSuffix(baseURL, "/"),
        headers: make(map[string]string),
    }
}

baseURL：傳入基礎(chǔ) URL。

timeout：設(shè)置超時(shí)時(shí)間。

功能實(shí)現(xiàn)

1. 設(shè)置請(qǐng)求頭

SetHeader 用于設(shè)置 HTTP 請(qǐng)求頭：

func (c *Curl) SetHeader(key, value string) {
    c.headers[key] = value
}

可以通過(guò)如下方式動(dòng)態(tài)設(shè)置請(qǐng)求頭：

curl.SetHeader("Content-Type", "application/json")

2. 構(gòu)建請(qǐng)求

func (c *Curl) buildRequest(ctx context.Context, method, urlPath string, queryParams map[string]string, body io.Reader) (*http.Request, error) {
    // 處理完整的 URL
    fullURL := c.baseURL + urlPath
    if queryParams != nil {
        query := url.Values{}
        for key, value := range queryParams {
            query.Add(key, value)
        }
        fullURL += "?" + query.Encode()
    }

    // 創(chuàng)建請(qǐng)求
    req, err := http.NewRequestWithContext(ctx, method, fullURL, body)
    if err != nil {
        return nil, err
    }

    // 設(shè)置請(qǐng)求頭
    for key, value := range c.headers {
        req.Header.Set(key, value)
    }

???????    return req, nil
}

3. 發(fā)送請(qǐng)求

func (c *Curl) doRequest(req *http.Request) (*http.Response, error) {
    resp, err := c.client.Do(req)
    if err != nil {
        return nil, err
    }
    return resp, nil
}

4. 發(fā)送 GET 請(qǐng)求

Get 方法通過(guò) HTTP GET 請(qǐng)求獲取資源：

func (c *Curl) Get(ctx context.Context, urlPath string, queryParams map[string]string) (*http.Response, error) {
    req, err := c.buildRequest(ctx, http.MethodGet, urlPath, queryParams, nil)
    if err != nil {
        return nil, err
    }
    return c.doRequest(req)
}

示例：

response, err := curl.Get(context.TODO(), "/user", map[string]string{"id": "123"})

5. 發(fā)送 POST 請(qǐng)求

Post 方法通過(guò) HTTP POST 請(qǐng)求提交數(shù)據(jù)：

func (c *Curl) Post(ctx context.Context, urlPath string, body []byte) (*http.Response, error) {
    req, err := c.buildRequest(ctx, http.MethodPost, urlPath, nil, bytes.NewBuffer(body))
    if err != nil {
        return nil, err
    }
    return c.doRequest(req)
}

示例：

data := []byte(`{"name":"Alice"}`)
response, err := curl.Post(context.TODO(), "/user", data)

6. 發(fā)送 PUT 請(qǐng)求

Put 方法通過(guò) HTTP PUT 請(qǐng)求更新資源：

func (c *Curl) Put(ctx context.Context, urlPath string, body []byte) (*http.Response, error) {
    req, err := c.buildRequest(ctx, http.MethodPut, urlPath, nil, bytes.NewBuffer(body))
    if err != nil {
        return nil, err
    }
    return c.doRequest(req)
}

示例：

data := []byte(`{"age":30}`)
response, err := curl.Put(context.TODO(), "/user", data)

7. 發(fā)送 DELETE 請(qǐng)求

Delete 方法通過(guò) HTTP DELETE 請(qǐng)求刪除資源：

func (c *Curl) Delete(ctx context.Context, urlPath string) (*http.Response, error) {
    req, err := c.buildRequest(ctx, http.MethodDelete, urlPath, nil, nil)
    if err != nil {
        return nil, err
    }
    return c.doRequest(req)
}

示例：

response, err := curl.Delete(context.TODO(), "/user")

8. 讀取響應(yīng)體

ReadResponseBody 讀取 HTTP 響應(yīng)體并返回字節(jié)數(shù)組：

func ReadResponseBody(resp *http.Response) ([]byte, error) {
    defer func(Body io.ReadCloser) {
        err := Body.Close()
        if err != nil {
            fmt.Printf("close response body failed: %v\n", err)
        }
    }(resp.Body)
    return io.ReadAll(resp.Body)
}

示例：

body, err := ReadResponseBody(response)
if err != nil {
    fmt.Println("Read response body error:", err)
} else {
    fmt.Println("Response body:", string(body))
}

實(shí)現(xiàn)思路

使用 http.NewRequestWithContext 構(gòu)建 HTTP 請(qǐng)求對(duì)象。

通過(guò) client.Do(req) 發(fā)送請(qǐng)求。

動(dòng)態(tài)設(shè)置請(qǐng)求頭，支持不同的 Content-Type。

處理查詢參數(shù)，方便 GET 請(qǐng)求傳參。

讀取響應(yīng)體，處理服務(wù)器返回的數(shù)據(jù)。

示例

完整示例

curl := NewCurl("https://example.com", 10*time.Second)
curl.SetHeader("Authorization", "Bearer token")
resp, err := curl.Get(context.TODO(), "/api/resource", map[string]string{"key": "value"})
if err != nil {
    log.Fatalf("Failed to send GET request: %v", err)
}

body, err := ReadResponseBody(resp)
if err != nil {
    log.Fatalf("Failed to read response: %v", err)
}

fmt.Println("Response:", string(body))