[FAB-8780] Recover a cached conn from server HUP

This change enables the janitor to detect when the
connector has already removed a connection due to
HUP. In this case, the connection needs to be removed
from sweep consideration.

Change-Id: Ie1333f09d066126affcacfae410ba5612eba7733
Signed-off-by: Troy Ronda <troy@troyronda.com>

Author: troyronda

pkg/fab/comm/connector.go

@@ -283,8 +283,7 @@ func janitor(sweepTime time.Duration, idleTime time.Duration, wg *sync.WaitGroup
 			flush(conns)
 			return
 		case c := <-conn:
-			logger.Debugf("updating connection in connection janitor")
-			conns[c.target] = c
+			cache(conns, c)
 		case <-ticker.C:
 			rm := sweep(conns, idleTime)
 			for _, target := range rm {
@@ -301,6 +300,30 @@ func janitor(sweepTime time.Duration, idleTime time.Duration, wg *sync.WaitGroup
 	}
 }
 
+func cache(conns map[string]*cachedConn, updateConn *cachedConn) {
+
+	c, ok := conns[updateConn.target]
+	if ok && updateConn.lastClose.IsZero() && updateConn.conn.GetState() == connectivity.Shutdown {
+		logger.Debugf("connection shutdown detected in connection janitor")
+		// We need to remove the connection from sweep consideration immediately
+		// since the connector has already removed it. Otherwise we can have a race
+		// between shutdown and creating a connection concurrently.
+		delete(conns, updateConn.target)
+		return
+	}
+
+	if !ok {
+		logger.Debugf("new connection in connection janitor")
+	} else if c.conn != updateConn.conn {
+		logger.Debugf("connection change in connection janitor")
+		c.conn.Close() // Not blocking
+	} else {
+		logger.Debugf("updating existing connection in connection janitor")
+	}
+
+	conns[updateConn.target] = updateConn
+}
+
 func flush(conns map[string]*cachedConn) {
 	for _, c := range conns {
 		logger.Debugf("connection janitor closing connection [%s]", c.target)