added the transmit mux to stop disconnects while operations are

ongoing with the host

connect/host.go

@@ -82,6 +82,7 @@ type Host struct {
 
 	// Send lock
 	sendMux sync.RWMutex
+	transmitMux sync.RWMutex
 
 	coolOffBucket *rateLimiting.Bucket
 	inCoolOff     bool
@@ -221,8 +222,8 @@ func (h *Host) SetMetricsTesting(m *Metric, face interface{}) {
 
 // Disconnect closes a the Host connection under the write lock
 func (h *Host) Disconnect() {
-	h.sendMux.Lock()
-	defer h.sendMux.Unlock()
+	h.transmitMux.Lock()
+	defer h.transmitMux.Unlock()
 
 	h.disconnect()
 }

connect/transmit.go

@@ -27,6 +27,9 @@ func (c *ProtoComms) transmit(host *Host, f func(conn *grpc.ClientConn) (interfa
 		return nil, errors.New("Host address is blank, host might be receive only.")
 	}
 
+	host.transmitMux.RLock()
+	defer host.transmitMux.RUnlock()
+
 	for numRetries := 0; numRetries < MaxRetries; numRetries++ {
 		err = nil
 		//reconnect if necessary