agent: shutdown connection on node stop

Signed-off-by: Stephen J Day <[email protected]>
moby · Jul 30, 2016 · 2950f77 · 2950f77
1 parent 058bc22
commit 2950f77
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/agent/node.go b/agent/node.go
@@ -339,6 +339,7 @@ func (n *Node) Stop(ctx context.Context) error {
 			return ctx.Err()
 		default:
 			close(n.stopped)
+			n.setControlSocket(nil)
 			// recurse and wait for closure
 			return n.Stop(ctx)
 		}
@@ -430,7 +431,9 @@ func (n *Node) CertificateRequested() <-chan struct{} {
 func (n *Node) setControlSocket(conn *grpc.ClientConn) {
 	n.Lock()
 	if n.conn != nil {
-		n.conn.Close()
+		if err := n.conn.Close(); err != nil {
+			log.L.WithError(err).Error("closing grpc connection failed")
+		}
 	}
 	n.conn = conn
 	n.connCond.Broadcast()