连接超时这种异常怎么处理比较好_Golang中SSH.NewSession超时问题记录

背景

项目中使用http://golang.org/x/crypto/ssh来建立SSH连接，最近使用过程中功能发生异常，使用pprof查找时，发现会有大量的业务goroutine卡在client.NewSession()方法上，被hang住，没有被超时处理。review之前同事写的代码已经将clientConfig赋予了超时属性，继续trace发现源码是将config中的Timeout定义为

// Timeout is the maximum amount of time for the TCP connection to establish.
//建立TCP连接的超时时间
// A Timeout of zero means no timeout.
Timeout time.Duration

这样，也就是说，并未给NewSession赋予超时时间，开始琢磨怎么加上超时时间，让goroutine正确关闭。

解决方式

原本计划是通过包装，然后自己写超时channel来强制关掉该协程，但是觉得go包应该做了类似设计。

幸运地，发现了和我们类似情况的老铁。

SSH Connection Timeoutstackoverflow.com

提问的开发小哥，想要通过conn.SetDeadline()方式去设定超时，但这种方案，对于需要长连接的SSH通道，需要发送心跳去保持连接。

卡卡西发动写轮眼[1]

// Conn wraps a net.Conn, and sets a deadline for every read
// and write operation.
type Conn struct {net.ConnReadTimeout  time.DurationWriteTimeout time.Duration
}func (c *Conn) Read(b []byte) (int, error) {err := c.Conn.SetReadDeadline(time.Now().Add(c.ReadTimeout))if err != nil {return 0, err}return c.Conn.Read(b)
}func (c *Conn) Write(b []byte) (int, error) {err := c.Conn.SetWriteDeadline(time.Now().Add(c.WriteTimeout))if err != nil {return 0, err}return c.Conn.Write(b)
}
func SSHDialTimeout(network, addr string, config *ssh.ClientConfig, timeout time.Duration) (*ssh.Client, error) {conn, err := net.DialTimeout(network, addr, timeout)if err != nil {return nil, err}timeoutConn := &Conn{conn, timeout, timeout}c, chans, reqs, err := ssh.NewClientConn(timeoutConn, addr, config)if err != nil {return nil, err}client := ssh.NewClient(c, chans, reqs)// this sends keepalive packets every 2 seconds// there's no useful response from these, so we can just abort if there's an errorgo func() {t := time.NewTicker(2 * time.Second)defer t.Stop()for range t.C {_, _, err := client.Conn.SendRequest("keepalive@golang.org", true, nil)if err != nil {return}}}()return client, nil
}

参考文献

[1]https://stackoverflow.com/questions/31554196/ssh-connection-timeout