YARN-10647. Fix TestRMNodeLabelsManager failed after YARN-10501. Contributed by

Qi Zhu.

(cherry picked from commit 47420ae3ed)
This commit is contained in:
Eric Badger 2021-02-22 19:07:51 +00:00
parent b1085bc874
commit 4ef5ed382b
1 changed files with 3 additions and 1 deletions

View File

@ -645,7 +645,9 @@ public class CommonNodeLabelsManager extends AbstractService {
host.labels.addAll(labels);
for (Node node : host.nms.values()) {
replaceNodeForLabels(node.nodeId, node.labels, labels);
replaceLabelsForNode(node.nodeId, node.labels, labels);
if (node.labels != null && labels != null) {
replaceLabelsForNode(node.nodeId, node.labels, labels);
}
node.labels = null;
}
break;