发起HTTP请求

在日常开发中,发起 HTTP 请求是一个非常频繁的任务。

Go 语言提供了 net/http 包,协助用户实施与 HTTP 协议相关的开发任务。 该包既提供了 HTTP 服务器 实现,又提供了 客户端 实现。

本文介绍如何使用 net/http 包发起 HTTP 请求,覆盖大部分应用场景:

Get请求

发起 Get 请求是最常见的场景之一,使用 http.Get 函数即可。代码示例如下:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
package main

import (
    "fmt"
    "net/http"
    "io/ioutil"
)

func main() {
    rsps, err := http.Get("http://example.com")
    if err != nil {
        fmt.Println("Request failed:", err)
        return
    }
    defer rsps.Body.Close()

    body, err := ioutil.ReadAll(rsps.Body)
    if err != nil {
        fmt.Println("Read body failed:", err)
        return
    }

    fmt.Println(string(body))
}

10 行调用 http.Get 函数发起 Get 请求; 第 15 行在 main 函数结束时关闭 Body 对象; 第 17 行读取整个响应体; 第 23 行输出响应内容。

警告

http.Get 函数使用默认的客户端设置,默认不超时。 挂起的请求将劫持 goroutine ,严重时可导致程序停止响应。 因此,禁止在生产业务中使用默认客户端设置。 超时设置 一节介绍如何设置请求超时时间。

Post请求

发起 Post 请求的方式也是类似的,代码如下:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
package main

import (
    "fmt"
    "net/http"
    "io/ioutil"
)

func main() {
    rsps, err := http.Post("http://example.com/", "plain/text", nil)
    if err != nil {
        fmt.Println("Request failed:", err)
        return
    }
    defer rsps.Body.Close()

    body, err := ioutil.ReadAll(rsps.Body)
    if err != nil {
        fmt.Println("Read body failed:", err)
        return
    }

    fmt.Println(string(body))
}

注意到, http.Post 函数接口与 http.Get 稍有区别,多了两个参数:

  • Content Type 头部,这里我们填 plain/text
  • 请求体,为 io.Reader 类型,没有请求体则填 nil

超时设置

超时设置需先准备一个 http.Client 结构体, Timeout 字段即是超时设置,单位为秒。 之后调用结构体相关请求方法发起请求,示例代码片段如下:

1
2
3
4
5
    var client = &http.Client{
        Timeout: time.Second * 5,
    }

    rsps, err := client.Get("http://example.com")

小技巧

生产业务请务必设置超时时间,以免 goroutine 被超时请求劫持而停止响应。

设置头部

设置头部须直接操作 http.Request 结构体,以添加头部 X-My-Auth 为例:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package main

import (
    "fmt"
    "net/http"
    "io/ioutil"
    "time"
)

func main() {
    var client = &http.Client{
        Timeout: time.Second * 5,
    }

    rqst, err := http.NewRequest("GET", "http://example.com", nil)
    if err != nil {
        fmt.Println("New request failed:", err)
        return
    }

    rqst.Header.Add("X-My-Auth", "xxxx")

    rsps, err := client.Do(rqst)
    if err != nil {
        fmt.Println("Request failed:", err)
        return
    }
    defer rsps.Body.Close()

    body, err := ioutil.ReadAll(rsps.Body)
    if err != nil {
        fmt.Println("Read body failed:", err)
        return
    }

    fmt.Println(string(body))
}

15 行初始化 http.Request 结构体,指定请求方法, URL 以及请求体; 第 21 行调用 Header 字段 Add 方法添加给定头部; 第 23 行调用客户端 Do 方法发起请求。

Basic认证

同样,进行 Basic 认证也需要操作 http.Request 结构体。 代码片段如下:

1
2
3
4
5
6
7
8
9
    rqst, err := http.NewRequest("GET", "http://example.com", nil)
    if err != nil {
        fmt.Println("New request failed:", err)
        return
    }

    rqst.SetBasicAuth("user", "password")

    rsps, err := client.Do(rqst)

1 行初始化 http.Request 结构体; 第 7 行调用 SetBasicAuth 方法设置认证信息。

下一步

订阅更新,获取更多学习资料,请关注我们的 微信公众号

../_images/wechat-mp-qrcode.png

小菜学编程