Go语言如何实现关闭http请求

栏目分类：其他教程    发布日期：2023-11-16    浏览次数：198次     收藏

这篇文章主要介绍“Go语言如何实现关闭http请求”的相关知识，小编通过实际案例向大家展示操作过程，操作方法简单快捷，实用性强，希望这篇“Go语言如何实现关闭http请求”文章能帮助大家解决问题。

写在前面

面试的时候问到如何关闭http请求，一般人脱口而出的是关闭response.body，这是错误的。response是返回结果的一个结构体，跟http连接没有关系。

type Response struct {
	Status     string // e.g. "200 OK"
	StatusCode int    // e.g. 200
	Proto      string // e.g. "HTTP/1.0"
	ProtoMajor int    // e.g. 1
	ProtoMinor int    // e.g. 0
	Header Header
	Body io.ReadCloser
	ContentLength int64
	Close bool
	Uncompressed bool
	Trailer Header
	Request *Request
	TLS *tls.ConnectionState
}

Body是Response中定义的一个IO流，用来读取返回内容。调用完成之后，无论http连接是否需要关闭，都要关闭response.body。

方式一：设置请求变量的 Close 字段值为 true

设置之后

req.Close = true

func main() {
	req, err := http.NewRequest("GET", "http://www.baidu.com", nil)
	checkError(err)

	req.Close = true

	resp, err := http.DefaultClient.Do(req)
	if resp != nil {
		defer resp.Body.Close()
	}
	checkError(err)

	body, err := ioutil.ReadAll(resp.Body)
	checkError(err)

	fmt.Println(string(body))

}

func checkError(err error) {
	if err != nil {
		fmt.Printf("err:%+v
", err)
	}
}

方式二：设置 Header 请求头部选项 Connection: close

设置

req.Header.Add("Connection", "close")

func main() {
	req, err := http.NewRequest("GET", "http://www.baidu.com", nil)
	checkError(err)

	req.Header.Add("Connection", "close")

	resp, err := http.DefaultClient.Do(req)
	if resp != nil {
		defer resp.Body.Close()
	}
	checkError(err)

	body, err := ioutil.ReadAll(resp.Body)
	checkError(err)

	fmt.Println(string(body))

}

func checkError(err error) {
	if err != nil {
		fmt.Printf("err:%+v
", err)
	}
}

方式三：自定义配置的 HTTP transport 客户端

这个主要是用来取消 HTTP 全局的复用连接，调用解释之后会自动关闭http连接

func main() {
	tr := http.Transport{DisableKeepAlives: true}
	client := http.Client{Transport: &tr}
	resp, err := client.Get("http://www.baidu.com")
	checkError(err)
	if resp != nil {
		defer resp.Body.Close()
	}

	body, err := ioutil.ReadAll(resp.Body)
	checkError(err)

	fmt.Println(string(body))

}

func checkError(err error) {
	if err != nil {
		fmt.Printf("err:%+v
", err)
	}
}