Golang中长连接的使用

第一次,站长亲自招 Gopher 了>>>

问题

在最近一次上线中,检查日志发现,存在很多的HTTP请求异常,具体报错内容为:

dial tcp Host:Port: connect: cannot assign requested address

在查看监控,发现CPU、服务器连接均较高,CPU负载100%、连接数达到7w左右,基本可以判断为连接数过多,本机的端口基本被占满,无法发起新的HTTP请求导致,因为连接会占用CPU资源,故而CPU也被占满。

问题解决

其后经过服务器升级、代码检查、持久化连接等,终于找到问题原因,主要是一个不用的数据被使用时导致大量Go协程并发,解决后,服务器基本正常。

问题思考

虽然问题已经解决,但是如果真的服务器的连接数爆满,那么该如何解决呢?本文基于此问题进行展开阐述。

应用基本场景

在项目开发过程中,总会遇到使用HTTP方式访问其他服务,不管是外部的三方服务还是自己公司内部的其他服务,基于HTTP的请求大都是短链接,即访问完成后链接就会被释放。但是在高并发场景下由于对方服务的情况,无法在短时间内获得请求结果,就会导致客户端积压大量的链接,甚至占满客户端的端口,无法发起新的请求。

HTTP的连接

1. 串行连接

每次发起一次HTTP请求都需要一次链接,使用TCP的三次握手完成一次链接,发起请求获取数据。

• 必须等待上一次连接结束后在发起新的HTTP请求,建立新的TCP连接。

2. 持久连接

HTTP/1.1 允许客户端在发起请求结束后仍然保持在打开状态的TCP连接,便于后续请求继续使用。

• 减少TCP连接建立握手的时间延迟
• 减少了打开连接的潜在数量

3. 并行连接

通过多条TCP连接发起并发的HTTP请求。

• 每个连接之间有较小的时间延迟
• 每次发起的HTTP请求也是一个独立的TCP连接
• 并行连接数不能太多,占用本地CPU、内存、端口等各类资源,目前浏览器基本也都支持并行连接,一般限制连接数的值,比如4个

4. 管道化连接

通过共享的TCP连接发起并发的HTTP请求。

• 建立在持久化连接的基础上,将多条请求放入队列,一同发往服务端
• 降低网络的环回时间,提高性能
• 必须确保服务端支持持久化连接
• 做好连接会在任意时间关闭的准备,准备好重发所有未完成的管道化请求

Golang中持久化连接

在Golang中使用持久化连接发起HTTP请求,主要依赖Transport,官方封装的net库中已经支持。

Transport实现了RoundTripper接口,该接口只有一个方法RoundTrip(),故Transport的入口函数就是RoundTrip()。
Transport的主要功能:

• 缓存了长连接,用于大量http请求场景下的连接复用
• 对连接做一些限制,连接超时时间,每个host的最大连接数

在实际应用中,需要在初始化HTTP的client时传入transport,以进行保持连接,Transport的主要结构为:

type Transport struct {
 // DialContext specifies the dial function for creating unencrypted TCP connections.
 // If DialContext is nil (and the deprecated Dial below is also nil),
 // then the transport dials using package net.
 //
 // DialContext runs concurrently with calls to RoundTrip.
 // A RoundTrip call that initiates a dial may end up using
 // a connection dialed previously when the earlier connection
 // becomes idle before the later DialContext completes.
 DialContext func(ctx context.Context, network, addr string) (net.Conn, error)
 // MaxIdleConns controls the maximum number of idle (keep-alive)
 // connections across all hosts. Zero means no limit.
 MaxIdleConns int
 // MaxIdleConnsPerHost, if non-zero, controls the maximum idle
 // (keep-alive) connections to keep per-host. If zero,
 // DefaultMaxIdleConnsPerHost is used.
 MaxIdleConnsPerHost int
 // MaxConnsPerHost optionally limits the total number of
 // connections per host, including connections in the dialing,
 // active, and idle states. On limit violation, dials will block.
 //
 // Zero means no limit.
 MaxConnsPerHost int
 // IdleConnTimeout is the maximum amount of time an idle
 // (keep-alive) connection will remain idle before closing
 // itself.
 // Zero means no limit.
 IdleConnTimeout time.Duration
}

具体使用Demo如下

package main
import (
 "fmt"
 "io/ioutil"
 "net"
 "net/http"
 "time"
)
var HTTPTransport = &http.Transport{
 DialContext: (&net.Dialer{
 Timeout: 30 * time.Second, // 连接超时时间
 KeepAlive: 60 * time.Second, // 保持长连接的时间
 }).DialContext, // 设置连接的参数
 MaxIdleConns: 500, // 最大空闲连接
 IdleConnTimeout: 60 * time.Second, // 空闲连接的超时时间
 ExpectContinueTimeout: 30 * time.Second, // 等待服务第一个响应的超时时间
 MaxIdleConnsPerHost: 100, // 每个host保持的空闲连接数
}
func main() {
 times := 50
 uri := "http://local.test.com/t.php"
 // uri := "http://www.baidu.com"
 
 // 短连接的情况
 start := time.Now()
 client := http.Client{} // 初始化http的client
 for i := 0; i < times; i++ {
 req, err := http.NewRequest(http.MethodGet, uri, nil)
 if err != nil {
 panic("Http Req Failed " + err.Error())
 }
 resp, err := client.Do(req) // 发起请求
 if err != nil {
 panic("Http Request Failed " + err.Error())
 }
 defer resp.Body.Close()
 ioutil.ReadAll(resp.Body)
 }
 fmt.Println("Orig GoNet Short Link", time.Since(start))
 
 // 长连接的情况
 start2 := time.Now()
 client2 := http.Client{Transport: HTTPTransport} // 初始化一个带有transport的http的client
 for i := 0; i < times; i++ {
 req, err := http.NewRequest(http.MethodGet, uri, nil)
 if err != nil {
 panic("Http Req Failed " + err.Error())
 }
 resp, err := client2.Do(req)
 if err != nil {
 panic("Http Request Failed " + err.Error())
 }
 defer resp.Body.Close()
 ioutil.ReadAll(resp.Body) // 如果不及时从请求中获取结果,此连接会占用,其他请求服务复用连接
 }
 fmt.Println("Orig GoNet Long Link", time.Since(start2))
}

经过本地测试,使用transport确实能控制客户端的连接数,使得本地资源使用得到大幅度的降低。通过netstat可以查看具体的连接情况:

在测试时也发现一个有意思的情况,如果发起请求后,而不获取请求的结果,即缺少如下代码

ioutil.ReadAll(resp.Body)

则客户端会重新建立连接发起新的请求,也即没有利用到持久化连接的优势,通过netstat可以查看到连接在持续增加:

所以大家在发起HTTP请求的时候,一定记得接收返回值,哪怕返回值数据不需要进行处理。

---

本文来自:简书

感谢作者:Reaburoa

查看原文:Golang中长连接的使用

11234 次点击 ∙ 2 赞

上一篇:Golang是如何操作excel的?

下一篇:Golang标准库——html

net 占满 context 返回值

• 编辑
• 预览