YARN-3565. NodeHeartbeatRequest/RegisterNodeManagerRequest should use NodeLabel object instead of String. (Naganarasimha G R via wangda)

This commit is contained in:
Wangda Tan 2015-05-19 16:34:17 -07:00
parent 12d6c5ce4f
commit b37da52a1c
15 changed files with 149 additions and 74 deletions

View File

@ -247,6 +247,9 @@ Release 2.8.0 - UNRELEASED
YARN-3362. Add node label usage in RM CapacityScheduler web UI. YARN-3362. Add node label usage in RM CapacityScheduler web UI.
(Naganarasimha G R via wangda) (Naganarasimha G R via wangda)
YARN-3565. NodeHeartbeatRequest/RegisterNodeManagerRequest should use
NodeLabel object instead of String. (Naganarasimha G R via wangda)
OPTIMIZATIONS OPTIMIZATIONS
YARN-3339. TestDockerContainerExecutor should pull a single image and not YARN-3339. TestDockerContainerExecutor should pull a single image and not

View File

@ -253,10 +253,6 @@ message NodeIdToLabelsProto {
repeated string nodeLabels = 2; repeated string nodeLabels = 2;
} }
message StringArrayProto {
repeated string elements = 1;
}
message LabelsToNodeIdsProto { message LabelsToNodeIdsProto {
optional string nodeLabels = 1; optional string nodeLabels = 1;
repeated NodeIdProto nodeId = 2; repeated NodeIdProto nodeId = 2;

View File

@ -39,6 +39,7 @@ import java.util.regex.Pattern;
import org.apache.commons.lang.StringUtils; import org.apache.commons.lang.StringUtils;
import org.apache.commons.logging.Log; import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory; import org.apache.commons.logging.LogFactory;
import org.apache.hadoop.classification.InterfaceAudience.Private;
import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.service.AbstractService; import org.apache.hadoop.service.AbstractService;
import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeId;
@ -59,6 +60,7 @@ import org.apache.hadoop.yarn.util.resource.Resources;
import com.google.common.annotations.VisibleForTesting; import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.ImmutableSet; import com.google.common.collect.ImmutableSet;
@Private
public class CommonNodeLabelsManager extends AbstractService { public class CommonNodeLabelsManager extends AbstractService {
protected static final Log LOG = LogFactory.getLog(CommonNodeLabelsManager.class); protected static final Log LOG = LogFactory.getLog(CommonNodeLabelsManager.class);
private static final int MAX_LABEL_LENGTH = 255; private static final int MAX_LABEL_LENGTH = 255;

View File

@ -112,6 +112,18 @@ public class NodeLabelTestBase {
return set; return set;
} }
@SuppressWarnings("unchecked")
public static Set<NodeLabel> toNodeLabelSet(String... nodeLabelsStr) {
if (null == nodeLabelsStr) {
return null;
}
Set<NodeLabel> labels = new HashSet<NodeLabel>();
for (String label : nodeLabelsStr) {
labels.add(NodeLabel.newInstance(label));
}
return labels;
}
public NodeId toNodeId(String str) { public NodeId toNodeId(String str) {
if (str.contains(":")) { if (str.contains(":")) {
int idx = str.indexOf(':'); int idx = str.indexOf(':');

View File

@ -21,6 +21,7 @@ package org.apache.hadoop.yarn.server.api.protocolrecords;
import java.util.List; import java.util.List;
import java.util.Set; import java.util.Set;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.server.api.records.MasterKey; import org.apache.hadoop.yarn.server.api.records.MasterKey;
import org.apache.hadoop.yarn.server.api.records.NodeStatus; import org.apache.hadoop.yarn.server.api.records.NodeStatus;
import org.apache.hadoop.yarn.util.Records; import org.apache.hadoop.yarn.util.Records;
@ -29,7 +30,7 @@ public abstract class NodeHeartbeatRequest {
public static NodeHeartbeatRequest newInstance(NodeStatus nodeStatus, public static NodeHeartbeatRequest newInstance(NodeStatus nodeStatus,
MasterKey lastKnownContainerTokenMasterKey, MasterKey lastKnownContainerTokenMasterKey,
MasterKey lastKnownNMTokenMasterKey, Set<String> nodeLabels) { MasterKey lastKnownNMTokenMasterKey, Set<NodeLabel> nodeLabels) {
NodeHeartbeatRequest nodeHeartbeatRequest = NodeHeartbeatRequest nodeHeartbeatRequest =
Records.newRecord(NodeHeartbeatRequest.class); Records.newRecord(NodeHeartbeatRequest.class);
nodeHeartbeatRequest.setNodeStatus(nodeStatus); nodeHeartbeatRequest.setNodeStatus(nodeStatus);
@ -50,8 +51,8 @@ public abstract class NodeHeartbeatRequest {
public abstract MasterKey getLastKnownNMTokenMasterKey(); public abstract MasterKey getLastKnownNMTokenMasterKey();
public abstract void setLastKnownNMTokenMasterKey(MasterKey secretKey); public abstract void setLastKnownNMTokenMasterKey(MasterKey secretKey);
public abstract Set<String> getNodeLabels(); public abstract Set<NodeLabel> getNodeLabels();
public abstract void setNodeLabels(Set<String> nodeLabels); public abstract void setNodeLabels(Set<NodeLabel> nodeLabels);
public abstract List<LogAggregationReport> public abstract List<LogAggregationReport>
getLogAggregationReportsForApps(); getLogAggregationReportsForApps();

View File

@ -23,6 +23,7 @@ import java.util.Set;
import org.apache.hadoop.yarn.api.records.ApplicationId; import org.apache.hadoop.yarn.api.records.ApplicationId;
import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeId;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.api.records.Resource; import org.apache.hadoop.yarn.api.records.Resource;
import org.apache.hadoop.yarn.util.Records; import org.apache.hadoop.yarn.util.Records;
@ -39,7 +40,7 @@ public abstract class RegisterNodeManagerRequest {
public static RegisterNodeManagerRequest newInstance(NodeId nodeId, public static RegisterNodeManagerRequest newInstance(NodeId nodeId,
int httpPort, Resource resource, String nodeManagerVersionId, int httpPort, Resource resource, String nodeManagerVersionId,
List<NMContainerStatus> containerStatuses, List<NMContainerStatus> containerStatuses,
List<ApplicationId> runningApplications, Set<String> nodeLabels) { List<ApplicationId> runningApplications, Set<NodeLabel> nodeLabels) {
RegisterNodeManagerRequest request = RegisterNodeManagerRequest request =
Records.newRecord(RegisterNodeManagerRequest.class); Records.newRecord(RegisterNodeManagerRequest.class);
request.setHttpPort(httpPort); request.setHttpPort(httpPort);
@ -57,8 +58,8 @@ public abstract class RegisterNodeManagerRequest {
public abstract Resource getResource(); public abstract Resource getResource();
public abstract String getNMVersion(); public abstract String getNMVersion();
public abstract List<NMContainerStatus> getNMContainerStatuses(); public abstract List<NMContainerStatus> getNMContainerStatuses();
public abstract Set<String> getNodeLabels(); public abstract Set<NodeLabel> getNodeLabels();
public abstract void setNodeLabels(Set<String> nodeLabels); public abstract void setNodeLabels(Set<NodeLabel> nodeLabels);
/** /**
* We introduce this here because currently YARN RM doesn't persist nodes info * We introduce this here because currently YARN RM doesn't persist nodes info

View File

@ -24,12 +24,16 @@ import java.util.Iterator;
import java.util.List; import java.util.List;
import java.util.Set; import java.util.Set;
import org.apache.hadoop.yarn.proto.YarnProtos.StringArrayProto; import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.api.records.impl.pb.NodeLabelPBImpl;
import org.apache.hadoop.yarn.proto.YarnProtos.NodeLabelProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonProtos.MasterKeyProto; import org.apache.hadoop.yarn.proto.YarnServerCommonProtos.MasterKeyProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonProtos.NodeStatusProto; import org.apache.hadoop.yarn.proto.YarnServerCommonProtos.NodeStatusProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.LogAggregationReportProto; import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.LogAggregationReportProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeHeartbeatRequestProto; import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeHeartbeatRequestProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeHeartbeatRequestProtoOrBuilder; import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeHeartbeatRequestProtoOrBuilder;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeLabelsProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeLabelsProto.Builder;
import org.apache.hadoop.yarn.server.api.protocolrecords.LogAggregationReport; import org.apache.hadoop.yarn.server.api.protocolrecords.LogAggregationReport;
import org.apache.hadoop.yarn.server.api.protocolrecords.NodeHeartbeatRequest; import org.apache.hadoop.yarn.server.api.protocolrecords.NodeHeartbeatRequest;
import org.apache.hadoop.yarn.server.api.records.MasterKey; import org.apache.hadoop.yarn.server.api.records.MasterKey;
@ -45,7 +49,7 @@ public class NodeHeartbeatRequestPBImpl extends NodeHeartbeatRequest {
private NodeStatus nodeStatus = null; private NodeStatus nodeStatus = null;
private MasterKey lastKnownContainerTokenMasterKey = null; private MasterKey lastKnownContainerTokenMasterKey = null;
private MasterKey lastKnownNMTokenMasterKey = null; private MasterKey lastKnownNMTokenMasterKey = null;
private Set<String> labels = null; private Set<NodeLabel> labels = null;
private List<LogAggregationReport> logAggregationReportsForApps = null; private List<LogAggregationReport> logAggregationReportsForApps = null;
public NodeHeartbeatRequestPBImpl() { public NodeHeartbeatRequestPBImpl() {
@ -93,8 +97,11 @@ public class NodeHeartbeatRequestPBImpl extends NodeHeartbeatRequest {
} }
if (this.labels != null) { if (this.labels != null) {
builder.clearNodeLabels(); builder.clearNodeLabels();
builder.setNodeLabels(StringArrayProto.newBuilder() Builder newBuilder = NodeLabelsProto.newBuilder();
.addAllElements(this.labels).build()); for (NodeLabel label : labels) {
newBuilder.addNodeLabels(convertToProtoFormat(label));
}
builder.setNodeLabels(newBuilder.build());
} }
if (this.logAggregationReportsForApps != null) { if (this.logAggregationReportsForApps != null) {
addLogAggregationStatusForAppsToProto(); addLogAggregationStatusForAppsToProto();
@ -238,13 +245,13 @@ public class NodeHeartbeatRequestPBImpl extends NodeHeartbeatRequest {
} }
@Override @Override
public Set<String> getNodeLabels() { public Set<NodeLabel> getNodeLabels() {
initNodeLabels(); initNodeLabels();
return this.labels; return this.labels;
} }
@Override @Override
public void setNodeLabels(Set<String> nodeLabels) { public void setNodeLabels(Set<NodeLabel> nodeLabels) {
maybeInitBuilder(); maybeInitBuilder();
builder.clearNodeLabels(); builder.clearNodeLabels();
this.labels = nodeLabels; this.labels = nodeLabels;
@ -259,8 +266,19 @@ public class NodeHeartbeatRequestPBImpl extends NodeHeartbeatRequest {
labels = null; labels = null;
return; return;
} }
StringArrayProto nodeLabels = p.getNodeLabels(); NodeLabelsProto nodeLabels = p.getNodeLabels();
labels = new HashSet<String>(nodeLabels.getElementsList()); labels = new HashSet<NodeLabel>();
for(NodeLabelProto nlp : nodeLabels.getNodeLabelsList()) {
labels.add(convertFromProtoFormat(nlp));
}
}
private NodeLabelPBImpl convertFromProtoFormat(NodeLabelProto p) {
return new NodeLabelPBImpl(p);
}
private NodeLabelProto convertToProtoFormat(NodeLabel t) {
return ((NodeLabelPBImpl)t).getProto();
} }
@Override @Override

View File

@ -27,16 +27,19 @@ import java.util.Set;
import org.apache.hadoop.yarn.api.records.ApplicationId; import org.apache.hadoop.yarn.api.records.ApplicationId;
import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeId;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.api.records.Resource; import org.apache.hadoop.yarn.api.records.Resource;
import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationIdPBImpl; import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationIdPBImpl;
import org.apache.hadoop.yarn.api.records.impl.pb.NodeIdPBImpl; import org.apache.hadoop.yarn.api.records.impl.pb.NodeIdPBImpl;
import org.apache.hadoop.yarn.api.records.impl.pb.NodeLabelPBImpl;
import org.apache.hadoop.yarn.api.records.impl.pb.ResourcePBImpl; import org.apache.hadoop.yarn.api.records.impl.pb.ResourcePBImpl;
import org.apache.hadoop.yarn.proto.YarnProtos.ApplicationIdProto; import org.apache.hadoop.yarn.proto.YarnProtos.ApplicationIdProto;
import org.apache.hadoop.yarn.proto.YarnProtos.NodeIdProto; import org.apache.hadoop.yarn.proto.YarnProtos.NodeIdProto;
import org.apache.hadoop.yarn.proto.YarnProtos.NodeIdToLabelsProto; import org.apache.hadoop.yarn.proto.YarnProtos.NodeLabelProto;
import org.apache.hadoop.yarn.proto.YarnProtos.ResourceProto; import org.apache.hadoop.yarn.proto.YarnProtos.ResourceProto;
import org.apache.hadoop.yarn.proto.YarnProtos.StringArrayProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NMContainerStatusProto; import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NMContainerStatusProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeLabelsProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.NodeLabelsProto.Builder;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.RegisterNodeManagerRequestProto; import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.RegisterNodeManagerRequestProto;
import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.RegisterNodeManagerRequestProtoOrBuilder; import org.apache.hadoop.yarn.proto.YarnServerCommonServiceProtos.RegisterNodeManagerRequestProtoOrBuilder;
import org.apache.hadoop.yarn.server.api.protocolrecords.NMContainerStatus; import org.apache.hadoop.yarn.server.api.protocolrecords.NMContainerStatus;
@ -51,7 +54,7 @@ public class RegisterNodeManagerRequestPBImpl extends RegisterNodeManagerRequest
private NodeId nodeId = null; private NodeId nodeId = null;
private List<NMContainerStatus> containerStatuses = null; private List<NMContainerStatus> containerStatuses = null;
private List<ApplicationId> runningApplications = null; private List<ApplicationId> runningApplications = null;
private Set<String> labels = null; private Set<NodeLabel> labels = null;
public RegisterNodeManagerRequestPBImpl() { public RegisterNodeManagerRequestPBImpl() {
builder = RegisterNodeManagerRequestProto.newBuilder(); builder = RegisterNodeManagerRequestProto.newBuilder();
@ -84,8 +87,11 @@ public class RegisterNodeManagerRequestPBImpl extends RegisterNodeManagerRequest
} }
if (this.labels != null) { if (this.labels != null) {
builder.clearNodeLabels(); builder.clearNodeLabels();
builder.setNodeLabels(StringArrayProto.newBuilder() Builder newBuilder = NodeLabelsProto.newBuilder();
.addAllElements(this.labels).build()); for (NodeLabel label : labels) {
newBuilder.addNodeLabels(convertToProtoFormat(label));
}
builder.setNodeLabels(newBuilder.build());
} }
} }
@ -293,13 +299,13 @@ public class RegisterNodeManagerRequestPBImpl extends RegisterNodeManagerRequest
} }
@Override @Override
public Set<String> getNodeLabels() { public Set<NodeLabel> getNodeLabels() {
initNodeLabels(); initNodeLabels();
return this.labels; return this.labels;
} }
@Override @Override
public void setNodeLabels(Set<String> nodeLabels) { public void setNodeLabels(Set<NodeLabel> nodeLabels) {
maybeInitBuilder(); maybeInitBuilder();
builder.clearNodeLabels(); builder.clearNodeLabels();
this.labels = nodeLabels; this.labels = nodeLabels;
@ -314,8 +320,19 @@ public class RegisterNodeManagerRequestPBImpl extends RegisterNodeManagerRequest
labels=null; labels=null;
return; return;
} }
StringArrayProto nodeLabels = p.getNodeLabels(); NodeLabelsProto nodeLabels = p.getNodeLabels();
labels = new HashSet<String>(nodeLabels.getElementsList()); labels = new HashSet<NodeLabel>();
for(NodeLabelProto nlp : nodeLabels.getNodeLabelsList()) {
labels.add(convertFromProtoFormat(nlp));
}
}
private NodeLabelPBImpl convertFromProtoFormat(NodeLabelProto p) {
return new NodeLabelPBImpl(p);
}
private NodeLabelProto convertToProtoFormat(NodeLabel t) {
return ((NodeLabelPBImpl)t).getProto();
} }
private ApplicationIdPBImpl convertFromProtoFormat(ApplicationIdProto p) { private ApplicationIdPBImpl convertFromProtoFormat(ApplicationIdProto p) {

View File

@ -25,6 +25,10 @@ package hadoop.yarn;
import "yarn_protos.proto"; import "yarn_protos.proto";
import "yarn_server_common_protos.proto"; import "yarn_server_common_protos.proto";
message NodeLabelsProto {
repeated NodeLabelProto nodeLabels = 1;
}
message RegisterNodeManagerRequestProto { message RegisterNodeManagerRequestProto {
optional NodeIdProto node_id = 1; optional NodeIdProto node_id = 1;
optional int32 http_port = 3; optional int32 http_port = 3;
@ -32,7 +36,7 @@ message RegisterNodeManagerRequestProto {
optional string nm_version = 5; optional string nm_version = 5;
repeated NMContainerStatusProto container_statuses = 6; repeated NMContainerStatusProto container_statuses = 6;
repeated ApplicationIdProto runningApplications = 7; repeated ApplicationIdProto runningApplications = 7;
optional StringArrayProto nodeLabels = 8; optional NodeLabelsProto nodeLabels = 8;
} }
message RegisterNodeManagerResponseProto { message RegisterNodeManagerResponseProto {
@ -49,7 +53,7 @@ message NodeHeartbeatRequestProto {
optional NodeStatusProto node_status = 1; optional NodeStatusProto node_status = 1;
optional MasterKeyProto last_known_container_token_master_key = 2; optional MasterKeyProto last_known_container_token_master_key = 2;
optional MasterKeyProto last_known_nm_token_master_key = 3; optional MasterKeyProto last_known_nm_token_master_key = 3;
optional StringArrayProto nodeLabels = 4; optional NodeLabelsProto nodeLabels = 4;
repeated LogAggregationReportProto log_aggregation_reports_for_apps = 5; repeated LogAggregationReportProto log_aggregation_reports_for_apps = 5;
} }

View File

@ -32,6 +32,7 @@ import org.apache.hadoop.yarn.api.records.ApplicationId;
import org.apache.hadoop.yarn.api.records.ContainerId; import org.apache.hadoop.yarn.api.records.ContainerId;
import org.apache.hadoop.yarn.api.records.ContainerStatus; import org.apache.hadoop.yarn.api.records.ContainerStatus;
import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeId;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.api.records.Resource; import org.apache.hadoop.yarn.api.records.Resource;
import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationAttemptIdPBImpl; import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationAttemptIdPBImpl;
import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationIdPBImpl; import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationIdPBImpl;
@ -113,7 +114,7 @@ public class TestYarnServerApiClasses {
Assert.assertTrue(original.getNodeLabels() Assert.assertTrue(original.getNodeLabels()
.containsAll(copy.getNodeLabels())); .containsAll(copy.getNodeLabels()));
// check for empty labels // check for empty labels
original.setNodeLabels(new HashSet<String> ()); original.setNodeLabels(new HashSet<NodeLabel> ());
copy = new NodeHeartbeatRequestPBImpl( copy = new NodeHeartbeatRequestPBImpl(
original.getProto()); original.getProto());
Assert.assertNotNull(copy.getNodeLabels()); Assert.assertNotNull(copy.getNodeLabels());
@ -271,7 +272,7 @@ public class TestYarnServerApiClasses {
@Test @Test
public void testRegisterNodeManagerRequestWithValidLabels() { public void testRegisterNodeManagerRequestWithValidLabels() {
HashSet<String> nodeLabels = getValidNodeLabels(); HashSet<NodeLabel> nodeLabels = getValidNodeLabels();
RegisterNodeManagerRequest request = RegisterNodeManagerRequest request =
RegisterNodeManagerRequest.newInstance( RegisterNodeManagerRequest.newInstance(
NodeId.newInstance("host", 1234), 1234, Resource.newInstance(0, 0), NodeId.newInstance("host", 1234), 1234, Resource.newInstance(0, 0),
@ -286,19 +287,19 @@ public class TestYarnServerApiClasses {
Assert.assertEquals(true, nodeLabels.containsAll(copy.getNodeLabels())); Assert.assertEquals(true, nodeLabels.containsAll(copy.getNodeLabels()));
// check for empty labels // check for empty labels
request.setNodeLabels(new HashSet<String> ()); request.setNodeLabels(new HashSet<NodeLabel> ());
copy = new RegisterNodeManagerRequestPBImpl( copy = new RegisterNodeManagerRequestPBImpl(
((RegisterNodeManagerRequestPBImpl) request).getProto()); ((RegisterNodeManagerRequestPBImpl) request).getProto());
Assert.assertNotNull(copy.getNodeLabels()); Assert.assertNotNull(copy.getNodeLabels());
Assert.assertEquals(0, copy.getNodeLabels().size()); Assert.assertEquals(0, copy.getNodeLabels().size());
} }
private HashSet<String> getValidNodeLabels() { private HashSet<NodeLabel> getValidNodeLabels() {
HashSet<String> nodeLabels = new HashSet<String>(); HashSet<NodeLabel> nodeLabels = new HashSet<NodeLabel>();
nodeLabels.add("java"); nodeLabels.add(NodeLabel.newInstance("java"));
nodeLabels.add("windows"); nodeLabels.add(NodeLabel.newInstance("windows"));
nodeLabels.add("gpu"); nodeLabels.add(NodeLabel.newInstance("gpu"));
nodeLabels.add("x86"); nodeLabels.add(NodeLabel.newInstance("x86"));
return nodeLabels; return nodeLabels;
} }

View File

@ -30,9 +30,9 @@ import java.util.LinkedHashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Map.Entry; import java.util.Map.Entry;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.Random; import java.util.Random;
import java.util.Set; import java.util.Set;
import java.util.concurrent.ConcurrentLinkedQueue;
import org.apache.commons.logging.Log; import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory; import org.apache.commons.logging.LogFactory;
@ -50,6 +50,7 @@ import org.apache.hadoop.yarn.api.records.ContainerId;
import org.apache.hadoop.yarn.api.records.ContainerState; import org.apache.hadoop.yarn.api.records.ContainerState;
import org.apache.hadoop.yarn.api.records.ContainerStatus; import org.apache.hadoop.yarn.api.records.ContainerStatus;
import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeId;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.api.records.Resource; import org.apache.hadoop.yarn.api.records.Resource;
import org.apache.hadoop.yarn.conf.YarnConfiguration; import org.apache.hadoop.yarn.conf.YarnConfiguration;
import org.apache.hadoop.yarn.event.Dispatcher; import org.apache.hadoop.yarn.event.Dispatcher;
@ -279,11 +280,11 @@ public class NodeStatusUpdaterImpl extends AbstractService implements
protected void registerWithRM() protected void registerWithRM()
throws YarnException, IOException { throws YarnException, IOException {
List<NMContainerStatus> containerReports = getNMContainerStatuses(); List<NMContainerStatus> containerReports = getNMContainerStatuses();
Set<String> nodeLabels = null; Set<NodeLabel> nodeLabels = null;
if (hasNodeLabelsProvider) { if (hasNodeLabelsProvider) {
nodeLabels = nodeLabelsProvider.getNodeLabels(); nodeLabels = nodeLabelsProvider.getNodeLabels();
nodeLabels = nodeLabels =
(null == nodeLabels) ? CommonNodeLabelsManager.EMPTY_STRING_SET (null == nodeLabels) ? CommonNodeLabelsManager.EMPTY_NODELABEL_SET
: nodeLabels; : nodeLabels;
} }
RegisterNodeManagerRequest request = RegisterNodeManagerRequest request =
@ -628,29 +629,29 @@ public class NodeStatusUpdaterImpl extends AbstractService implements
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
public void run() { public void run() {
int lastHeartbeatID = 0; int lastHeartbeatID = 0;
Set<String> lastUpdatedNodeLabelsToRM = null; Set<NodeLabel> lastUpdatedNodeLabelsToRM = null;
if (hasNodeLabelsProvider) { if (hasNodeLabelsProvider) {
lastUpdatedNodeLabelsToRM = nodeLabelsProvider.getNodeLabels(); lastUpdatedNodeLabelsToRM = nodeLabelsProvider.getNodeLabels();
lastUpdatedNodeLabelsToRM = lastUpdatedNodeLabelsToRM =
(null == lastUpdatedNodeLabelsToRM) ? CommonNodeLabelsManager.EMPTY_STRING_SET (null == lastUpdatedNodeLabelsToRM) ? CommonNodeLabelsManager.EMPTY_NODELABEL_SET
: lastUpdatedNodeLabelsToRM; : lastUpdatedNodeLabelsToRM;
} }
while (!isStopped) { while (!isStopped) {
// Send heartbeat // Send heartbeat
try { try {
NodeHeartbeatResponse response = null; NodeHeartbeatResponse response = null;
Set<String> nodeLabelsForHeartbeat = null; Set<NodeLabel> nodeLabelsForHeartbeat = null;
NodeStatus nodeStatus = getNodeStatus(lastHeartbeatID); NodeStatus nodeStatus = getNodeStatus(lastHeartbeatID);
if (hasNodeLabelsProvider) { if (hasNodeLabelsProvider) {
nodeLabelsForHeartbeat = nodeLabelsProvider.getNodeLabels(); nodeLabelsForHeartbeat = nodeLabelsProvider.getNodeLabels();
//if the provider returns null then consider empty labels are set // if the provider returns null then consider empty labels are set
nodeLabelsForHeartbeat = nodeLabelsForHeartbeat =
(nodeLabelsForHeartbeat == null) ? CommonNodeLabelsManager.EMPTY_STRING_SET (nodeLabelsForHeartbeat == null) ? CommonNodeLabelsManager.EMPTY_NODELABEL_SET
: nodeLabelsForHeartbeat; : nodeLabelsForHeartbeat;
if (!areNodeLabelsUpdated(nodeLabelsForHeartbeat, if (!areNodeLabelsUpdated(nodeLabelsForHeartbeat,
lastUpdatedNodeLabelsToRM)) { lastUpdatedNodeLabelsToRM)) {
//if nodelabels have not changed then no need to send // if nodelabels have not changed then no need to send
nodeLabelsForHeartbeat = null; nodeLabelsForHeartbeat = null;
} }
} }
@ -781,8 +782,8 @@ public class NodeStatusUpdaterImpl extends AbstractService implements
* @param nodeLabelsOld * @param nodeLabelsOld
* @return if the New node labels are diff from the older one. * @return if the New node labels are diff from the older one.
*/ */
private boolean areNodeLabelsUpdated(Set<String> nodeLabelsNew, private boolean areNodeLabelsUpdated(Set<NodeLabel> nodeLabelsNew,
Set<String> nodeLabelsOld) { Set<NodeLabel> nodeLabelsOld) {
if (nodeLabelsNew.size() != nodeLabelsOld.size() if (nodeLabelsNew.size() != nodeLabelsOld.size()
|| !nodeLabelsOld.containsAll(nodeLabelsNew)) { || !nodeLabelsOld.containsAll(nodeLabelsNew)) {
return true; return true;

View File

@ -21,6 +21,7 @@ package org.apache.hadoop.yarn.server.nodemanager.nodelabels;
import java.util.Set; import java.util.Set;
import org.apache.hadoop.service.AbstractService; import org.apache.hadoop.service.AbstractService;
import org.apache.hadoop.yarn.api.records.NodeLabel;
/** /**
* Interface which will be responsible for fetching the labels * Interface which will be responsible for fetching the labels
@ -39,5 +40,5 @@ public abstract class NodeLabelsProvider extends AbstractService {
* *
* @return Set of node label strings applicable for a node * @return Set of node label strings applicable for a node
*/ */
public abstract Set<String> getNodeLabels(); public abstract Set<NodeLabel> getNodeLabels();
} }

View File

@ -23,16 +23,17 @@ import static org.junit.Assert.assertTrue;
import java.io.IOException; import java.io.IOException;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
import java.util.Collections;
import java.util.Set; import java.util.Set;
import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.service.ServiceOperations; import org.apache.hadoop.service.ServiceOperations;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.conf.YarnConfiguration; import org.apache.hadoop.yarn.conf.YarnConfiguration;
import org.apache.hadoop.yarn.event.Dispatcher; import org.apache.hadoop.yarn.event.Dispatcher;
import org.apache.hadoop.yarn.exceptions.YarnException; import org.apache.hadoop.yarn.exceptions.YarnException;
import org.apache.hadoop.yarn.factories.RecordFactory; import org.apache.hadoop.yarn.factories.RecordFactory;
import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider; import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider;
import org.apache.hadoop.yarn.nodelabels.CommonNodeLabelsManager;
import org.apache.hadoop.yarn.nodelabels.NodeLabelTestBase; import org.apache.hadoop.yarn.nodelabels.NodeLabelTestBase;
import org.apache.hadoop.yarn.server.api.ResourceTracker; import org.apache.hadoop.yarn.server.api.ResourceTracker;
import org.apache.hadoop.yarn.server.api.protocolrecords.NodeHeartbeatRequest; import org.apache.hadoop.yarn.server.api.protocolrecords.NodeHeartbeatRequest;
@ -71,7 +72,7 @@ public class TestNodeStatusUpdaterForLabels extends NodeLabelTestBase {
private class ResourceTrackerForLabels implements ResourceTracker { private class ResourceTrackerForLabels implements ResourceTracker {
int heartbeatID = 0; int heartbeatID = 0;
Set<String> labels; Set<NodeLabel> labels;
private boolean receivedNMHeartbeat = false; private boolean receivedNMHeartbeat = false;
private boolean receivedNMRegister = false; private boolean receivedNMRegister = false;
@ -185,18 +186,18 @@ public class TestNodeStatusUpdaterForLabels extends NodeLabelTestBase {
public static class DummyNodeLabelsProvider extends NodeLabelsProvider { public static class DummyNodeLabelsProvider extends NodeLabelsProvider {
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private Set<String> nodeLabels = Collections.EMPTY_SET; private Set<NodeLabel> nodeLabels = CommonNodeLabelsManager.EMPTY_NODELABEL_SET;
public DummyNodeLabelsProvider() { public DummyNodeLabelsProvider() {
super(DummyNodeLabelsProvider.class.getName()); super(DummyNodeLabelsProvider.class.getName());
} }
@Override @Override
public synchronized Set<String> getNodeLabels() { public synchronized Set<NodeLabel> getNodeLabels() {
return nodeLabels; return nodeLabels;
} }
synchronized void setNodeLabels(Set<String> nodeLabels) { synchronized void setNodeLabels(Set<NodeLabel> nodeLabels) {
this.nodeLabels = nodeLabels; this.nodeLabels = nodeLabels;
} }
} }
@ -245,19 +246,21 @@ public class TestNodeStatusUpdaterForLabels extends NodeLabelTestBase {
resourceTracker.resetNMHeartbeatReceiveFlag(); resourceTracker.resetNMHeartbeatReceiveFlag();
nm.start(); nm.start();
resourceTracker.waitTillRegister(); resourceTracker.waitTillRegister();
assertCollectionEquals(resourceTracker.labels, assertNLCollectionEquals(resourceTracker.labels,
dummyLabelsProviderRef.getNodeLabels()); dummyLabelsProviderRef
.getNodeLabels());
resourceTracker.waitTillHeartbeat();// wait till the first heartbeat resourceTracker.waitTillHeartbeat();// wait till the first heartbeat
resourceTracker.resetNMHeartbeatReceiveFlag(); resourceTracker.resetNMHeartbeatReceiveFlag();
// heartbeat with updated labels // heartbeat with updated labels
dummyLabelsProviderRef.setNodeLabels(toSet("P")); dummyLabelsProviderRef.setNodeLabels(toNodeLabelSet("P"));
nm.getNodeStatusUpdater().sendOutofBandHeartBeat(); nm.getNodeStatusUpdater().sendOutofBandHeartBeat();
resourceTracker.waitTillHeartbeat(); resourceTracker.waitTillHeartbeat();
assertCollectionEquals(resourceTracker.labels, assertNLCollectionEquals(resourceTracker.labels,
dummyLabelsProviderRef.getNodeLabels()); dummyLabelsProviderRef
.getNodeLabels());
resourceTracker.resetNMHeartbeatReceiveFlag(); resourceTracker.resetNMHeartbeatReceiveFlag();
// heartbeat without updating labels // heartbeat without updating labels

View File

@ -22,6 +22,7 @@ import java.io.InputStream;
import java.net.InetSocketAddress; import java.net.InetSocketAddress;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
import java.util.HashMap; import java.util.HashMap;
import java.util.HashSet;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.concurrent.ConcurrentMap; import java.util.concurrent.ConcurrentMap;
@ -42,6 +43,7 @@ import org.apache.hadoop.yarn.api.records.Container;
import org.apache.hadoop.yarn.api.records.ContainerState; import org.apache.hadoop.yarn.api.records.ContainerState;
import org.apache.hadoop.yarn.api.records.ContainerStatus; import org.apache.hadoop.yarn.api.records.ContainerStatus;
import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeId;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.api.records.Resource; import org.apache.hadoop.yarn.api.records.Resource;
import org.apache.hadoop.yarn.conf.YarnConfiguration; import org.apache.hadoop.yarn.conf.YarnConfiguration;
import org.apache.hadoop.yarn.exceptions.YarnException; import org.apache.hadoop.yarn.exceptions.YarnException;
@ -247,6 +249,17 @@ public class ResourceTrackerService extends AbstractService implements
} }
} }
static Set<String> convertToStringSet(Set<NodeLabel> nodeLabels) {
if (null == nodeLabels) {
return null;
}
Set<String> labels = new HashSet<String>();
for (NodeLabel label : nodeLabels) {
labels.add(label.getName());
}
return labels;
}
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
@Override @Override
public RegisterNodeManagerResponse registerNodeManager( public RegisterNodeManagerResponse registerNodeManager(
@ -346,7 +359,7 @@ public class ResourceTrackerService extends AbstractService implements
} }
// Update node's labels to RM's NodeLabelManager. // Update node's labels to RM's NodeLabelManager.
Set<String> nodeLabels = request.getNodeLabels(); Set<String> nodeLabels = convertToStringSet(request.getNodeLabels());
if (isDistributedNodeLabelsConf && nodeLabels != null) { if (isDistributedNodeLabelsConf && nodeLabels != null) {
try { try {
updateNodeLabelsFromNMReport(nodeLabels, nodeId); updateNodeLabelsFromNMReport(nodeLabels, nodeId);
@ -467,7 +480,8 @@ public class ResourceTrackerService extends AbstractService implements
// 5. Update node's labels to RM's NodeLabelManager. // 5. Update node's labels to RM's NodeLabelManager.
if (isDistributedNodeLabelsConf && request.getNodeLabels() != null) { if (isDistributedNodeLabelsConf && request.getNodeLabels() != null) {
try { try {
updateNodeLabelsFromNMReport(request.getNodeLabels(), nodeId); updateNodeLabelsFromNMReport(
convertToStringSet(request.getNodeLabels()), nodeId);
nodeHeartBeatResponse.setAreNodeLabelsAcceptedByRM(true); nodeHeartBeatResponse.setAreNodeLabelsAcceptedByRM(true);
} catch (IOException ex) { } catch (IOException ex) {
//ensure the error message is captured and sent across in response //ensure the error message is captured and sent across in response

View File

@ -43,6 +43,7 @@ import org.apache.hadoop.yarn.api.records.ContainerId;
import org.apache.hadoop.yarn.api.records.ContainerState; import org.apache.hadoop.yarn.api.records.ContainerState;
import org.apache.hadoop.yarn.api.records.ContainerStatus; import org.apache.hadoop.yarn.api.records.ContainerStatus;
import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeId;
import org.apache.hadoop.yarn.api.records.NodeLabel;
import org.apache.hadoop.yarn.api.records.NodeState; import org.apache.hadoop.yarn.api.records.NodeState;
import org.apache.hadoop.yarn.api.records.Priority; import org.apache.hadoop.yarn.api.records.Priority;
import org.apache.hadoop.yarn.api.records.Resource; import org.apache.hadoop.yarn.api.records.Resource;
@ -353,14 +354,14 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
registerReq.setNodeId(nodeId); registerReq.setNodeId(nodeId);
registerReq.setHttpPort(1234); registerReq.setHttpPort(1234);
registerReq.setNMVersion(YarnVersionInfo.getVersion()); registerReq.setNMVersion(YarnVersionInfo.getVersion());
registerReq.setNodeLabels(toSet("A")); registerReq.setNodeLabels(toSet(NodeLabel.newInstance("A")));
RegisterNodeManagerResponse response = RegisterNodeManagerResponse response =
resourceTrackerService.registerNodeManager(registerReq); resourceTrackerService.registerNodeManager(registerReq);
Assert.assertEquals("Action should be normal on valid Node Labels", Assert.assertEquals("Action should be normal on valid Node Labels",
NodeAction.NORMAL, response.getNodeAction()); NodeAction.NORMAL, response.getNodeAction());
assertCollectionEquals(nodeLabelsMgr.getNodeLabels().get(nodeId), assertCollectionEquals(nodeLabelsMgr.getNodeLabels().get(nodeId),
registerReq.getNodeLabels()); ResourceTrackerService.convertToStringSet(registerReq.getNodeLabels()));
Assert.assertTrue("Valid Node Labels were not accepted by RM", Assert.assertTrue("Valid Node Labels were not accepted by RM",
response.getAreNodeLabelsAcceptedByRM()); response.getAreNodeLabelsAcceptedByRM());
rm.stop(); rm.stop();
@ -402,7 +403,7 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
registerReq.setNodeId(nodeId); registerReq.setNodeId(nodeId);
registerReq.setHttpPort(1234); registerReq.setHttpPort(1234);
registerReq.setNMVersion(YarnVersionInfo.getVersion()); registerReq.setNMVersion(YarnVersionInfo.getVersion());
registerReq.setNodeLabels(toSet("A", "B", "C")); registerReq.setNodeLabels(toNodeLabelSet("A", "B", "C"));
RegisterNodeManagerResponse response = RegisterNodeManagerResponse response =
resourceTrackerService.registerNodeManager(registerReq); resourceTrackerService.registerNodeManager(registerReq);
@ -455,7 +456,7 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
req.setNodeId(nodeId); req.setNodeId(nodeId);
req.setHttpPort(1234); req.setHttpPort(1234);
req.setNMVersion(YarnVersionInfo.getVersion()); req.setNMVersion(YarnVersionInfo.getVersion());
req.setNodeLabels(toSet("#Y")); req.setNodeLabels(toNodeLabelSet("#Y"));
RegisterNodeManagerResponse response = RegisterNodeManagerResponse response =
resourceTrackerService.registerNodeManager(req); resourceTrackerService.registerNodeManager(req);
@ -506,7 +507,7 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
req.setNodeId(nodeId); req.setNodeId(nodeId);
req.setHttpPort(1234); req.setHttpPort(1234);
req.setNMVersion(YarnVersionInfo.getVersion()); req.setNMVersion(YarnVersionInfo.getVersion());
req.setNodeLabels(toSet("A")); req.setNodeLabels(toNodeLabelSet("A"));
RegisterNodeManagerResponse response = RegisterNodeManagerResponse response =
resourceTrackerService.registerNodeManager(req); resourceTrackerService.registerNodeManager(req);
// registered to RM with central label config // registered to RM with central label config
@ -568,14 +569,14 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
registerReq.setNodeId(nodeId); registerReq.setNodeId(nodeId);
registerReq.setHttpPort(1234); registerReq.setHttpPort(1234);
registerReq.setNMVersion(YarnVersionInfo.getVersion()); registerReq.setNMVersion(YarnVersionInfo.getVersion());
registerReq.setNodeLabels(toSet("A")); // Node register label registerReq.setNodeLabels(toNodeLabelSet("A")); // Node register label
RegisterNodeManagerResponse registerResponse = RegisterNodeManagerResponse registerResponse =
resourceTrackerService.registerNodeManager(registerReq); resourceTrackerService.registerNodeManager(registerReq);
// modification of labels during heartbeat // modification of labels during heartbeat
NodeHeartbeatRequest heartbeatReq = NodeHeartbeatRequest heartbeatReq =
Records.newRecord(NodeHeartbeatRequest.class); Records.newRecord(NodeHeartbeatRequest.class);
heartbeatReq.setNodeLabels(toSet("B")); // Node heartbeat label update heartbeatReq.setNodeLabels(toNodeLabelSet("B")); // Node heartbeat label update
NodeStatus nodeStatusObject = getNodeStatusObject(nodeId); NodeStatus nodeStatusObject = getNodeStatusObject(nodeId);
heartbeatReq.setNodeStatus(nodeStatusObject); heartbeatReq.setNodeStatus(nodeStatusObject);
heartbeatReq.setLastKnownNMTokenMasterKey(registerResponse heartbeatReq.setLastKnownNMTokenMasterKey(registerResponse
@ -588,7 +589,7 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
Assert.assertEquals("InValid Node Labels were not accepted by RM", Assert.assertEquals("InValid Node Labels were not accepted by RM",
NodeAction.NORMAL, nodeHeartbeatResponse.getNodeAction()); NodeAction.NORMAL, nodeHeartbeatResponse.getNodeAction());
assertCollectionEquals(nodeLabelsMgr.getNodeLabels().get(nodeId), assertCollectionEquals(nodeLabelsMgr.getNodeLabels().get(nodeId),
heartbeatReq.getNodeLabels()); ResourceTrackerService.convertToStringSet(heartbeatReq.getNodeLabels()));
Assert.assertTrue("Valid Node Labels were not accepted by RM", Assert.assertTrue("Valid Node Labels were not accepted by RM",
nodeHeartbeatResponse.getAreNodeLabelsAcceptedByRM()); nodeHeartbeatResponse.getAreNodeLabelsAcceptedByRM());
@ -652,13 +653,13 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
registerReq.setNodeId(nodeId); registerReq.setNodeId(nodeId);
registerReq.setHttpPort(1234); registerReq.setHttpPort(1234);
registerReq.setNMVersion(YarnVersionInfo.getVersion()); registerReq.setNMVersion(YarnVersionInfo.getVersion());
registerReq.setNodeLabels(toSet("A")); registerReq.setNodeLabels(toNodeLabelSet("A"));
RegisterNodeManagerResponse registerResponse = RegisterNodeManagerResponse registerResponse =
resourceTrackerService.registerNodeManager(registerReq); resourceTrackerService.registerNodeManager(registerReq);
NodeHeartbeatRequest heartbeatReq = NodeHeartbeatRequest heartbeatReq =
Records.newRecord(NodeHeartbeatRequest.class); Records.newRecord(NodeHeartbeatRequest.class);
heartbeatReq.setNodeLabels(toSet("B", "#C")); // Invalid heart beat labels heartbeatReq.setNodeLabels(toNodeLabelSet("B", "#C")); // Invalid heart beat labels
heartbeatReq.setNodeStatus(getNodeStatusObject(nodeId)); heartbeatReq.setNodeStatus(getNodeStatusObject(nodeId));
heartbeatReq.setLastKnownNMTokenMasterKey(registerResponse heartbeatReq.setLastKnownNMTokenMasterKey(registerResponse
.getNMTokenMasterKey()); .getNMTokenMasterKey());
@ -705,13 +706,13 @@ public class TestResourceTrackerService extends NodeLabelTestBase {
req.setNodeId(nodeId); req.setNodeId(nodeId);
req.setHttpPort(1234); req.setHttpPort(1234);
req.setNMVersion(YarnVersionInfo.getVersion()); req.setNMVersion(YarnVersionInfo.getVersion());
req.setNodeLabels(toSet("A", "B", "C")); req.setNodeLabels(toNodeLabelSet("A", "B", "C"));
RegisterNodeManagerResponse registerResponse = RegisterNodeManagerResponse registerResponse =
resourceTrackerService.registerNodeManager(req); resourceTrackerService.registerNodeManager(req);
NodeHeartbeatRequest heartbeatReq = NodeHeartbeatRequest heartbeatReq =
Records.newRecord(NodeHeartbeatRequest.class); Records.newRecord(NodeHeartbeatRequest.class);
heartbeatReq.setNodeLabels(toSet("B")); // Valid heart beat labels heartbeatReq.setNodeLabels(toNodeLabelSet("B")); // Valid heart beat labels
heartbeatReq.setNodeStatus(getNodeStatusObject(nodeId)); heartbeatReq.setNodeStatus(getNodeStatusObject(nodeId));
heartbeatReq.setLastKnownNMTokenMasterKey(registerResponse heartbeatReq.setLastKnownNMTokenMasterKey(registerResponse
.getNMTokenMasterKey()); .getNMTokenMasterKey());