import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.NodeHeartbeatRequestPBImpl; //导入依赖的package包/类
@Override
public NodeHeartbeatResponse nodeHeartbeat(NodeHeartbeatRequest request)
    throws YarnException, IOException {
  NodeHeartbeatRequestProto requestProto = ((NodeHeartbeatRequestPBImpl)request).getProto();
  try {
    return new NodeHeartbeatResponsePBImpl(proxy.nodeHeartbeat(null, requestProto));
  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.NodeHeartbeatRequestPBImpl; //导入依赖的package包/类
@Test
public void testPbRecordFactory() {
  RecordFactory pbRecordFactory = RecordFactoryPBImpl.get();
  try {
    NodeHeartbeatRequest request = pbRecordFactory.newRecordInstance(NodeHeartbeatRequest.class);
    Assert.assertEquals(NodeHeartbeatRequestPBImpl.class, request.getClass());
  } catch (YarnRuntimeException e) {
    e.printStackTrace();
    Assert.fail("Failed to crete record");
  }
  
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.NodeHeartbeatRequestPBImpl; //导入依赖的package包/类
/**
 * Test NodeHeartbeatRequestPBImpl.
 */
@Test
public void testNodeHeartbeatRequestPBImpl() {
  NodeHeartbeatRequestPBImpl original = new NodeHeartbeatRequestPBImpl();
  original.setLastKnownContainerTokenMasterKey(getMasterKey());
  original.setLastKnownNMTokenMasterKey(getMasterKey());
  original.setNodeStatus(getNodeStatus());
  NodeHeartbeatRequestPBImpl copy = new NodeHeartbeatRequestPBImpl(
      original.getProto());
  assertEquals(1, copy.getLastKnownContainerTokenMasterKey().getKeyId());
  assertEquals(1, copy.getLastKnownNMTokenMasterKey().getKeyId());
  assertEquals("localhost", copy.getNodeStatus().getNodeId().getHost());
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.NodeHeartbeatRequestPBImpl; //导入依赖的package包/类
@Override
public NodeHeartbeatResponseProto nodeHeartbeat(RpcController controller,
    NodeHeartbeatRequestProto proto) throws ServiceException {
  NodeHeartbeatRequestPBImpl request = new NodeHeartbeatRequestPBImpl(proto);
  try {
    NodeHeartbeatResponse response = real.nodeHeartbeat(request);
    return ((NodeHeartbeatResponsePBImpl)response).getProto();
  } catch (YarnException | IOException e) {
    throw new ServiceException(e);
  }
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.NodeHeartbeatRequestPBImpl; //导入依赖的package包/类
/**
 * Test NodeHeartbeatRequestPBImpl.
 */
@Test
public void testNodeHeartbeatRequestPBImplWithNullLabels() {
  NodeHeartbeatRequestPBImpl original = new NodeHeartbeatRequestPBImpl();
  NodeHeartbeatRequestPBImpl copy =
      new NodeHeartbeatRequestPBImpl(original.getProto());
  Assert.assertNull(copy.getNodeLabels());
}