在 Go 语言开发中，与 HTTP 服务进行交互是非常常见的需求。本文将分享一个用 Go 语言封装的 Curl 工具包，它提供了简洁易用的接口来进行 HTTP 请求，包括 GET、POST、PUT 和 DELETE 等常见操作。通过这个工具包，可以轻松管理 HTTP 请求头、查询参数和请求体，并处理响应。

工具包结构简介

在这个 curl 工具包中，主要定义了一个 Curl 结构体，封装了 HTTP 客户端的常见操作。

核心结构体定义

1. type Curl struct {
   
2.     client  *http.Client      // http client
   
3.     baseURL string            // base url
   
4.     headers map[string]string // headers
   
5. }

复制代码

client：使用 http.Client 发送 HTTP 请求。
baseURL：基础 URL，便于在请求时自动拼接路径。
headers：一个存储 HTTP 请求头的 map，支持动态设置。

初始化函数

1. func NewCurl(baseURL string, timeout time.Duration) *Curl {
   
2.     return &Curl{
   
3.         client: &http.Client{
   
4.             Timeout: timeout,
   
5.         },
   
6.         baseURL: strings.TrimSuffix(baseURL, "/"),
   
7.         headers: make(map[string]string),
   
8.     }
   
9. }

复制代码

baseURL：传入基础 URL。
timeout：设置超时时间。

功能实现


1. 设置请求头

SetHeader 用于设置 HTTP 请求头：

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

复制代码

可以通过如下方式动态设置请求头：

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

复制代码

2. 构建请求

1. func (c *Curl) buildRequest(ctx context.Context, method, urlPath string, queryParams map[string]string, body io.Reader) (*http.Request, error) {
   
2.     // 处理完整的 URL
   
3.     fullURL := c.baseURL + urlPath
   
4.     if queryParams != nil {
   
5.         query := url.Values{}
   
6.         for key, value := range queryParams {
   
7.             query.Add(key, value)
   
8.         }
   
9.         fullURL += "?" + query.Encode()
   
10.     }
    
11. 
    
12.     // 创建请求
    
13.     req, err := http.NewRequestWithContext(ctx, method, fullURL, body)
    
14.     if err != nil {
    
15.         return nil, err
    
16.     }
    
17. 
    
18.     // 设置请求头
    
19.     for key, value := range c.headers {
    
20.         req.Header.Set(key, value)
    
21.     }
    
22. 
    
23. ​​​​​​​    return req, nil
    
24. }

复制代码

3. 发送请求

1. func (c *Curl) doRequest(req *http.Request) (*http.Response, error) {
   
2.     resp, err := c.client.Do(req)
   
3.     if err != nil {
   
4.         return nil, err
   
5.     }
   
6.     return resp, nil
   
7. }

复制代码

4. 发送 GET 请求

Get 方法通过 HTTP GET 请求获取资源：

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

复制代码

示例：

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

复制代码

5. 发送 POST 请求

Post 方法通过 HTTP POST 请求提交数据：

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

复制代码

示例：

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

复制代码

6. 发送 PUT 请求

Put 方法通过 HTTP PUT 请求更新资源：

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

复制代码

示例：

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

复制代码

7. 发送 DELETE 请求

Delete 方法通过 HTTP DELETE 请求删除资源：

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

复制代码

示例：

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

复制代码

8. 读取响应体

ReadResponseBody 读取 HTTP 响应体并返回字节数组：

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

复制代码

示例：

1. body, err := ReadResponseBody(response)
   
2. if err != nil {
   
3.     fmt.Println("Read response body error:", err)
   
4. } else {
   
5.     fmt.Println("Response body:", string(body))
   
6. }

复制代码

实现思路

使用 http.NewRequestWithContext 构建 HTTP 请求对象。
通过 client.Do(req) 发送请求。
动态设置请求头，支持不同的 Content-Type。
处理查询参数，方便 GET 请求传参。
读取响应体，处理服务器返回的数据。

示例

完整示例

1. curl := NewCurl("https://example.com", 10*time.Second)
   
2. curl.SetHeader("Authorization", "Bearer token")
   
3. resp, err := curl.Get(context.TODO(), "/api/resource", map[string]string{"key": "value"})
   
4. if err != nil {
   
5.     log.Fatalf("Failed to send GET request: %v", err)
   
6. }
   
7. 
   
8. body, err := ReadResponseBody(resp)
   
9. if err != nil {
   
10.     log.Fatalf("Failed to read response: %v", err)
    
11. }
    
12. 
    
13. fmt.Println("Response:", string(body))

复制代码

总结

通过封装 Curl 结构体，简化了 Go 语言中与 HTTP 服务的交互过程，提供了灵活的配置和扩展能力。通过这种封装，可以快速集成 HTTP 请求，减少模板代码的编写，提升开发效率。
到此这篇关于Go语言封装HTTP请求的Curl工具包详解的文章就介绍到这了,更多相关Go封装HTTP请求内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

来源：互联网
免责声明：如果侵犯了您的权益，请联系站长（1277306191@qq.com），我们会及时删除侵权内容，谢谢合作！