From 5e8bb86995145d31fbde233b02f749d59680d45c Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@gmail.com>
Date: Mon, 20 Apr 2020 02:53:50 -0700
Subject: [PATCH] close the connection with better accouting

---
 weed/util/net_timeout.go | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/weed/util/net_timeout.go b/weed/util/net_timeout.go
index edbb1a3b7..f057a8f5b 100644
--- a/weed/util/net_timeout.go
+++ b/weed/util/net_timeout.go
@@ -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) {