Example usage for org.apache.hadoop.yarn.api.protocolrecords RegisterApplicationMasterResponse getSchedulerResourceTypes

List of usage examples for org.apache.hadoop.yarn.api.protocolrecords RegisterApplicationMasterResponse getSchedulerResourceTypes

Introduction

In this page you can find the example usage for org.apache.hadoop.yarn.api.protocolrecords RegisterApplicationMasterResponse getSchedulerResourceTypes.

Prototype

@Public
@Unstable
public abstract EnumSet<SchedulerResourceTypes> getSchedulerResourceTypes();

Source Link

Document

Get a set of the resource types considered by the scheduler.

Usage

From source file:org.apache.drill.yarn.appMaster.ClusterControllerImpl.java

License:Apache License

private void yarnReport() {
    RegisterApplicationMasterResponse response = yarn.getRegistrationResponse();
    LOG.info("YARN queue: " + response.getQueue());
    Resource resource = response.getMaximumResourceCapability();
    LOG.info("YARN max resource: " + resource.getMemory() + " MB, " + resource.getVirtualCores() + " cores");
    EnumSet<SchedulerResourceTypes> types = response.getSchedulerResourceTypes();
    StringBuilder buf = new StringBuilder();
    String sep = "";
    for (SchedulerResourceTypes type : types) {
        buf.append(sep);//from   ww  w  . j  a  va  2  s.c o m
        buf.append(type.toString());
        sep = ", ";
    }
    LOG.info("YARN scheduler resource types: " + buf.toString());
}

From source file:org.apache.tez.dag.app.rm.DagAwareYarnTaskScheduler.java

License:Apache License

@Override
public void start() throws Exception {
    super.start();
    client.start();// w w w. j  av a  2s .  co  m
    if (shouldReuseContainers) {
        reuseExecutor = createExecutor();
    }
    TaskSchedulerContext ctx = getContext();
    RegisterApplicationMasterResponse response = client.registerApplicationMaster(ctx.getAppHostName(),
            ctx.getAppClientPort(), ctx.getAppTrackingUrl());
    ctx.setApplicationRegistrationData(response.getMaximumResourceCapability(), response.getApplicationACLs(),
            response.getClientToAMTokenMasterKey(), response.getQueue());
    if (response.getSchedulerResourceTypes().contains(SchedulerResourceTypes.CPU)) {
        resourceCalculator = new MemCpuResourceCalculator();
    } else {
        resourceCalculator = new MemResourceCalculator();
    }
}

From source file:org.apache.tez.hadoop.shim.HadoopShim26.java

License:Apache License

@Override
public Set<String> getSupportedResourceTypes(RegisterApplicationMasterResponse response) {
    EnumSet<SchedulerResourceTypes> supportedResourceTypes = response.getSchedulerResourceTypes();
    Set<String> supportedTypes = new HashSet<String>();
    for (SchedulerResourceTypes resourceType : supportedResourceTypes) {
        supportedTypes.add(resourceType.name());
    }/*from   w  w w . ja  v a  2 s.  co m*/
    return supportedTypes;
}