YARN-3826. Race condition in ResourceTrackerService leads to wrong
diagnostics messages. Contributed by Chengbing Liu.
This commit is contained in:
parent
a815cc157c
commit
57f1a01eda
@ -553,6 +553,9 @@ Release 2.8.0 - UNRELEASED
|
||||
YARN-3790. usedResource from rootQueue metrics may get stale data for FS
|
||||
scheduler after recovering the container (Zhihai Xu via rohithsharmaks)
|
||||
|
||||
YARN-3826. Race condition in ResourceTrackerService leads to
|
||||
wrong diagnostics messages. (Chengbing Liu via devaraj)
|
||||
|
||||
Release 2.7.1 - UNRELEASED
|
||||
|
||||
INCOMPATIBLE CHANGES
|
||||
|
@ -22,13 +22,11 @@
|
||||
|
||||
import org.apache.hadoop.yarn.api.records.ApplicationId;
|
||||
import org.apache.hadoop.yarn.api.records.ContainerId;
|
||||
import org.apache.hadoop.yarn.api.records.SerializedException;
|
||||
import org.apache.hadoop.yarn.factories.RecordFactory;
|
||||
import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider;
|
||||
import org.apache.hadoop.yarn.server.api.protocolrecords.NodeHeartbeatResponse;
|
||||
import org.apache.hadoop.yarn.server.api.records.MasterKey;
|
||||
import org.apache.hadoop.yarn.server.api.records.NodeAction;
|
||||
import org.apache.hadoop.yarn.util.Records;
|
||||
|
||||
/**
|
||||
* Server Builder utilities to construct various objects.
|
||||
@ -39,6 +37,15 @@ public class YarnServerBuilderUtils {
|
||||
private static final RecordFactory recordFactory = RecordFactoryProvider
|
||||
.getRecordFactory(null);
|
||||
|
||||
public static NodeHeartbeatResponse newNodeHeartbeatResponse(
|
||||
NodeAction action, String diagnosticsMessage) {
|
||||
NodeHeartbeatResponse response = recordFactory
|
||||
.newRecordInstance(NodeHeartbeatResponse.class);
|
||||
response.setNodeAction(action);
|
||||
response.setDiagnosticsMessage(diagnosticsMessage);
|
||||
return response;
|
||||
}
|
||||
|
||||
public static NodeHeartbeatResponse newNodeHeartbeatResponse(int responseId,
|
||||
NodeAction action, List<ContainerId> containersToCleanUp,
|
||||
List<ApplicationId> applicationsToCleanUp,
|
||||
|
@ -100,22 +100,11 @@ public class ResourceTrackerService extends AbstractService implements
|
||||
private InetSocketAddress resourceTrackerAddress;
|
||||
private String minimumNodeManagerVersion;
|
||||
|
||||
private static final NodeHeartbeatResponse resync = recordFactory
|
||||
.newRecordInstance(NodeHeartbeatResponse.class);
|
||||
private static final NodeHeartbeatResponse shutDown = recordFactory
|
||||
.newRecordInstance(NodeHeartbeatResponse.class);
|
||||
|
||||
private int minAllocMb;
|
||||
private int minAllocVcores;
|
||||
|
||||
private boolean isDistributedNodeLabelsConf;
|
||||
|
||||
static {
|
||||
resync.setNodeAction(NodeAction.RESYNC);
|
||||
|
||||
shutDown.setNodeAction(NodeAction.SHUTDOWN);
|
||||
}
|
||||
|
||||
public ResourceTrackerService(RMContext rmContext,
|
||||
NodesListManager nodesListManager,
|
||||
NMLivelinessMonitor nmLivelinessMonitor,
|
||||
@ -414,8 +403,8 @@ public NodeHeartbeatResponse nodeHeartbeat(NodeHeartbeatRequest request)
|
||||
"Disallowed NodeManager nodeId: " + nodeId + " hostname: "
|
||||
+ nodeId.getHost();
|
||||
LOG.info(message);
|
||||
shutDown.setDiagnosticsMessage(message);
|
||||
return shutDown;
|
||||
return YarnServerBuilderUtils.newNodeHeartbeatResponse(
|
||||
NodeAction.SHUTDOWN, message);
|
||||
}
|
||||
|
||||
// 2. Check if it's a registered node
|
||||
@ -424,8 +413,8 @@ public NodeHeartbeatResponse nodeHeartbeat(NodeHeartbeatRequest request)
|
||||
/* node does not exist */
|
||||
String message = "Node not found resyncing " + remoteNodeStatus.getNodeId();
|
||||
LOG.info(message);
|
||||
resync.setDiagnosticsMessage(message);
|
||||
return resync;
|
||||
return YarnServerBuilderUtils.newNodeHeartbeatResponse(NodeAction.RESYNC,
|
||||
message);
|
||||
}
|
||||
|
||||
// Send ping
|
||||
@ -445,11 +434,11 @@ public NodeHeartbeatResponse nodeHeartbeat(NodeHeartbeatRequest request)
|
||||
+ lastNodeHeartbeatResponse.getResponseId() + " nm response id:"
|
||||
+ remoteNodeStatus.getResponseId();
|
||||
LOG.info(message);
|
||||
resync.setDiagnosticsMessage(message);
|
||||
// TODO: Just sending reboot is not enough. Think more.
|
||||
this.rmContext.getDispatcher().getEventHandler().handle(
|
||||
new RMNodeEvent(nodeId, RMNodeEventType.REBOOTING));
|
||||
return resync;
|
||||
return YarnServerBuilderUtils.newNodeHeartbeatResponse(NodeAction.RESYNC,
|
||||
message);
|
||||
}
|
||||
|
||||
// Heartbeat response
|
||||
|
Loading…
Reference in New Issue
Block a user