Examples with ReservationAllocationStateProto org.apache.hadoop.yarn.proto.YarnProtos.ReservationAllocationStateProto used on opensource projects

Example 11 with ReservationAllocationStateProto

use of org.apache.hadoop.yarn.proto.YarnProtos.ReservationAllocationStateProto in project hadoop by apache.

the class MemoryRMStateStore method removeReservationState.

@Override
protected synchronized void removeReservationState(String planName, String reservationIdName) throws Exception {
    LOG.info("Removing reservationallocation " + reservationIdName + " for plan " + planName);
    Map<ReservationId, ReservationAllocationStateProto> planState = state.getReservationState().get(planName);
    if (planState == null) {
        throw new YarnRuntimeException("State for plan " + planName + " does " + "not exist");
    }
    ReservationId reservationId = ReservationId.parseReservationId(reservationIdName);
    planState.remove(reservationId);
    if (planState.isEmpty()) {
        state.getReservationState().remove(planName);
    }
}