Explorar o código

control/controlclient: make Auto.Shutdown call Direct.Close

Updates #3488

Signed-off-by: Maisem Ali <[email protected]>
Maisem Ali %!s(int64=4) %!d(string=hai) anos
pai
achega
91a8cdc84b
Modificáronse 1 ficheiros con 4 adicións e 0 borrados
  1. 4 0
      control/controlclient/auto.go

+ 4 - 0
control/controlclient/auto.go

@@ -677,6 +677,7 @@ func (c *Auto) Shutdown() {
 	c.mu.Lock()
 	inSendStatus := c.inSendStatus
 	closed := c.closed
+	direct := c.direct
 	if !closed {
 		c.closed = true
 		c.statusFunc = nil
@@ -685,6 +686,9 @@ func (c *Auto) Shutdown() {
 
 	c.logf("client.Shutdown: inSendStatus=%v", inSendStatus)
 	if !closed {
+		if direct != nil {
+			direct.Close()
+		}
 		c.unregisterHealthWatch()
 		close(c.quit)
 		c.cancelAuth()