List of usage examples for org.apache.hadoop.yarn.api.records ApplicationSubmissionContext setAMContainerSpec
@Public @Stable public abstract void setAMContainerSpec(ContainerLaunchContext amContainer);
ContainerLaunchContext
to describe the Container
with which the ApplicationMaster
is launched. From source file:x10.x10rt.yarn.Client.java
License:Open Source License
public void run() throws IOException, YarnException { yarnClient.start();//www.ja va 2s . co m YarnClusterMetrics clusterMetrics = yarnClient.getYarnClusterMetrics(); // print out cluster information LOG.info("Got Cluster metric info from ASM, numNodeManagers=" + clusterMetrics.getNumNodeManagers()); List<NodeReport> clusterNodeReports = yarnClient.getNodeReports(NodeState.RUNNING); LOG.info("Got Cluster node info from ASM"); for (NodeReport node : clusterNodeReports) { LOG.info("Got node report from ASM for" + ", nodeId=" + node.getNodeId() + ", nodeAddress" + node.getHttpAddress() + ", nodeRackName" + node.getRackName() + ", nodeNumContainers" + node.getNumContainers()); } QueueInfo queueInfo = yarnClient.getQueueInfo(this.amQueue); LOG.info("Queue info" + ", queueName=" + queueInfo.getQueueName() + ", queueCurrentCapacity=" + queueInfo.getCurrentCapacity() + ", queueMaxCapacity=" + queueInfo.getMaximumCapacity() + ", queueApplicationCount=" + queueInfo.getApplications().size() + ", queueChildQueueCount=" + queueInfo.getChildQueues().size()); List<QueueUserACLInfo> listAclInfo = yarnClient.getQueueAclsInfo(); for (QueueUserACLInfo aclInfo : listAclInfo) { for (QueueACL userAcl : aclInfo.getUserAcls()) { LOG.info("User ACL Info for Queue" + ", queueName=" + aclInfo.getQueueName() + ", userAcl=" + userAcl.name()); } } // Get a new application id YarnClientApplication app = yarnClient.createApplication(); GetNewApplicationResponse appResponse = app.getNewApplicationResponse(); int maxMem = appResponse.getMaximumResourceCapability().getMemory(); LOG.info("Max mem capabililty of resources in this cluster " + maxMem); int maxVCores = appResponse.getMaximumResourceCapability().getVirtualCores(); LOG.info("Max virtual cores capabililty of resources in this cluster " + maxVCores); // set the application name ApplicationSubmissionContext appContext = app.getApplicationSubmissionContext(); final ApplicationId appId = appContext.getApplicationId(); appContext.setKeepContainersAcrossApplicationAttempts(false); appContext.setApplicationName(appName); Map<String, LocalResource> localResources = new HashMap<String, LocalResource>(); LOG.info("Copy App Master jar from local filesystem and add to local environment"); // Copy the application master jar to the filesystem // Create a local resource to point to the destination jar path FileSystem fs = FileSystem.get(conf); StringBuilder x10jars = new StringBuilder(); boolean isNative = Boolean.getBoolean(ApplicationMaster.X10_YARN_NATIVE); String[] jarfiles = classPath.split(":"); // upload jar files for (String jar : jarfiles) { if (jar.endsWith(".jar")) { String nopath = jar.substring(jar.lastIndexOf('/') + 1); LOG.info("Uploading " + nopath + " to " + fs.getUri()); x10jars.append(addToLocalResources(fs, jar, nopath, appId.toString(), localResources, null)); if (isNative) { // add the user's program. LOG.info("Uploading application " + appName + " to " + fs.getUri()); x10jars.append(':'); x10jars.append(addToLocalResources(fs, args[mainClassArg], appName, appId.toString(), localResources, null)); break; // no other jar files are needed beyond the one holding ApplicationMaster, which is the first one } else x10jars.append(':'); } } StringBuilder uploadedFiles = new StringBuilder(); // upload any files specified via -upload argument to the x10 script String upload = System.getProperty(ApplicationMaster.X10_YARNUPLOAD); if (upload != null) { String[] files = upload.split(","); for (String file : files) { String nopath = file.substring(file.lastIndexOf('/') + 1); LOG.info("Uploading file " + nopath + " to " + fs.getUri()); uploadedFiles.append(addToLocalResources(fs, file, nopath, appId.toString(), localResources, null)); uploadedFiles.append(','); } } LOG.info("Set the environment for the application master"); Map<String, String> env = new HashMap<String, String>(); //env.putAll(System.getenv()); // copy all environment variables from the client side to the application side // copy over existing environment variables for (String key : System.getenv().keySet()) { if (!key.startsWith("BASH_FUNC_") && !key.equals("LS_COLORS")) // skip some env.put(ApplicationMaster.X10YARNENV_ + key, System.getenv(key)); } String places = System.getenv(ApplicationMaster.X10_NPLACES); env.put(ApplicationMaster.X10_NPLACES, (places == null) ? "1" : places); String cores = System.getenv(ApplicationMaster.X10_NTHREADS); env.put(ApplicationMaster.X10_NTHREADS, (cores == null) ? "0" : cores); env.put(ApplicationMaster.X10_HDFS_JARS, x10jars.toString()); // At some point we should not be required to add // the hadoop specific classpaths to the env. // It should be provided out of the box. // For now setting all required classpaths including // the classpath to "." for the application jar StringBuilder classPathEnv = new StringBuilder(Environment.CLASSPATH.$$()) .append(ApplicationConstants.CLASS_PATH_SEPARATOR).append("./*"); for (String c : conf.getStrings(YarnConfiguration.YARN_APPLICATION_CLASSPATH, YarnConfiguration.DEFAULT_YARN_CROSS_PLATFORM_APPLICATION_CLASSPATH)) { classPathEnv.append(ApplicationConstants.CLASS_PATH_SEPARATOR); classPathEnv.append(c.trim()); } env.put("CLASSPATH", classPathEnv.toString()); LOG.info("Completed setting up the ApplicationManager environment " + env.toString()); // Set the necessary command to execute the application master Vector<CharSequence> vargs = new Vector<CharSequence>(this.args.length + 30); // Set java executable command LOG.info("Setting up app master command"); vargs.add(Environment.JAVA_HOME.$$() + "/bin/java"); // Set Xmx based on am memory size vargs.add("-Xmx" + amMemory + "m"); // propigate the native flag if (isNative) vargs.add("-DX10_YARN_NATIVE=true"); if (upload != null) vargs.add("-D" + ApplicationMaster.X10_YARNUPLOAD + "=" + uploadedFiles.toString()); vargs.add("-D" + ApplicationMaster.X10_YARN_MAIN + "=" + appName); vargs.add("-Dorg.apache.commons.logging.simplelog.showdatetime=true"); // Set class name vargs.add(appMasterMainClass); // add java arguments for (int i = 0; i < mainClassArg; i++) { if (i != classPathArg) // skip the classpath, as it gets reworked vargs.add(args[i]); } // add our own main class wrapper if (!isNative) vargs.add(X10MainRunner.class.getName()); // add remaining application command line arguments for (int i = mainClassArg; i < args.length; i++) vargs.add(args[i]); vargs.add("1>" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/AppMaster.stdout"); vargs.add("2>" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/AppMaster.stderr"); // Get final commmand StringBuilder command = new StringBuilder(); for (CharSequence str : vargs) { command.append(str).append(" "); } LOG.info("Completed setting up app master command " + command.toString()); List<String> commands = new ArrayList<String>(); commands.add(command.toString()); // Set up the container launch context for the application master ContainerLaunchContext amContainer = ContainerLaunchContext.newInstance(localResources, env, commands, null, null, null); // Set up resource type requirements // For now, both memory and vcores are supported, so we set memory and // vcores requirements Resource capability = Resource.newInstance(amMemory, amVCores); appContext.setResource(capability); appContext.setAMContainerSpec(amContainer); // kill the application if the user hits ctrl-c Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() { @Override public void run() { System.out.println(); System.out.println("Exiting..."); forceKillApplication(appId); } })); LOG.info("Submitting application to ASM"); yarnClient.submitApplication(appContext); // Monitor the application monitorApplication(appId); // delete jar files uploaded earlier cleanupLocalResources(fs, appId.toString()); }
From source file:yarnkit.client.YarnClientService.java
License:Apache License
@Nonnull private ApplicationSubmissionContext makeApplicationContext(@Nonnull YarnClient yarnClient) throws YarnkitException { final YarnClientApplication clientApp; try {//from w w w .j a v a 2s. c om clientApp = yarnClient.createApplication(); } catch (Exception e) { throw new YarnkitException("Failed to initialize YarnClient", e); } GetNewApplicationResponse appResp = clientApp.getNewApplicationResponse(); ApplicationSubmissionContext appContext = clientApp.getApplicationSubmissionContext(); appContext.setApplicationName(parameters.getApplicationName()); appContext.setQueue(parameters.getQueue()); final ByteBuffer serializedTokens; try { serializedTokens = getSecurityToken(conf); } catch (IOException e) { throw new YarnkitException("Failed to get a security token", e); } ContainerLaunchContextFactory clcFactory = new ContainerLaunchContextFactory( appResp.getMaximumResourceCapability(), serializedTokens); ApplicationId applicationId = appContext.getApplicationId(); ContainerLaunchParameters appMasterParams = parameters.getApplicationMasterParameters(applicationId); appContext.setResource(clcFactory.createResource(appMasterParams)); appContext.setPriority(clcFactory.createPriority(appMasterParams)); ContainerLaunchContext clc = clcFactory.create(appMasterParams); appContext.setAMContainerSpec(clc); return appContext; }
From source file:yrun.YarnRunner.java
License:Apache License
public void execute() throws IOException, YarnException, InterruptedException { LOG.info("Using application path [" + _installPath + "]"); Path jarPath = installThisJar(_installPath, _appJarFile); LOG.info("Driver installed [" + jarPath + "]"); List<Path> installedArchivePathList = install(_installPath, _archivePathList); for (Path p : installedArchivePathList) { LOG.info("Archive installed [" + p + "]"); }/*w ww.j a v a 2 s.c o m*/ YarnRunnerArgs yarnRunnerArgs = new YarnRunnerArgs(); yarnRunnerArgs.setCommand(_command); Path argsPath = installThisArgs(_installPath, yarnRunnerArgs); final YarnClient client = YarnClient.createYarnClient(); _configuration.setInt("yarn.nodemanager.delete.debug-delay-sec", (int) TimeUnit.HOURS.toSeconds(1)); client.init(_configuration); client.start(); YarnClientApplication app = client.createApplication(); ContainerLaunchContext amContainer = Records.newRecord(ContainerLaunchContext.class); Map<String, String> appMasterEnv = new HashMap<String, String>(); setupAppMasterEnv(appMasterEnv, _appJarFile); Map<String, LocalResource> localResources = new HashMap<String, LocalResource>(); { LocalResource appMasterJar = Records.newRecord(LocalResource.class); setupAppMasterJar(jarPath, appMasterJar); localResources.put(jarPath.getName(), appMasterJar); } { LocalResource appMasterArgs = Records.newRecord(LocalResource.class); setupAppMasterArgs(argsPath, appMasterArgs); localResources.put(MASTER_JSON, appMasterArgs); } List<String> vargs = new ArrayList<String>(); vargs.add(Environment.JAVA_HOME.$() + "/bin/java"); vargs.add("-Xmx256m"); vargs.add("-Djava.net.preferIPv4Stack=true"); vargs.add(YarnRunnerApplicationMaster.class.getName()); String strCommand = "(echo ENV && set && echo CURRENT_DIR_LISTING && ls -la && echo PWD && pwd && (" + StringUtils.join(" ", vargs) + "))"; strCommand += " 1>" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/stdout"; strCommand += " 2>" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/stderr"; LOG.debug("Application Master command [" + strCommand + "]"); amContainer.setCommands(Collections.singletonList(strCommand)); amContainer.setLocalResources(localResources); amContainer.setEnvironment(appMasterEnv); Resource capability = Records.newRecord(Resource.class); capability.setMemory(256); capability.setVirtualCores(1); ApplicationSubmissionContext appContext = app.getApplicationSubmissionContext(); appContext.setApplicationName(_yarnName); appContext.setAMContainerSpec(amContainer); appContext.setResource(capability); if (_queue != null) { appContext.setQueue(_queue); } appContext.setApplicationType("yrun"); ApplicationId appId = appContext.getApplicationId(); AtomicBoolean shutdown = new AtomicBoolean(); if (!_isDaemon) { addShutdownHook(client, appId, shutdown); } LOG.info("Submitting application with id [" + appId + "]"); client.submitApplication(appContext); ApplicationReport report; YarnApplicationState state; do { report = client.getApplicationReport(appId); state = report.getYarnApplicationState(); if (state == YarnApplicationState.RUNNING) { if (_isDaemon) { LOG.info("Application is running. This is a daemon application driver program exiting."); return; } } Thread.sleep(100); } while (isNoLongerRunning(state)); shutdown.set(true); LOG.info("Application has finished with state [" + state + "]"); }