fix possible connection counting error

This commit is contained in:
Chris Lu 2018-09-03 00:28:58 -07:00
parent 88f1d32cc4
commit 06b81c063a

View file

@ -66,8 +66,11 @@ func (c *Conn) Write(b []byte) (count int, e error) {
}
func (c *Conn) Close() error {
stats.ConnectionClose()
return c.Conn.Close()
err := c.Conn.Close()
if err == nil {
stats.ConnectionClose()
}
return err
}
func NewListener(addr string, timeout time.Duration) (net.Listener, error) {