From e5d53708244d0dd4f732e2de645954d7eb47071c Mon Sep 17 00:00:00 2001
From: DJ Enriquez <denrie.enriquezjr@gmail.com>
Date: Tue, 24 Apr 2018 22:14:29 -0700
Subject: [PATCH] Check if drain alloc node exists

---
 nomad/drainer/watch_jobs.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/nomad/drainer/watch_jobs.go b/nomad/drainer/watch_jobs.go
index b5173c82aae..7959d160f50 100644
--- a/nomad/drainer/watch_jobs.go
+++ b/nomad/drainer/watch_jobs.go
@@ -351,6 +351,10 @@ func handleTaskGroup(snap *state.StateSnapshot, batch bool, tg *structs.TaskGrou
 			if err != nil {
 				return err
 			}
+			// If the node doesn't exist, move on
+			if node == nil {
+				continue
+			}
 
 			onDrainingNode = node.DrainStrategy != nil
 			drainingNodes[node.ID] = onDrainingNode