Merge pull request #2596 from weargoggles/deregister-before-kill

Clean up consul earlier when destroying a task
hashicorp · May 2, 2017 · ba73ed5 · ba73ed5
2 parents 3d3a32e + aa2da9e
commit ba73ed5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/client/task_runner.go b/client/task_runner.go
@@ -1031,6 +1031,10 @@ func (r *TaskRunner) run() {
 					return
 				}
 
+				// Remove from consul before killing the task so that traffic
+				// can be rerouted
+				r.consul.RemoveTask(r.alloc.ID, r.task)
+
 				// Store the task event that provides context on the task
 				// destroy. The Killed event is set from the alloc_runner and
 				// doesn't add detail