From 913a4d30c6f26897735c59b8c4a1f92579f7ca64 Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Wed, 25 Apr 2018 13:36:03 -0700
Subject: [PATCH] Safety guard

---
 nomad/drainer/draining_node.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/nomad/drainer/draining_node.go b/nomad/drainer/draining_node.go
index 21e1f254d..5a9ee1c15 100644
--- a/nomad/drainer/draining_node.go
+++ b/nomad/drainer/draining_node.go
@@ -125,6 +125,11 @@ func (n *drainingNode) DrainingJobs() ([]structs.NamespacedID, error) {
 	n.l.RLock()
 	defer n.l.RUnlock()
 
+	// Should never happen
+	if n.node == nil || n.node.DrainStrategy == nil {
+		return nil, fmt.Errorf("node doesn't have a drain strategy set")
+	}
+
 	// Retrieve the allocs on the node
 	allocs, err := n.state.AllocsByNode(nil, n.node.ID)
 	if err != nil {