Examples with SCMHeartbeatResponseProto org.apache.hadoop.hdds.protocol.proto.StorageContainerDatanodeProtocolProtos.SCMHeartbeatResponseProto used on opensource projects

Example 1 with SCMHeartbeatResponseProto

use of org.apache.hadoop.hdds.protocol.proto.StorageContainerDatanodeProtocolProtos.SCMHeartbeatResponseProto in project ozone by apache.

the class TestEndPoint method testHeartbeat.

@Test
public void testHeartbeat() throws Exception {
    DatanodeDetails dataNode = randomDatanodeDetails();
    try (EndpointStateMachine rpcEndPoint = createEndpoint(SCMTestUtils.getConf(), serverAddress, 1000)) {
        SCMHeartbeatRequestProto request = SCMHeartbeatRequestProto.newBuilder().setDatanodeDetails(dataNode.getProtoBufMessage()).setNodeReport(HddsTestUtils.createNodeReport(Arrays.asList(getStorageReports(dataNode.getUuid())), Arrays.asList(getMetadataStorageReports(dataNode.getUuid())))).build();
        SCMHeartbeatResponseProto responseProto = rpcEndPoint.getEndPoint().sendHeartbeat(request);
        Assert.assertNotNull(responseProto);
        Assert.assertEquals(0, responseProto.getCommandsCount());
    }
}

Example 2 with SCMHeartbeatResponseProto

use of org.apache.hadoop.hdds.protocol.proto.StorageContainerDatanodeProtocolProtos.SCMHeartbeatResponseProto in project ozone by apache.

the class HeartbeatEndpointTask method call.

/**
 * Computes a result, or throws an exception if unable to do so.
 *
 * @return computed result
 * @throws Exception if unable to compute a result
 */
@Override
public EndpointStateMachine.EndPointStates call() throws Exception {
    rpcEndpoint.lock();
    SCMHeartbeatRequestProto.Builder requestBuilder = null;
    try {
        Preconditions.checkState(this.datanodeDetailsProto != null);
        LayoutVersionProto layoutinfo = toLayoutVersionProto(layoutVersionManager.getMetadataLayoutVersion(), layoutVersionManager.getSoftwareLayoutVersion());
        requestBuilder = SCMHeartbeatRequestProto.newBuilder().setDatanodeDetails(datanodeDetailsProto).setDataNodeLayoutVersion(layoutinfo);
        addReports(requestBuilder);
        addContainerActions(requestBuilder);
        addPipelineActions(requestBuilder);
        SCMHeartbeatRequestProto request = requestBuilder.build();
        if (LOG.isDebugEnabled()) {
            LOG.debug("Sending heartbeat message :: {}", request.toString());
        }
        SCMHeartbeatResponseProto response = rpcEndpoint.getEndPoint().sendHeartbeat(request);
        processResponse(response, datanodeDetailsProto);
        rpcEndpoint.setLastSuccessfulHeartbeat(ZonedDateTime.now());
        rpcEndpoint.zeroMissedCount();
    } catch (IOException ex) {
        Preconditions.checkState(requestBuilder != null);
        // put back the reports which failed to be sent
        putBackIncrementalReports(requestBuilder);
        rpcEndpoint.logIfNeeded(ex);
    } finally {
        rpcEndpoint.unlock();
    }
    return rpcEndpoint.getState();
}

Example 3 with SCMHeartbeatResponseProto

use of org.apache.hadoop.hdds.protocol.proto.StorageContainerDatanodeProtocolProtos.SCMHeartbeatResponseProto in project ozone by apache.

the class TestEndPoint method testHeartbeatWithCommandStatusReport.

@Test
public void testHeartbeatWithCommandStatusReport() throws Exception {
    DatanodeDetails dataNode = randomDatanodeDetails();
    try (EndpointStateMachine rpcEndPoint = createEndpoint(SCMTestUtils.getConf(), serverAddress, 1000)) {
        // Add some scmCommands for heartbeat response
        addScmCommands();
        SCMHeartbeatRequestProto request = SCMHeartbeatRequestProto.newBuilder().setDatanodeDetails(dataNode.getProtoBufMessage()).setNodeReport(HddsTestUtils.createNodeReport(Arrays.asList(getStorageReports(dataNode.getUuid())), Arrays.asList(getMetadataStorageReports(dataNode.getUuid())))).build();
        SCMHeartbeatResponseProto responseProto = rpcEndPoint.getEndPoint().sendHeartbeat(request);
        assertNotNull(responseProto);
        assertEquals(3, responseProto.getCommandsCount());
        assertEquals(0, scmServerImpl.getCommandStatusReportCount());
        // Send heartbeat again from heartbeat endpoint task
        final StateContext stateContext = heartbeatTaskHelper(serverAddress, 3000);
        Map<Long, CommandStatus> map = stateContext.getCommandStatusMap();
        assertNotNull(map);
        assertEquals("Should have 1 objects", 1, map.size());
        assertTrue(map.containsKey(3L));
        assertEquals(Type.deleteBlocksCommand, map.get(3L).getType());
        assertEquals(Status.PENDING, map.get(3L).getStatus());
        scmServerImpl.clearScmCommandRequests();
    }
}