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 18:59:45 +00:00
parent 3339ff5966
commit 305c4e72c2
1 changed files with 3 additions and 1 deletions

View File

@ -646,7 +646,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;