From 46c055d57b74bb0b08c8d09196e98ebc6b6c61d1 Mon Sep 17 00:00:00 2001
From: Frank Schroeder <frank.schroeder@gmail.com>
Date: Wed, 7 Jun 2017 13:11:04 +0200
Subject: [PATCH] agent: log differnt shutdown stages

Log the different code paths during shutdown.
---
 command/agent/command.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/command/agent/command.go b/command/agent/command.go
index 2a778ec44f..1424c8da38 100644
--- a/command/agent/command.go
+++ b/command/agent/command.go
@@ -800,6 +800,7 @@ func (cmd *Command) Run(args []string) int {
 
 			graceful := (sig == os.Interrupt && !(*config.SkipLeaveOnInt)) || (sig == syscall.SIGTERM && (*config.LeaveOnTerm))
 			if !graceful {
+				cmd.UI.Output("Graceful shutdown disabled. Exiting")
 				return 1
 			}
 
@@ -807,7 +808,7 @@ func (cmd *Command) Run(args []string) int {
 			gracefulCh := make(chan struct{})
 			go func() {
 				if err := agent.Leave(); err != nil {
-					cmd.UI.Error(fmt.Sprintf("Error: %s", err))
+					cmd.UI.Error(fmt.Sprintf("Error on leave: %s", err))
 					return
 				}
 				close(gracefulCh)
@@ -816,10 +817,13 @@ func (cmd *Command) Run(args []string) int {
 			gracefulTimeout := 5 * time.Second
 			select {
 			case <-signalCh:
+				cmd.UI.Output(fmt.Sprintf("Caught second signal: %v. Exiting", sig))
 				return 1
 			case <-time.After(gracefulTimeout):
+				cmd.UI.Output("Timeout on graceful leave. Exiting")
 				return 1
 			case <-gracefulCh:
+				cmd.UI.Output("Graceful exit completed")
 				return 0
 			}
 		}