taintedNodes: revert unintentional delete

hashicorp · Feb 14, 2022 · 4ea42c6 · 4ea42c6
1 parent c7bbf74
commit 4ea42c6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scheduler/util.go b/scheduler/util.go
@@ -356,6 +356,10 @@ func progressMade(result *structs.PlanResult) bool {
 func taintedNodes(state State, allocs []*structs.Allocation) (map[string]*structs.Node, error) {
 	out := make(map[string]*structs.Node)
 	for _, alloc := range allocs {
+		if _, ok := out[alloc.NodeID]; ok {
+			continue
+		}
+
 		ws := memdb.NewWatchSet()
 		node, err := state.NodeByID(ws, alloc.NodeID)
 		if err != nil {