close the connection with better accouting

This commit is contained in:
Chris Lu 2020-04-20 02:53:50 -07:00
parent 978da73cdb
commit 5e8bb86995

View file

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