Examples with StartContainerRequestPBImpl org.apache.hadoop.yarn.api.protocolrecords.impl.pb.StartContainerRequestPBImpl used on opensource projects

Example 1 with StartContainerRequestPBImpl

use of org.apache.hadoop.yarn.api.protocolrecords.impl.pb.StartContainerRequestPBImpl in project hadoop by apache.

the class NMLeveldbStateStoreService method storeContainer.

@Override
public void storeContainer(ContainerId containerId, int containerVersion, StartContainerRequest startRequest) throws IOException {
    String idStr = containerId.toString();
    if (LOG.isDebugEnabled()) {
        LOG.debug("storeContainer: containerId= " + idStr + ", startRequest= " + startRequest);
    }
    String keyRequest = CONTAINERS_KEY_PREFIX + idStr + CONTAINER_REQUEST_KEY_SUFFIX;
    String keyVersion = getContainerVersionKey(idStr);
    try {
        WriteBatch batch = db.createWriteBatch();
        try {
            batch.put(bytes(keyRequest), ((StartContainerRequestPBImpl) startRequest).getProto().toByteArray());
            if (containerVersion != 0) {
                batch.put(bytes(keyVersion), bytes(Integer.toString(containerVersion)));
            }
            db.write(batch);
        } finally {
            batch.close();
        }
    } catch (DBException e) {
        throw new IOException(e);
    }
}

Example 2 with StartContainerRequestPBImpl

use of org.apache.hadoop.yarn.api.protocolrecords.impl.pb.StartContainerRequestPBImpl in project hadoop by apache.

the class NMLeveldbStateStoreService method loadContainerState.

private RecoveredContainerState loadContainerState(ContainerId containerId, LeveldbIterator iter, String keyPrefix) throws IOException {
    RecoveredContainerState rcs = new RecoveredContainerState();
    rcs.status = RecoveredContainerStatus.REQUESTED;
    while (iter.hasNext()) {
        Entry<byte[], byte[]> entry = iter.peekNext();
        String key = asString(entry.getKey());
        if (!key.startsWith(keyPrefix)) {
            break;
        }
        iter.next();
        // start with '/'
        String suffix = key.substring(keyPrefix.length() - 1);
        if (suffix.equals(CONTAINER_REQUEST_KEY_SUFFIX)) {
            rcs.startRequest = new StartContainerRequestPBImpl(StartContainerRequestProto.parseFrom(entry.getValue()));
        } else if (suffix.equals(CONTAINER_VERSION_KEY_SUFFIX)) {
            rcs.version = Integer.parseInt(asString(entry.getValue()));
        } else if (suffix.equals(CONTAINER_DIAGS_KEY_SUFFIX)) {
            rcs.diagnostics = asString(entry.getValue());
        } else if (suffix.equals(CONTAINER_QUEUED_KEY_SUFFIX)) {
            if (rcs.status == RecoveredContainerStatus.REQUESTED) {
                rcs.status = RecoveredContainerStatus.QUEUED;
            }
        } else if (suffix.equals(CONTAINER_LAUNCHED_KEY_SUFFIX)) {
            if ((rcs.status == RecoveredContainerStatus.REQUESTED) || (rcs.status == RecoveredContainerStatus.QUEUED)) {
                rcs.status = RecoveredContainerStatus.LAUNCHED;
            }
        } else if (suffix.equals(CONTAINER_KILLED_KEY_SUFFIX)) {
            rcs.killed = true;
        } else if (suffix.equals(CONTAINER_EXIT_CODE_KEY_SUFFIX)) {
            rcs.status = RecoveredContainerStatus.COMPLETED;
            rcs.exitCode = Integer.parseInt(asString(entry.getValue()));
        } else if (suffix.equals(CONTAINER_RESOURCE_CHANGED_KEY_SUFFIX)) {
            rcs.capability = new ResourcePBImpl(ResourceProto.parseFrom(entry.getValue()));
        } else if (suffix.equals(CONTAINER_REMAIN_RETRIES_KEY_SUFFIX)) {
            rcs.setRemainingRetryAttempts(Integer.parseInt(asString(entry.getValue())));
        } else if (suffix.equals(CONTAINER_WORK_DIR_KEY_SUFFIX)) {
            rcs.setWorkDir(asString(entry.getValue()));
        } else if (suffix.equals(CONTAINER_LOG_DIR_KEY_SUFFIX)) {
            rcs.setLogDir(asString(entry.getValue()));
        } else {
            LOG.warn("the container " + containerId + " will be killed because of the unknown key " + key + " during recovery.");
            containerUnknownKeySuffixes.put(containerId, suffix);
            rcs.setRecoveryType(RecoveredContainerType.KILL);
        }
    }
    return rcs;
}