client: do not run ar cleanup hooks if client is shutting down

hashicorp · Jan 24, 2023 · 137ce0b · 137ce0b
1 parent 672dc16
commit 137ce0b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/client/allocrunner/alloc_runner.go b/client/allocrunner/alloc_runner.go
@@ -590,9 +590,9 @@ func (ar *allocRunner) handleTaskStateUpdates() {
 		} else {
 			// there are no live runners left
 
-			// run AR pre-kill hooks if this alloc is terminal; any post-stop
-			// tasks would regularly run in this state anyway (?)
-			if done {
+			// run AR pre-kill hooks if this alloc is done, but not if it's because
+			// the agent is shutting down.
+			if !ar.isShuttingDown() && done {
 				ar.preKillHooks()
 			}