scheduler: better error handling

hashicorp · Jul 5, 2018 · 3e7d430 · 3e7d430
1 parent 1acbf1d
commit 3e7d430
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/scheduler/generic_sched.go b/scheduler/generic_sched.go
@@ -592,14 +592,18 @@ func updateRescheduleTracker(alloc *structs.Allocation, prev *structs.Allocation
 }
 
 // findPreferredNode finds the preferred node for an allocation
-func (s *GenericScheduler) findPreferredNode(place placementResult) (node *structs.Node, err error) {
+func (s *GenericScheduler) findPreferredNode(place placementResult) (*structs.Node, error) {
 	if prev := place.PreviousAllocation(); prev != nil && place.TaskGroup().EphemeralDisk.Sticky == true {
 		var preferredNode *structs.Node
 		ws := memdb.NewWatchSet()
 		preferredNode, err = s.state.NodeByID(ws, prev.NodeID)
+		if err != nil {
+			return nil, err
+		}
+
 		if preferredNode != nil && preferredNode.Ready() {
-			node = preferredNode
+			return preferredNode, nil
 		}
 	}
-	return
+	return nil, nil
 }