<HTML>
<BODY BGCOLOR="white">
<PRE>
<FONT color="green">001</FONT>    /**<a name="line.1"></a>
<FONT color="green">002</FONT>     * Licensed to the Apache Software Foundation (ASF) under one<a name="line.2"></a>
<FONT color="green">003</FONT>     * or more contributor license agreements.  See the NOTICE file<a name="line.3"></a>
<FONT color="green">004</FONT>     * distributed with this work for additional information<a name="line.4"></a>
<FONT color="green">005</FONT>     * regarding copyright ownership.  The ASF licenses this file<a name="line.5"></a>
<FONT color="green">006</FONT>     * to you under the Apache License, Version 2.0 (the<a name="line.6"></a>
<FONT color="green">007</FONT>     * "License"); you may not use this file except in compliance<a name="line.7"></a>
<FONT color="green">008</FONT>     * with the License.  You may obtain a copy of the License at<a name="line.8"></a>
<FONT color="green">009</FONT>     *<a name="line.9"></a>
<FONT color="green">010</FONT>     *     http://www.apache.org/licenses/LICENSE-2.0<a name="line.10"></a>
<FONT color="green">011</FONT>     *<a name="line.11"></a>
<FONT color="green">012</FONT>     * Unless required by applicable law or agreed to in writing, software<a name="line.12"></a>
<FONT color="green">013</FONT>     * distributed under the License is distributed on an "AS IS" BASIS,<a name="line.13"></a>
<FONT color="green">014</FONT>     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.<a name="line.14"></a>
<FONT color="green">015</FONT>     * See the License for the specific language governing permissions and<a name="line.15"></a>
<FONT color="green">016</FONT>     * limitations under the License.<a name="line.16"></a>
<FONT color="green">017</FONT>     */<a name="line.17"></a>
<FONT color="green">018</FONT>    <a name="line.18"></a>
<FONT color="green">019</FONT>    package org.apache.hadoop.yarn.applications.distributedshell;<a name="line.19"></a>
<FONT color="green">020</FONT>    <a name="line.20"></a>
<FONT color="green">021</FONT>    import java.io.BufferedReader;<a name="line.21"></a>
<FONT color="green">022</FONT>    import java.io.IOException;<a name="line.22"></a>
<FONT color="green">023</FONT>    import java.io.InputStreamReader;<a name="line.23"></a>
<FONT color="green">024</FONT>    import java.net.InetSocketAddress;<a name="line.24"></a>
<FONT color="green">025</FONT>    import java.net.URI;<a name="line.25"></a>
<FONT color="green">026</FONT>    import java.net.URISyntaxException;<a name="line.26"></a>
<FONT color="green">027</FONT>    import java.util.ArrayList;<a name="line.27"></a>
<FONT color="green">028</FONT>    import java.util.HashMap;<a name="line.28"></a>
<FONT color="green">029</FONT>    import java.util.List;<a name="line.29"></a>
<FONT color="green">030</FONT>    import java.util.Map;<a name="line.30"></a>
<FONT color="green">031</FONT>    import java.util.Vector;<a name="line.31"></a>
<FONT color="green">032</FONT>    import java.util.concurrent.CopyOnWriteArrayList;<a name="line.32"></a>
<FONT color="green">033</FONT>    import java.util.concurrent.atomic.AtomicInteger;<a name="line.33"></a>
<FONT color="green">034</FONT>    <a name="line.34"></a>
<FONT color="green">035</FONT>    import org.apache.commons.cli.CommandLine;<a name="line.35"></a>
<FONT color="green">036</FONT>    import org.apache.commons.cli.GnuParser;<a name="line.36"></a>
<FONT color="green">037</FONT>    import org.apache.commons.cli.HelpFormatter;<a name="line.37"></a>
<FONT color="green">038</FONT>    import org.apache.commons.cli.Options;<a name="line.38"></a>
<FONT color="green">039</FONT>    import org.apache.commons.cli.ParseException;<a name="line.39"></a>
<FONT color="green">040</FONT>    import org.apache.commons.logging.Log;<a name="line.40"></a>
<FONT color="green">041</FONT>    import org.apache.commons.logging.LogFactory;<a name="line.41"></a>
<FONT color="green">042</FONT>    <a name="line.42"></a>
<FONT color="green">043</FONT>    import org.apache.hadoop.classification.InterfaceAudience;<a name="line.43"></a>
<FONT color="green">044</FONT>    import org.apache.hadoop.classification.InterfaceStability;<a name="line.44"></a>
<FONT color="green">045</FONT>    import org.apache.hadoop.conf.Configuration;<a name="line.45"></a>
<FONT color="green">046</FONT>    import org.apache.hadoop.net.NetUtils;<a name="line.46"></a>
<FONT color="green">047</FONT>    import org.apache.hadoop.security.UserGroupInformation;<a name="line.47"></a>
<FONT color="green">048</FONT>    import org.apache.hadoop.yarn.api.AMRMProtocol;<a name="line.48"></a>
<FONT color="green">049</FONT>    import org.apache.hadoop.yarn.api.ApplicationConstants;<a name="line.49"></a>
<FONT color="green">050</FONT>    import org.apache.hadoop.yarn.api.ContainerManager;<a name="line.50"></a>
<FONT color="green">051</FONT>    <a name="line.51"></a>
<FONT color="green">052</FONT>    import org.apache.hadoop.yarn.api.protocolrecords.AllocateRequest;<a name="line.52"></a>
<FONT color="green">053</FONT>    import org.apache.hadoop.yarn.api.protocolrecords.AllocateResponse;<a name="line.53"></a>
<FONT color="green">054</FONT>    import org.apache.hadoop.yarn.api.protocolrecords.FinishApplicationMasterRequest;<a name="line.54"></a>
<FONT color="green">055</FONT>    //import org.apache.hadoop.yarn.api.protocolrecords.GetContainerStatusRequest;<a name="line.55"></a>
<FONT color="green">056</FONT>    //import org.apache.hadoop.yarn.api.protocolrecords.GetContainerStatusResponse;<a name="line.56"></a>
<FONT color="green">057</FONT>    import org.apache.hadoop.yarn.api.protocolrecords.RegisterApplicationMasterRequest;<a name="line.57"></a>
<FONT color="green">058</FONT>    import org.apache.hadoop.yarn.api.protocolrecords.RegisterApplicationMasterResponse;<a name="line.58"></a>
<FONT color="green">059</FONT>    import org.apache.hadoop.yarn.api.protocolrecords.StartContainerRequest;<a name="line.59"></a>
<FONT color="green">060</FONT>    <a name="line.60"></a>
<FONT color="green">061</FONT>    import org.apache.hadoop.yarn.api.records.AMResponse;<a name="line.61"></a>
<FONT color="green">062</FONT>    import org.apache.hadoop.yarn.api.records.ApplicationAttemptId;<a name="line.62"></a>
<FONT color="green">063</FONT>    import org.apache.hadoop.yarn.api.records.Container;<a name="line.63"></a>
<FONT color="green">064</FONT>    import org.apache.hadoop.yarn.api.records.ContainerId;<a name="line.64"></a>
<FONT color="green">065</FONT>    import org.apache.hadoop.yarn.api.records.ContainerLaunchContext;<a name="line.65"></a>
<FONT color="green">066</FONT>    import org.apache.hadoop.yarn.api.records.ContainerState;<a name="line.66"></a>
<FONT color="green">067</FONT>    import org.apache.hadoop.yarn.api.records.ContainerStatus;<a name="line.67"></a>
<FONT color="green">068</FONT>    import org.apache.hadoop.yarn.api.records.FinalApplicationStatus;<a name="line.68"></a>
<FONT color="green">069</FONT>    import org.apache.hadoop.yarn.api.records.LocalResource;<a name="line.69"></a>
<FONT color="green">070</FONT>    import org.apache.hadoop.yarn.api.records.LocalResourceType;<a name="line.70"></a>
<FONT color="green">071</FONT>    import org.apache.hadoop.yarn.api.records.LocalResourceVisibility;<a name="line.71"></a>
<FONT color="green">072</FONT>    import org.apache.hadoop.yarn.api.records.Priority;<a name="line.72"></a>
<FONT color="green">073</FONT>    import org.apache.hadoop.yarn.api.records.Resource;<a name="line.73"></a>
<FONT color="green">074</FONT>    import org.apache.hadoop.yarn.api.records.ResourceRequest;<a name="line.74"></a>
<FONT color="green">075</FONT>    import org.apache.hadoop.yarn.conf.YarnConfiguration;<a name="line.75"></a>
<FONT color="green">076</FONT>    import org.apache.hadoop.yarn.exceptions.YarnRemoteException;<a name="line.76"></a>
<FONT color="green">077</FONT>    import org.apache.hadoop.yarn.ipc.YarnRPC;<a name="line.77"></a>
<FONT color="green">078</FONT>    import org.apache.hadoop.yarn.util.ConverterUtils;<a name="line.78"></a>
<FONT color="green">079</FONT>    import org.apache.hadoop.yarn.util.Records;<a name="line.79"></a>
<FONT color="green">080</FONT>    <a name="line.80"></a>
<FONT color="green">081</FONT>    /**<a name="line.81"></a>
<FONT color="green">082</FONT>     * An ApplicationMaster for executing shell commands on a set of launched containers using the YARN framework. <a name="line.82"></a>
<FONT color="green">083</FONT>     * <a name="line.83"></a>
<FONT color="green">084</FONT>     * &lt;p&gt;This class is meant to act as an example on how to write yarn-based application masters. &lt;/p&gt;<a name="line.84"></a>
<FONT color="green">085</FONT>     * <a name="line.85"></a>
<FONT color="green">086</FONT>     * &lt;p&gt; The ApplicationMaster is started on a container by the &lt;code&gt;ResourceManager&lt;/code&gt;'s launcher. <a name="line.86"></a>
<FONT color="green">087</FONT>     * The first thing that the &lt;code&gt;ApplicationMaster&lt;/code&gt; needs to do is to connect and register itself with <a name="line.87"></a>
<FONT color="green">088</FONT>     * the &lt;code&gt;ResourceManager&lt;/code&gt;. The registration sets up information within the &lt;code&gt;ResourceManager&lt;/code&gt;<a name="line.88"></a>
<FONT color="green">089</FONT>     * regarding what host:port the ApplicationMaster is listening on to provide any form of functionality to a client<a name="line.89"></a>
<FONT color="green">090</FONT>     * as well as a tracking url that a client can use to keep track of status/job history if needed. &lt;/p&gt;<a name="line.90"></a>
<FONT color="green">091</FONT>     * <a name="line.91"></a>
<FONT color="green">092</FONT>     * &lt;p&gt; The &lt;code&gt;ApplicationMaster&lt;/code&gt; needs to send a heartbeat to the &lt;code&gt;ResourceManager&lt;/code&gt; at regular intervals<a name="line.92"></a>
<FONT color="green">093</FONT>     * to inform the &lt;code&gt;ResourceManager&lt;/code&gt; that it is up and alive. The {@link AMRMProtocol#allocate} to the <a name="line.93"></a>
<FONT color="green">094</FONT>     * &lt;code&gt;ResourceManager&lt;/code&gt; from the &lt;code&gt;ApplicationMaster&lt;/code&gt; acts as a heartbeat.<a name="line.94"></a>
<FONT color="green">095</FONT>     * <a name="line.95"></a>
<FONT color="green">096</FONT>     * &lt;p&gt; For the actual handling of the job, the &lt;code&gt;ApplicationMaster&lt;/code&gt; has to request the <a name="line.96"></a>
<FONT color="green">097</FONT>     * &lt;code&gt;ResourceManager&lt;/code&gt; via {@link AllocateRequest} for the required no. of containers using {@link ResourceRequest}<a name="line.97"></a>
<FONT color="green">098</FONT>     * with the necessary resource specifications such as node location, computational (memory/disk/cpu) resource requirements.<a name="line.98"></a>
<FONT color="green">099</FONT>     * The &lt;code&gt;ResourceManager&lt;/code&gt; responds with an {@link AllocateResponse} that informs the &lt;code&gt;ApplicationMaster&lt;/code&gt; <a name="line.99"></a>
<FONT color="green">100</FONT>     * of the set of newly allocated containers, completed containers as well as current state of available resources. &lt;/p&gt;<a name="line.100"></a>
<FONT color="green">101</FONT>     * <a name="line.101"></a>
<FONT color="green">102</FONT>     * &lt;p&gt; For each allocated container, the &lt;code&gt;ApplicationMaster&lt;/code&gt; can then set up the necessary launch context via <a name="line.102"></a>
<FONT color="green">103</FONT>     * {@link ContainerLaunchContext} to specify the allocated container id, local resources required by the executable, <a name="line.103"></a>
<FONT color="green">104</FONT>     * the environment to be setup for the executable, commands to execute, etc. and submit a {@link StartContainerRequest} <a name="line.104"></a>
<FONT color="green">105</FONT>     * to the {@link ContainerManager} to launch and execute the defined commands on the given allocated container. &lt;/p&gt;<a name="line.105"></a>
<FONT color="green">106</FONT>     *  <a name="line.106"></a>
<FONT color="green">107</FONT>     * &lt;p&gt; The &lt;code&gt;ApplicationMaster&lt;/code&gt; can monitor the launched container by either querying the &lt;code&gt;ResourceManager&lt;/code&gt; <a name="line.107"></a>
<FONT color="green">108</FONT>     * using {@link AMRMProtocol#allocate} to get updates on completed containers or via the {@link ContainerManager} <a name="line.108"></a>
<FONT color="green">109</FONT>     * by querying for the status of the allocated container's {@link ContainerId}.<a name="line.109"></a>
<FONT color="green">110</FONT>     * <a name="line.110"></a>
<FONT color="green">111</FONT>     * &lt;p&gt; After the job has been completed, the &lt;code&gt;ApplicationMaster&lt;/code&gt; has to send a {@link FinishApplicationMasterRequest} <a name="line.111"></a>
<FONT color="green">112</FONT>     * to the &lt;code&gt;ResourceManager&lt;/code&gt; to inform it that the &lt;code&gt;ApplicationMaster&lt;/code&gt; has been completed. <a name="line.112"></a>
<FONT color="green">113</FONT>     */<a name="line.113"></a>
<FONT color="green">114</FONT>    @InterfaceAudience.Public<a name="line.114"></a>
<FONT color="green">115</FONT>    @InterfaceStability.Unstable<a name="line.115"></a>
<FONT color="green">116</FONT>    public class ApplicationMaster {<a name="line.116"></a>
<FONT color="green">117</FONT>    <a name="line.117"></a>
<FONT color="green">118</FONT>      private static final Log LOG = LogFactory.getLog(ApplicationMaster.class);<a name="line.118"></a>
<FONT color="green">119</FONT>    <a name="line.119"></a>
<FONT color="green">120</FONT>      // Configuration <a name="line.120"></a>
<FONT color="green">121</FONT>      private Configuration conf;<a name="line.121"></a>
<FONT color="green">122</FONT>      // YARN RPC to communicate with the Resource Manager or Node Manager<a name="line.122"></a>
<FONT color="green">123</FONT>      private YarnRPC rpc;<a name="line.123"></a>
<FONT color="green">124</FONT>    <a name="line.124"></a>
<FONT color="green">125</FONT>      // Handle to communicate with the Resource Manager<a name="line.125"></a>
<FONT color="green">126</FONT>      private AMRMProtocol resourceManager;<a name="line.126"></a>
<FONT color="green">127</FONT>    <a name="line.127"></a>
<FONT color="green">128</FONT>      // Application Attempt Id ( combination of attemptId and fail count )<a name="line.128"></a>
<FONT color="green">129</FONT>      private ApplicationAttemptId appAttemptID;<a name="line.129"></a>
<FONT color="green">130</FONT>    <a name="line.130"></a>
<FONT color="green">131</FONT>      // TODO<a name="line.131"></a>
<FONT color="green">132</FONT>      // For status update for clients - yet to be implemented<a name="line.132"></a>
<FONT color="green">133</FONT>      // Hostname of the container <a name="line.133"></a>
<FONT color="green">134</FONT>      private String appMasterHostname = "";<a name="line.134"></a>
<FONT color="green">135</FONT>      // Port on which the app master listens for status update requests from clients<a name="line.135"></a>
<FONT color="green">136</FONT>      private int appMasterRpcPort = 0;<a name="line.136"></a>
<FONT color="green">137</FONT>      // Tracking url to which app master publishes info for clients to monitor <a name="line.137"></a>
<FONT color="green">138</FONT>      private String appMasterTrackingUrl = "";<a name="line.138"></a>
<FONT color="green">139</FONT>    <a name="line.139"></a>
<FONT color="green">140</FONT>      // App Master configuration<a name="line.140"></a>
<FONT color="green">141</FONT>      // No. of containers to run shell command on<a name="line.141"></a>
<FONT color="green">142</FONT>      private int numTotalContainers = 1;<a name="line.142"></a>
<FONT color="green">143</FONT>      // Memory to request for the container on which the shell command will run <a name="line.143"></a>
<FONT color="green">144</FONT>      private int containerMemory = 10;<a name="line.144"></a>
<FONT color="green">145</FONT>      // Priority of the request<a name="line.145"></a>
<FONT color="green">146</FONT>      private int requestPriority; <a name="line.146"></a>
<FONT color="green">147</FONT>    <a name="line.147"></a>
<FONT color="green">148</FONT>      // Incremental counter for rpc calls to the RM<a name="line.148"></a>
<FONT color="green">149</FONT>      private AtomicInteger rmRequestID = new AtomicInteger();<a name="line.149"></a>
<FONT color="green">150</FONT>    <a name="line.150"></a>
<FONT color="green">151</FONT>      // Simple flag to denote whether all works is done<a name="line.151"></a>
<FONT color="green">152</FONT>      private boolean appDone = false; <a name="line.152"></a>
<FONT color="green">153</FONT>      // Counter for completed containers ( complete denotes successful or failed )<a name="line.153"></a>
<FONT color="green">154</FONT>      private AtomicInteger numCompletedContainers = new AtomicInteger();<a name="line.154"></a>
<FONT color="green">155</FONT>      // Allocated container count so that we know how many containers has the RM<a name="line.155"></a>
<FONT color="green">156</FONT>      // allocated to us<a name="line.156"></a>
<FONT color="green">157</FONT>      private AtomicInteger numAllocatedContainers = new AtomicInteger();<a name="line.157"></a>
<FONT color="green">158</FONT>      // Count of failed containers <a name="line.158"></a>
<FONT color="green">159</FONT>      private AtomicInteger numFailedContainers = new AtomicInteger();<a name="line.159"></a>
<FONT color="green">160</FONT>      // Count of containers already requested from the RM<a name="line.160"></a>
<FONT color="green">161</FONT>      // Needed as once requested, we should not request for containers again and again. <a name="line.161"></a>
<FONT color="green">162</FONT>      // Only request for more if the original requirement changes. <a name="line.162"></a>
<FONT color="green">163</FONT>      private AtomicInteger numRequestedContainers = new AtomicInteger();<a name="line.163"></a>
<FONT color="green">164</FONT>    <a name="line.164"></a>
<FONT color="green">165</FONT>      // Shell command to be executed <a name="line.165"></a>
<FONT color="green">166</FONT>      private String shellCommand = ""; <a name="line.166"></a>
<FONT color="green">167</FONT>      // Args to be passed to the shell command<a name="line.167"></a>
<FONT color="green">168</FONT>      private String shellArgs = "";<a name="line.168"></a>
<FONT color="green">169</FONT>      // Env variables to be setup for the shell command <a name="line.169"></a>
<FONT color="green">170</FONT>      private Map&lt;String, String&gt; shellEnv = new HashMap&lt;String, String&gt;();<a name="line.170"></a>
<FONT color="green">171</FONT>    <a name="line.171"></a>
<FONT color="green">172</FONT>      // Location of shell script ( obtained from info set in env )<a name="line.172"></a>
<FONT color="green">173</FONT>      // Shell script path in fs<a name="line.173"></a>
<FONT color="green">174</FONT>      private String shellScriptPath = ""; <a name="line.174"></a>
<FONT color="green">175</FONT>      // Timestamp needed for creating a local resource<a name="line.175"></a>
<FONT color="green">176</FONT>      private long shellScriptPathTimestamp = 0;<a name="line.176"></a>
<FONT color="green">177</FONT>      // File length needed for local resource<a name="line.177"></a>
<FONT color="green">178</FONT>      private long shellScriptPathLen = 0;<a name="line.178"></a>
<FONT color="green">179</FONT>    <a name="line.179"></a>
<FONT color="green">180</FONT>      // Hardcoded path to shell script in launch container's local env<a name="line.180"></a>
<FONT color="green">181</FONT>      private final String ExecShellStringPath = "ExecShellScript.sh";<a name="line.181"></a>
<FONT color="green">182</FONT>    <a name="line.182"></a>
<FONT color="green">183</FONT>      // Containers to be released<a name="line.183"></a>
<FONT color="green">184</FONT>      private CopyOnWriteArrayList&lt;ContainerId&gt; releasedContainers = new CopyOnWriteArrayList&lt;ContainerId&gt;();<a name="line.184"></a>
<FONT color="green">185</FONT>    <a name="line.185"></a>
<FONT color="green">186</FONT>      // Launch threads<a name="line.186"></a>
<FONT color="green">187</FONT>      private List&lt;Thread&gt; launchThreads = new ArrayList&lt;Thread&gt;();<a name="line.187"></a>
<FONT color="green">188</FONT>    <a name="line.188"></a>
<FONT color="green">189</FONT>      /**<a name="line.189"></a>
<FONT color="green">190</FONT>       * @param args Command line args<a name="line.190"></a>
<FONT color="green">191</FONT>       */<a name="line.191"></a>
<FONT color="green">192</FONT>      public static void main(String[] args) {<a name="line.192"></a>
<FONT color="green">193</FONT>        boolean result = false;<a name="line.193"></a>
<FONT color="green">194</FONT>        try {<a name="line.194"></a>
<FONT color="green">195</FONT>          ApplicationMaster appMaster = new ApplicationMaster();<a name="line.195"></a>
<FONT color="green">196</FONT>          LOG.info("Initializing ApplicationMaster");<a name="line.196"></a>
<FONT color="green">197</FONT>          boolean doRun = appMaster.init(args);<a name="line.197"></a>
<FONT color="green">198</FONT>          if (!doRun) {<a name="line.198"></a>
<FONT color="green">199</FONT>            System.exit(0);<a name="line.199"></a>
<FONT color="green">200</FONT>          }<a name="line.200"></a>
<FONT color="green">201</FONT>          result = appMaster.run();<a name="line.201"></a>
<FONT color="green">202</FONT>        } catch (Throwable t) {<a name="line.202"></a>
<FONT color="green">203</FONT>          LOG.fatal("Error running ApplicationMaster", t);<a name="line.203"></a>
<FONT color="green">204</FONT>          System.exit(1);<a name="line.204"></a>
<FONT color="green">205</FONT>        }<a name="line.205"></a>
<FONT color="green">206</FONT>        if (result) {<a name="line.206"></a>
<FONT color="green">207</FONT>          LOG.info("Application Master completed successfully. exiting");<a name="line.207"></a>
<FONT color="green">208</FONT>          System.exit(0);<a name="line.208"></a>
<FONT color="green">209</FONT>        }<a name="line.209"></a>
<FONT color="green">210</FONT>        else {<a name="line.210"></a>
<FONT color="green">211</FONT>          LOG.info("Application Master failed. exiting");<a name="line.211"></a>
<FONT color="green">212</FONT>          System.exit(2);<a name="line.212"></a>
<FONT color="green">213</FONT>        }<a name="line.213"></a>
<FONT color="green">214</FONT>      }<a name="line.214"></a>
<FONT color="green">215</FONT>    <a name="line.215"></a>
<FONT color="green">216</FONT>      /**<a name="line.216"></a>
<FONT color="green">217</FONT>       * Dump out contents of $CWD and the environment to stdout for debugging<a name="line.217"></a>
<FONT color="green">218</FONT>       */<a name="line.218"></a>
<FONT color="green">219</FONT>      private void dumpOutDebugInfo() {<a name="line.219"></a>
<FONT color="green">220</FONT>    <a name="line.220"></a>
<FONT color="green">221</FONT>        LOG.info("Dump debug output");<a name="line.221"></a>
<FONT color="green">222</FONT>        Map&lt;String, String&gt; envs = System.getenv();<a name="line.222"></a>
<FONT color="green">223</FONT>        for (Map.Entry&lt;String, String&gt; env : envs.entrySet()) {<a name="line.223"></a>
<FONT color="green">224</FONT>          LOG.info("System env: key=" + env.getKey() + ", val=" + env.getValue());<a name="line.224"></a>
<FONT color="green">225</FONT>          System.out.println("System env: key=" + env.getKey() + ", val=" + env.getValue());<a name="line.225"></a>
<FONT color="green">226</FONT>        }<a name="line.226"></a>
<FONT color="green">227</FONT>    <a name="line.227"></a>
<FONT color="green">228</FONT>        String cmd = "ls -al";<a name="line.228"></a>
<FONT color="green">229</FONT>        Runtime run = Runtime.getRuntime();<a name="line.229"></a>
<FONT color="green">230</FONT>        Process pr = null;<a name="line.230"></a>
<FONT color="green">231</FONT>        try {<a name="line.231"></a>
<FONT color="green">232</FONT>          pr = run.exec(cmd);<a name="line.232"></a>
<FONT color="green">233</FONT>          pr.waitFor();<a name="line.233"></a>
<FONT color="green">234</FONT>    <a name="line.234"></a>
<FONT color="green">235</FONT>          BufferedReader buf = new BufferedReader(new InputStreamReader(pr.getInputStream()));<a name="line.235"></a>
<FONT color="green">236</FONT>          String line = "";<a name="line.236"></a>
<FONT color="green">237</FONT>          while ((line=buf.readLine())!=null) {<a name="line.237"></a>
<FONT color="green">238</FONT>            LOG.info("System CWD content: " + line);<a name="line.238"></a>
<FONT color="green">239</FONT>            System.out.println("System CWD content: " + line);<a name="line.239"></a>
<FONT color="green">240</FONT>          }<a name="line.240"></a>
<FONT color="green">241</FONT>          buf.close();<a name="line.241"></a>
<FONT color="green">242</FONT>        } catch (IOException e) {<a name="line.242"></a>
<FONT color="green">243</FONT>          e.printStackTrace();<a name="line.243"></a>
<FONT color="green">244</FONT>        } catch (InterruptedException e) {<a name="line.244"></a>
<FONT color="green">245</FONT>          e.printStackTrace();<a name="line.245"></a>
<FONT color="green">246</FONT>        } <a name="line.246"></a>
<FONT color="green">247</FONT>      }<a name="line.247"></a>
<FONT color="green">248</FONT>    <a name="line.248"></a>
<FONT color="green">249</FONT>      public ApplicationMaster() throws Exception {<a name="line.249"></a>
<FONT color="green">250</FONT>        // Set up the configuration and RPC<a name="line.250"></a>
<FONT color="green">251</FONT>        conf = new Configuration();<a name="line.251"></a>
<FONT color="green">252</FONT>        rpc = YarnRPC.create(conf);<a name="line.252"></a>
<FONT color="green">253</FONT>      }<a name="line.253"></a>
<FONT color="green">254</FONT>      /**<a name="line.254"></a>
<FONT color="green">255</FONT>       * Parse command line options<a name="line.255"></a>
<FONT color="green">256</FONT>       * @param args Command line args <a name="line.256"></a>
<FONT color="green">257</FONT>       * @return Whether init successful and run should be invoked <a name="line.257"></a>
<FONT color="green">258</FONT>       * @throws ParseException<a name="line.258"></a>
<FONT color="green">259</FONT>       * @throws IOException <a name="line.259"></a>
<FONT color="green">260</FONT>       */<a name="line.260"></a>
<FONT color="green">261</FONT>      public boolean init(String[] args) throws ParseException, IOException {<a name="line.261"></a>
<FONT color="green">262</FONT>    <a name="line.262"></a>
<FONT color="green">263</FONT>        Options opts = new Options();<a name="line.263"></a>
<FONT color="green">264</FONT>        opts.addOption("app_attempt_id", true, "App Attempt ID. Not to be used unless for testing purposes");<a name="line.264"></a>
<FONT color="green">265</FONT>        opts.addOption("shell_command", true, "Shell command to be executed by the Application Master");<a name="line.265"></a>
<FONT color="green">266</FONT>        opts.addOption("shell_script", true, "Location of the shell script to be executed");<a name="line.266"></a>
<FONT color="green">267</FONT>        opts.addOption("shell_args", true, "Command line args for the shell script");<a name="line.267"></a>
<FONT color="green">268</FONT>        opts.addOption("shell_env", true, "Environment for shell script. Specified as env_key=env_val pairs");<a name="line.268"></a>
<FONT color="green">269</FONT>        opts.addOption("container_memory", true, "Amount of memory in MB to be requested to run the shell command");<a name="line.269"></a>
<FONT color="green">270</FONT>        opts.addOption("num_containers", true, "No. of containers on which the shell command needs to be executed");<a name="line.270"></a>
<FONT color="green">271</FONT>        opts.addOption("priority", true, "Application Priority. Default 0");<a name="line.271"></a>
<FONT color="green">272</FONT>        opts.addOption("debug", false, "Dump out debug information");<a name="line.272"></a>
<FONT color="green">273</FONT>    <a name="line.273"></a>
<FONT color="green">274</FONT>        opts.addOption("help", false, "Print usage");<a name="line.274"></a>
<FONT color="green">275</FONT>        CommandLine cliParser = new GnuParser().parse(opts, args);<a name="line.275"></a>
<FONT color="green">276</FONT>    <a name="line.276"></a>
<FONT color="green">277</FONT>        if (args.length == 0) {<a name="line.277"></a>
<FONT color="green">278</FONT>          printUsage(opts);<a name="line.278"></a>
<FONT color="green">279</FONT>          throw new IllegalArgumentException("No args specified for application master to initialize");<a name="line.279"></a>
<FONT color="green">280</FONT>        }<a name="line.280"></a>
<FONT color="green">281</FONT>    <a name="line.281"></a>
<FONT color="green">282</FONT>        if (cliParser.hasOption("help")) {<a name="line.282"></a>
<FONT color="green">283</FONT>          printUsage(opts);<a name="line.283"></a>
<FONT color="green">284</FONT>          return false;<a name="line.284"></a>
<FONT color="green">285</FONT>        }<a name="line.285"></a>
<FONT color="green">286</FONT>    <a name="line.286"></a>
<FONT color="green">287</FONT>        if (cliParser.hasOption("debug")) {<a name="line.287"></a>
<FONT color="green">288</FONT>          dumpOutDebugInfo();<a name="line.288"></a>
<FONT color="green">289</FONT>        }<a name="line.289"></a>
<FONT color="green">290</FONT>    <a name="line.290"></a>
<FONT color="green">291</FONT>        Map&lt;String, String&gt; envs = System.getenv();<a name="line.291"></a>
<FONT color="green">292</FONT>    <a name="line.292"></a>
<FONT color="green">293</FONT>        appAttemptID = Records.newRecord(ApplicationAttemptId.class);<a name="line.293"></a>
<FONT color="green">294</FONT>        if (!envs.containsKey(ApplicationConstants.AM_CONTAINER_ID_ENV)) {<a name="line.294"></a>
<FONT color="green">295</FONT>          if (cliParser.hasOption("app_attempt_id")) {<a name="line.295"></a>
<FONT color="green">296</FONT>            String appIdStr = cliParser.getOptionValue("app_attempt_id", "");<a name="line.296"></a>
<FONT color="green">297</FONT>            appAttemptID = ConverterUtils.toApplicationAttemptId(appIdStr);<a name="line.297"></a>
<FONT color="green">298</FONT>          } <a name="line.298"></a>
<FONT color="green">299</FONT>          else {<a name="line.299"></a>
<FONT color="green">300</FONT>            throw new IllegalArgumentException("Application Attempt Id not set in the environment");<a name="line.300"></a>
<FONT color="green">301</FONT>          }<a name="line.301"></a>
<FONT color="green">302</FONT>        } else {<a name="line.302"></a>
<FONT color="green">303</FONT>          ContainerId containerId = ConverterUtils.toContainerId(envs.get(ApplicationConstants.AM_CONTAINER_ID_ENV));<a name="line.303"></a>
<FONT color="green">304</FONT>          appAttemptID = containerId.getApplicationAttemptId();<a name="line.304"></a>
<FONT color="green">305</FONT>        }<a name="line.305"></a>
<FONT color="green">306</FONT>    <a name="line.306"></a>
<FONT color="green">307</FONT>        LOG.info("Application master for app"<a name="line.307"></a>
<FONT color="green">308</FONT>            + ", appId=" + appAttemptID.getApplicationId().getId()<a name="line.308"></a>
<FONT color="green">309</FONT>            + ", clustertimestamp=" + appAttemptID.getApplicationId().getClusterTimestamp()<a name="line.309"></a>
<FONT color="green">310</FONT>            + ", attemptId=" + appAttemptID.getAttemptId());<a name="line.310"></a>
<FONT color="green">311</FONT>    <a name="line.311"></a>
<FONT color="green">312</FONT>        if (!cliParser.hasOption("shell_command")) {<a name="line.312"></a>
<FONT color="green">313</FONT>          throw new IllegalArgumentException("No shell command specified to be executed by application master");<a name="line.313"></a>
<FONT color="green">314</FONT>        }<a name="line.314"></a>
<FONT color="green">315</FONT>        shellCommand = cliParser.getOptionValue("shell_command");<a name="line.315"></a>
<FONT color="green">316</FONT>    <a name="line.316"></a>
<FONT color="green">317</FONT>        if (cliParser.hasOption("shell_args")) {<a name="line.317"></a>
<FONT color="green">318</FONT>          shellArgs = cliParser.getOptionValue("shell_args");<a name="line.318"></a>
<FONT color="green">319</FONT>        }<a name="line.319"></a>
<FONT color="green">320</FONT>        if (cliParser.hasOption("shell_env")) { <a name="line.320"></a>
<FONT color="green">321</FONT>          String shellEnvs[] = cliParser.getOptionValues("shell_env");<a name="line.321"></a>
<FONT color="green">322</FONT>          for (String env : shellEnvs) {<a name="line.322"></a>
<FONT color="green">323</FONT>            env = env.trim();<a name="line.323"></a>
<FONT color="green">324</FONT>            int index = env.indexOf('=');<a name="line.324"></a>
<FONT color="green">325</FONT>            if (index == -1) {<a name="line.325"></a>
<FONT color="green">326</FONT>              shellEnv.put(env, "");<a name="line.326"></a>
<FONT color="green">327</FONT>              continue;<a name="line.327"></a>
<FONT color="green">328</FONT>            }<a name="line.328"></a>
<FONT color="green">329</FONT>            String key = env.substring(0, index);<a name="line.329"></a>
<FONT color="green">330</FONT>            String val = "";<a name="line.330"></a>
<FONT color="green">331</FONT>            if (index &lt; (env.length()-1)) {<a name="line.331"></a>
<FONT color="green">332</FONT>              val = env.substring(index+1);<a name="line.332"></a>
<FONT color="green">333</FONT>            }<a name="line.333"></a>
<FONT color="green">334</FONT>            shellEnv.put(key, val);<a name="line.334"></a>
<FONT color="green">335</FONT>          }<a name="line.335"></a>
<FONT color="green">336</FONT>        }<a name="line.336"></a>
<FONT color="green">337</FONT>    <a name="line.337"></a>
<FONT color="green">338</FONT>        if (envs.containsKey(DSConstants.DISTRIBUTEDSHELLSCRIPTLOCATION)) {<a name="line.338"></a>
<FONT color="green">339</FONT>          shellScriptPath = envs.get(DSConstants.DISTRIBUTEDSHELLSCRIPTLOCATION);<a name="line.339"></a>
<FONT color="green">340</FONT>    <a name="line.340"></a>
<FONT color="green">341</FONT>          if (envs.containsKey(DSConstants.DISTRIBUTEDSHELLSCRIPTTIMESTAMP)) {<a name="line.341"></a>
<FONT color="green">342</FONT>            shellScriptPathTimestamp = Long.valueOf(envs.get(DSConstants.DISTRIBUTEDSHELLSCRIPTTIMESTAMP));<a name="line.342"></a>
<FONT color="green">343</FONT>          }<a name="line.343"></a>
<FONT color="green">344</FONT>          if (envs.containsKey(DSConstants.DISTRIBUTEDSHELLSCRIPTLEN)) {<a name="line.344"></a>
<FONT color="green">345</FONT>            shellScriptPathLen = Long.valueOf(envs.get(DSConstants.DISTRIBUTEDSHELLSCRIPTLEN));<a name="line.345"></a>
<FONT color="green">346</FONT>          }<a name="line.346"></a>
<FONT color="green">347</FONT>    <a name="line.347"></a>
<FONT color="green">348</FONT>          if (!shellScriptPath.isEmpty()<a name="line.348"></a>
<FONT color="green">349</FONT>              &amp;&amp; (shellScriptPathTimestamp &lt;= 0 <a name="line.349"></a>
<FONT color="green">350</FONT>              || shellScriptPathLen &lt;= 0)) {<a name="line.350"></a>
<FONT color="green">351</FONT>            LOG.error("Illegal values in env for shell script path"<a name="line.351"></a>
<FONT color="green">352</FONT>                + ", path=" + shellScriptPath<a name="line.352"></a>
<FONT color="green">353</FONT>                + ", len=" + shellScriptPathLen<a name="line.353"></a>
<FONT color="green">354</FONT>                + ", timestamp=" + shellScriptPathTimestamp);<a name="line.354"></a>
<FONT color="green">355</FONT>            throw new IllegalArgumentException("Illegal values in env for shell script path");<a name="line.355"></a>
<FONT color="green">356</FONT>          }<a name="line.356"></a>
<FONT color="green">357</FONT>        }<a name="line.357"></a>
<FONT color="green">358</FONT>    <a name="line.358"></a>
<FONT color="green">359</FONT>        containerMemory = Integer.parseInt(cliParser.getOptionValue("container_memory", "10"));<a name="line.359"></a>
<FONT color="green">360</FONT>        numTotalContainers = Integer.parseInt(cliParser.getOptionValue("num_containers", "1"));<a name="line.360"></a>
<FONT color="green">361</FONT>        requestPriority = Integer.parseInt(cliParser.getOptionValue("priority", "0"));<a name="line.361"></a>
<FONT color="green">362</FONT>    <a name="line.362"></a>
<FONT color="green">363</FONT>        return true;<a name="line.363"></a>
<FONT color="green">364</FONT>      }<a name="line.364"></a>
<FONT color="green">365</FONT>    <a name="line.365"></a>
<FONT color="green">366</FONT>      /**<a name="line.366"></a>
<FONT color="green">367</FONT>       * Helper function to print usage <a name="line.367"></a>
<FONT color="green">368</FONT>       * @param opts Parsed command line options<a name="line.368"></a>
<FONT color="green">369</FONT>       */<a name="line.369"></a>
<FONT color="green">370</FONT>      private void printUsage(Options opts) {<a name="line.370"></a>
<FONT color="green">371</FONT>        new HelpFormatter().printHelp("ApplicationMaster", opts);<a name="line.371"></a>
<FONT color="green">372</FONT>      }<a name="line.372"></a>
<FONT color="green">373</FONT>    <a name="line.373"></a>
<FONT color="green">374</FONT>      /**<a name="line.374"></a>
<FONT color="green">375</FONT>       * Main run function for the application master<a name="line.375"></a>
<FONT color="green">376</FONT>       * @throws YarnRemoteException<a name="line.376"></a>
<FONT color="green">377</FONT>       */<a name="line.377"></a>
<FONT color="green">378</FONT>      public boolean run() throws YarnRemoteException {<a name="line.378"></a>
<FONT color="green">379</FONT>        LOG.info("Starting ApplicationMaster");<a name="line.379"></a>
<FONT color="green">380</FONT>    <a name="line.380"></a>
<FONT color="green">381</FONT>        // Connect to ResourceManager<a name="line.381"></a>
<FONT color="green">382</FONT>        resourceManager = connectToRM();<a name="line.382"></a>
<FONT color="green">383</FONT>    <a name="line.383"></a>
<FONT color="green">384</FONT>        // Setup local RPC Server to accept status requests directly from clients <a name="line.384"></a>
<FONT color="green">385</FONT>        // TODO need to setup a protocol for client to be able to communicate to the RPC server <a name="line.385"></a>
<FONT color="green">386</FONT>        // TODO use the rpc port info to register with the RM for the client to send requests to this app master<a name="line.386"></a>
<FONT color="green">387</FONT>    <a name="line.387"></a>
<FONT color="green">388</FONT>        // Register self with ResourceManager <a name="line.388"></a>
<FONT color="green">389</FONT>        RegisterApplicationMasterResponse response = registerToRM();<a name="line.389"></a>
<FONT color="green">390</FONT>        // Dump out information about cluster capability as seen by the resource manager<a name="line.390"></a>
<FONT color="green">391</FONT>        int minMem = response.getMinimumResourceCapability().getMemory();<a name="line.391"></a>
<FONT color="green">392</FONT>        int maxMem = response.getMaximumResourceCapability().getMemory();<a name="line.392"></a>
<FONT color="green">393</FONT>        LOG.info("Min mem capabililty of resources in this cluster " + minMem);<a name="line.393"></a>
<FONT color="green">394</FONT>        LOG.info("Max mem capabililty of resources in this cluster " + maxMem);<a name="line.394"></a>
<FONT color="green">395</FONT>    <a name="line.395"></a>
<FONT color="green">396</FONT>        // A resource ask has to be atleast the minimum of the capability of the cluster, the value has to be <a name="line.396"></a>
<FONT color="green">397</FONT>        // a multiple of the min value and cannot exceed the max. <a name="line.397"></a>
<FONT color="green">398</FONT>        // If it is not an exact multiple of min, the RM will allocate to the nearest multiple of min<a name="line.398"></a>
<FONT color="green">399</FONT>        if (containerMemory &lt; minMem) {<a name="line.399"></a>
<FONT color="green">400</FONT>          LOG.info("Container memory specified below min threshold of cluster. Using min value."<a name="line.400"></a>
<FONT color="green">401</FONT>              + ", specified=" + containerMemory<a name="line.401"></a>
<FONT color="green">402</FONT>              + ", min=" + minMem);<a name="line.402"></a>
<FONT color="green">403</FONT>          containerMemory = minMem; <a name="line.403"></a>
<FONT color="green">404</FONT>        } <a name="line.404"></a>
<FONT color="green">405</FONT>        else if (containerMemory &gt; maxMem) {<a name="line.405"></a>
<FONT color="green">406</FONT>          LOG.info("Container memory specified above max threshold of cluster. Using max value."<a name="line.406"></a>
<FONT color="green">407</FONT>              + ", specified=" + containerMemory<a name="line.407"></a>
<FONT color="green">408</FONT>              + ", max=" + maxMem);<a name="line.408"></a>
<FONT color="green">409</FONT>          containerMemory = maxMem;<a name="line.409"></a>
<FONT color="green">410</FONT>        }<a name="line.410"></a>
<FONT color="green">411</FONT>    <a name="line.411"></a>
<FONT color="green">412</FONT>        // Setup heartbeat emitter<a name="line.412"></a>
<FONT color="green">413</FONT>        // TODO poll RM every now and then with an empty request to let RM know that we are alive<a name="line.413"></a>
<FONT color="green">414</FONT>        // The heartbeat interval after which an AM is timed out by the RM is defined by a config setting: <a name="line.414"></a>
<FONT color="green">415</FONT>        // RM_AM_EXPIRY_INTERVAL_MS with default defined by DEFAULT_RM_AM_EXPIRY_INTERVAL_MS<a name="line.415"></a>
<FONT color="green">416</FONT>        // The allocate calls to the RM count as heartbeats so, for now, this additional heartbeat emitter <a name="line.416"></a>
<FONT color="green">417</FONT>        // is not required.<a name="line.417"></a>
<FONT color="green">418</FONT>    <a name="line.418"></a>
<FONT color="green">419</FONT>        // Setup ask for containers from RM<a name="line.419"></a>
<FONT color="green">420</FONT>        // Send request for containers to RM<a name="line.420"></a>
<FONT color="green">421</FONT>        // Until we get our fully allocated quota, we keep on polling RM for containers<a name="line.421"></a>
<FONT color="green">422</FONT>        // Keep looping until all the containers are launched and shell script executed on them <a name="line.422"></a>
<FONT color="green">423</FONT>        // ( regardless of success/failure). <a name="line.423"></a>
<FONT color="green">424</FONT>    <a name="line.424"></a>
<FONT color="green">425</FONT>        int loopCounter = -1;<a name="line.425"></a>
<FONT color="green">426</FONT>    <a name="line.426"></a>
<FONT color="green">427</FONT>        while (numCompletedContainers.get() &lt; numTotalContainers<a name="line.427"></a>
<FONT color="green">428</FONT>            &amp;&amp; !appDone) {<a name="line.428"></a>
<FONT color="green">429</FONT>          loopCounter++;<a name="line.429"></a>
<FONT color="green">430</FONT>    <a name="line.430"></a>
<FONT color="green">431</FONT>          // log current state<a name="line.431"></a>
<FONT color="green">432</FONT>          LOG.info("Current application state: loop=" + loopCounter <a name="line.432"></a>
<FONT color="green">433</FONT>              + ", appDone=" + appDone<a name="line.433"></a>
<FONT color="green">434</FONT>              + ", total=" + numTotalContainers<a name="line.434"></a>
<FONT color="green">435</FONT>              + ", requested=" + numRequestedContainers<a name="line.435"></a>
<FONT color="green">436</FONT>              + ", completed=" + numCompletedContainers<a name="line.436"></a>
<FONT color="green">437</FONT>              + ", failed=" + numFailedContainers<a name="line.437"></a>
<FONT color="green">438</FONT>              + ", currentAllocated=" + numAllocatedContainers);<a name="line.438"></a>
<FONT color="green">439</FONT>    <a name="line.439"></a>
<FONT color="green">440</FONT>          // Sleep before each loop when asking RM for containers<a name="line.440"></a>
<FONT color="green">441</FONT>          // to avoid flooding RM with spurious requests when it <a name="line.441"></a>
<FONT color="green">442</FONT>          // need not have any available containers <a name="line.442"></a>
<FONT color="green">443</FONT>          // Sleeping for 1000 ms.<a name="line.443"></a>
<FONT color="green">444</FONT>          try {<a name="line.444"></a>
<FONT color="green">445</FONT>            Thread.sleep(1000);<a name="line.445"></a>
<FONT color="green">446</FONT>          } catch (InterruptedException e) {<a name="line.446"></a>
<FONT color="green">447</FONT>            LOG.info("Sleep interrupted " + e.getMessage());<a name="line.447"></a>
<FONT color="green">448</FONT>          }<a name="line.448"></a>
<FONT color="green">449</FONT>    <a name="line.449"></a>
<FONT color="green">450</FONT>          // No. of containers to request <a name="line.450"></a>
<FONT color="green">451</FONT>          // For the first loop, askCount will be equal to total containers needed <a name="line.451"></a>
<FONT color="green">452</FONT>          // From that point on, askCount will always be 0 as current implementation <a name="line.452"></a>
<FONT color="green">453</FONT>          // does not change its ask on container failures. <a name="line.453"></a>
<FONT color="green">454</FONT>          int askCount = numTotalContainers - numRequestedContainers.get();<a name="line.454"></a>
<FONT color="green">455</FONT>          numRequestedContainers.addAndGet(askCount);<a name="line.455"></a>
<FONT color="green">456</FONT>    <a name="line.456"></a>
<FONT color="green">457</FONT>          // Setup request to be sent to RM to allocate containers<a name="line.457"></a>
<FONT color="green">458</FONT>          List&lt;ResourceRequest&gt; resourceReq = new ArrayList&lt;ResourceRequest&gt;();<a name="line.458"></a>
<FONT color="green">459</FONT>          if (askCount &gt; 0) {<a name="line.459"></a>
<FONT color="green">460</FONT>            ResourceRequest containerAsk = setupContainerAskForRM(askCount);<a name="line.460"></a>
<FONT color="green">461</FONT>            resourceReq.add(containerAsk);<a name="line.461"></a>
<FONT color="green">462</FONT>          }<a name="line.462"></a>
<FONT color="green">463</FONT>    <a name="line.463"></a>
<FONT color="green">464</FONT>          // Send the request to RM <a name="line.464"></a>
<FONT color="green">465</FONT>          LOG.info("Asking RM for containers"<a name="line.465"></a>
<FONT color="green">466</FONT>              + ", askCount=" + askCount);<a name="line.466"></a>
<FONT color="green">467</FONT>          AMResponse amResp =sendContainerAskToRM(resourceReq);<a name="line.467"></a>
<FONT color="green">468</FONT>    <a name="line.468"></a>
<FONT color="green">469</FONT>          // Retrieve list of allocated containers from the response <a name="line.469"></a>
<FONT color="green">470</FONT>          List&lt;Container&gt; allocatedContainers = amResp.getAllocatedContainers();<a name="line.470"></a>
<FONT color="green">471</FONT>          LOG.info("Got response from RM for container ask, allocatedCnt=" + allocatedContainers.size());<a name="line.471"></a>
<FONT color="green">472</FONT>          numAllocatedContainers.addAndGet(allocatedContainers.size());<a name="line.472"></a>
<FONT color="green">473</FONT>          for (Container allocatedContainer : allocatedContainers) {<a name="line.473"></a>
<FONT color="green">474</FONT>            LOG.info("Launching shell command on a new container."<a name="line.474"></a>
<FONT color="green">475</FONT>                + ", containerId=" + allocatedContainer.getId()<a name="line.475"></a>
<FONT color="green">476</FONT>                + ", containerNode=" + allocatedContainer.getNodeId().getHost() <a name="line.476"></a>
<FONT color="green">477</FONT>                + ":" + allocatedContainer.getNodeId().getPort()<a name="line.477"></a>
<FONT color="green">478</FONT>                + ", containerNodeURI=" + allocatedContainer.getNodeHttpAddress()<a name="line.478"></a>
<FONT color="green">479</FONT>                + ", containerState" + allocatedContainer.getState()<a name="line.479"></a>
<FONT color="green">480</FONT>                + ", containerResourceMemory" + allocatedContainer.getResource().getMemory());<a name="line.480"></a>
<FONT color="green">481</FONT>            //+ ", containerToken" + allocatedContainer.getContainerToken().getIdentifier().toString());<a name="line.481"></a>
<FONT color="green">482</FONT>    <a name="line.482"></a>
<FONT color="green">483</FONT>            LaunchContainerRunnable runnableLaunchContainer = new LaunchContainerRunnable(allocatedContainer);<a name="line.483"></a>
<FONT color="green">484</FONT>            Thread launchThread = new Thread(runnableLaunchContainer);<a name="line.484"></a>
<FONT color="green">485</FONT>    <a name="line.485"></a>
<FONT color="green">486</FONT>            // launch and start the container on a separate thread to keep the main thread unblocked<a name="line.486"></a>
<FONT color="green">487</FONT>            // as all containers may not be allocated at one go.<a name="line.487"></a>
<FONT color="green">488</FONT>            launchThreads.add(launchThread);<a name="line.488"></a>
<FONT color="green">489</FONT>            launchThread.start();<a name="line.489"></a>
<FONT color="green">490</FONT>          }<a name="line.490"></a>
<FONT color="green">491</FONT>    <a name="line.491"></a>
<FONT color="green">492</FONT>          // Check what the current available resources in the cluster are<a name="line.492"></a>
<FONT color="green">493</FONT>          // TODO should we do anything if the available resources are not enough? <a name="line.493"></a>
<FONT color="green">494</FONT>          Resource availableResources = amResp.getAvailableResources();<a name="line.494"></a>
<FONT color="green">495</FONT>          LOG.info("Current available resources in the cluster " + availableResources);<a name="line.495"></a>
<FONT color="green">496</FONT>    <a name="line.496"></a>
<FONT color="green">497</FONT>          // Check the completed containers<a name="line.497"></a>
<FONT color="green">498</FONT>          List&lt;ContainerStatus&gt; completedContainers = amResp.getCompletedContainersStatuses();<a name="line.498"></a>
<FONT color="green">499</FONT>          LOG.info("Got response from RM for container ask, completedCnt=" + completedContainers.size());<a name="line.499"></a>
<FONT color="green">500</FONT>          for (ContainerStatus containerStatus : completedContainers) {<a name="line.500"></a>
<FONT color="green">501</FONT>            LOG.info("Got container status for containerID= " + containerStatus.getContainerId()<a name="line.501"></a>
<FONT color="green">502</FONT>                + ", state=" + containerStatus.getState()<a name="line.502"></a>
<FONT color="green">503</FONT>                + ", exitStatus=" + containerStatus.getExitStatus() <a name="line.503"></a>
<FONT color="green">504</FONT>                + ", diagnostics=" + containerStatus.getDiagnostics());<a name="line.504"></a>
<FONT color="green">505</FONT>    <a name="line.505"></a>
<FONT color="green">506</FONT>            // non complete containers should not be here <a name="line.506"></a>
<FONT color="green">507</FONT>            assert(containerStatus.getState() == ContainerState.COMPLETE);<a name="line.507"></a>
<FONT color="green">508</FONT>    <a name="line.508"></a>
<FONT color="green">509</FONT>            // increment counters for completed/failed containers<a name="line.509"></a>
<FONT color="green">510</FONT>            int exitStatus = containerStatus.getExitStatus();<a name="line.510"></a>
<FONT color="green">511</FONT>            if (0 != exitStatus) {<a name="line.511"></a>
<FONT color="green">512</FONT>              // container failed <a name="line.512"></a>
<FONT color="green">513</FONT>              if (-100 != exitStatus) {<a name="line.513"></a>
<FONT color="green">514</FONT>                // shell script failed<a name="line.514"></a>
<FONT color="green">515</FONT>                // counts as completed <a name="line.515"></a>
<FONT color="green">516</FONT>                numCompletedContainers.incrementAndGet();<a name="line.516"></a>
<FONT color="green">517</FONT>                numFailedContainers.incrementAndGet();<a name="line.517"></a>
<FONT color="green">518</FONT>              }<a name="line.518"></a>
<FONT color="green">519</FONT>              else { <a name="line.519"></a>
<FONT color="green">520</FONT>                // something else bad happened <a name="line.520"></a>
<FONT color="green">521</FONT>                // app job did not complete for some reason <a name="line.521"></a>
<FONT color="green">522</FONT>                // we should re-try as the container was lost for some reason<a name="line.522"></a>
<FONT color="green">523</FONT>                numAllocatedContainers.decrementAndGet();<a name="line.523"></a>
<FONT color="green">524</FONT>                numRequestedContainers.decrementAndGet();<a name="line.524"></a>
<FONT color="green">525</FONT>                // we do not need to release the container as it would be done<a name="line.525"></a>
<FONT color="green">526</FONT>                // by the RM/CM.<a name="line.526"></a>
<FONT color="green">527</FONT>              }<a name="line.527"></a>
<FONT color="green">528</FONT>            }<a name="line.528"></a>
<FONT color="green">529</FONT>            else { <a name="line.529"></a>
<FONT color="green">530</FONT>              // nothing to do <a name="line.530"></a>
<FONT color="green">531</FONT>              // container completed successfully <a name="line.531"></a>
<FONT color="green">532</FONT>              numCompletedContainers.incrementAndGet();<a name="line.532"></a>
<FONT color="green">533</FONT>              LOG.info("Container completed successfully."<a name="line.533"></a>
<FONT color="green">534</FONT>                  + ", containerId=" + containerStatus.getContainerId());<a name="line.534"></a>
<FONT color="green">535</FONT>            }<a name="line.535"></a>
<FONT color="green">536</FONT>    <a name="line.536"></a>
<FONT color="green">537</FONT>          }<a name="line.537"></a>
<FONT color="green">538</FONT>          if (numCompletedContainers.get() == numTotalContainers) {<a name="line.538"></a>
<FONT color="green">539</FONT>            appDone = true;<a name="line.539"></a>
<FONT color="green">540</FONT>          }<a name="line.540"></a>
<FONT color="green">541</FONT>    <a name="line.541"></a>
<FONT color="green">542</FONT>          LOG.info("Current application state: loop=" + loopCounter<a name="line.542"></a>
<FONT color="green">543</FONT>              + ", appDone=" + appDone<a name="line.543"></a>
<FONT color="green">544</FONT>              + ", total=" + numTotalContainers<a name="line.544"></a>
<FONT color="green">545</FONT>              + ", requested=" + numRequestedContainers<a name="line.545"></a>
<FONT color="green">546</FONT>              + ", completed=" + numCompletedContainers<a name="line.546"></a>
<FONT color="green">547</FONT>              + ", failed=" + numFailedContainers<a name="line.547"></a>
<FONT color="green">548</FONT>              + ", currentAllocated=" + numAllocatedContainers);<a name="line.548"></a>
<FONT color="green">549</FONT>    <a name="line.549"></a>
<FONT color="green">550</FONT>          // TODO <a name="line.550"></a>
<FONT color="green">551</FONT>          // Add a timeout handling layer <a name="line.551"></a>
<FONT color="green">552</FONT>          // for misbehaving shell commands<a name="line.552"></a>
<FONT color="green">553</FONT>        }<a name="line.553"></a>
<FONT color="green">554</FONT>    <a name="line.554"></a>
<FONT color="green">555</FONT>        // Join all launched threads<a name="line.555"></a>
<FONT color="green">556</FONT>        // needed for when we time out <a name="line.556"></a>
<FONT color="green">557</FONT>        // and we need to release containers<a name="line.557"></a>
<FONT color="green">558</FONT>        for (Thread launchThread : launchThreads) {<a name="line.558"></a>
<FONT color="green">559</FONT>          try {<a name="line.559"></a>
<FONT color="green">560</FONT>            launchThread.join(10000);<a name="line.560"></a>
<FONT color="green">561</FONT>          } catch (InterruptedException e) {<a name="line.561"></a>
<FONT color="green">562</FONT>            LOG.info("Exception thrown in thread join: " + e.getMessage());<a name="line.562"></a>
<FONT color="green">563</FONT>            e.printStackTrace();<a name="line.563"></a>
<FONT color="green">564</FONT>          }<a name="line.564"></a>
<FONT color="green">565</FONT>        }<a name="line.565"></a>
<FONT color="green">566</FONT>    <a name="line.566"></a>
<FONT color="green">567</FONT>        // When the application completes, it should send a finish application signal <a name="line.567"></a>
<FONT color="green">568</FONT>        // to the RM<a name="line.568"></a>
<FONT color="green">569</FONT>        LOG.info("Application completed. Signalling finish to RM");<a name="line.569"></a>
<FONT color="green">570</FONT>    <a name="line.570"></a>
<FONT color="green">571</FONT>        FinishApplicationMasterRequest finishReq = Records.newRecord(FinishApplicationMasterRequest.class);<a name="line.571"></a>
<FONT color="green">572</FONT>        finishReq.setAppAttemptId(appAttemptID);<a name="line.572"></a>
<FONT color="green">573</FONT>        boolean isSuccess = true;<a name="line.573"></a>
<FONT color="green">574</FONT>        if (numFailedContainers.get() == 0) {<a name="line.574"></a>
<FONT color="green">575</FONT>          finishReq.setFinishApplicationStatus(FinalApplicationStatus.SUCCEEDED);<a name="line.575"></a>
<FONT color="green">576</FONT>        }<a name="line.576"></a>
<FONT color="green">577</FONT>        else {<a name="line.577"></a>
<FONT color="green">578</FONT>          finishReq.setFinishApplicationStatus(FinalApplicationStatus.FAILED);<a name="line.578"></a>
<FONT color="green">579</FONT>          String diagnostics = "Diagnostics."<a name="line.579"></a>
<FONT color="green">580</FONT>              + ", total=" + numTotalContainers<a name="line.580"></a>
<FONT color="green">581</FONT>              + ", completed=" + numCompletedContainers.get()<a name="line.581"></a>
<FONT color="green">582</FONT>              + ", allocated=" + numAllocatedContainers.get()<a name="line.582"></a>
<FONT color="green">583</FONT>              + ", failed=" + numFailedContainers.get();<a name="line.583"></a>
<FONT color="green">584</FONT>          finishReq.setDiagnostics(diagnostics);<a name="line.584"></a>
<FONT color="green">585</FONT>          isSuccess = false;<a name="line.585"></a>
<FONT color="green">586</FONT>        }<a name="line.586"></a>
<FONT color="green">587</FONT>        resourceManager.finishApplicationMaster(finishReq);<a name="line.587"></a>
<FONT color="green">588</FONT>        return isSuccess;<a name="line.588"></a>
<FONT color="green">589</FONT>      }<a name="line.589"></a>
<FONT color="green">590</FONT>    <a name="line.590"></a>
<FONT color="green">591</FONT>      /**<a name="line.591"></a>
<FONT color="green">592</FONT>       * Thread to connect to the {@link ContainerManager} and <a name="line.592"></a>
<FONT color="green">593</FONT>       * launch the container that will execute the shell command. <a name="line.593"></a>
<FONT color="green">594</FONT>       */<a name="line.594"></a>
<FONT color="green">595</FONT>      private class LaunchContainerRunnable implements Runnable {<a name="line.595"></a>
<FONT color="green">596</FONT>    <a name="line.596"></a>
<FONT color="green">597</FONT>        // Allocated container <a name="line.597"></a>
<FONT color="green">598</FONT>        Container container;<a name="line.598"></a>
<FONT color="green">599</FONT>        // Handle to communicate with ContainerManager<a name="line.599"></a>
<FONT color="green">600</FONT>        ContainerManager cm;<a name="line.600"></a>
<FONT color="green">601</FONT>    <a name="line.601"></a>
<FONT color="green">602</FONT>        /**<a name="line.602"></a>
<FONT color="green">603</FONT>         * @param lcontainer Allocated container<a name="line.603"></a>
<FONT color="green">604</FONT>         */<a name="line.604"></a>
<FONT color="green">605</FONT>        public LaunchContainerRunnable(Container lcontainer) {<a name="line.605"></a>
<FONT color="green">606</FONT>          this.container = lcontainer;<a name="line.606"></a>
<FONT color="green">607</FONT>        }<a name="line.607"></a>
<FONT color="green">608</FONT>    <a name="line.608"></a>
<FONT color="green">609</FONT>        /**<a name="line.609"></a>
<FONT color="green">610</FONT>         * Helper function to connect to CM<a name="line.610"></a>
<FONT color="green">611</FONT>         */<a name="line.611"></a>
<FONT color="green">612</FONT>        private void connectToCM() {<a name="line.612"></a>
<FONT color="green">613</FONT>          LOG.debug("Connecting to ContainerManager for containerid=" + container.getId());<a name="line.613"></a>
<FONT color="green">614</FONT>          String cmIpPortStr = container.getNodeId().getHost() + ":"<a name="line.614"></a>
<FONT color="green">615</FONT>              + container.getNodeId().getPort();<a name="line.615"></a>
<FONT color="green">616</FONT>          InetSocketAddress cmAddress = NetUtils.createSocketAddr(cmIpPortStr);<a name="line.616"></a>
<FONT color="green">617</FONT>          LOG.info("Connecting to ContainerManager at " + cmIpPortStr);<a name="line.617"></a>
<FONT color="green">618</FONT>          this.cm = ((ContainerManager) rpc.getProxy(ContainerManager.class, cmAddress, conf));<a name="line.618"></a>
<FONT color="green">619</FONT>        }<a name="line.619"></a>
<FONT color="green">620</FONT>    <a name="line.620"></a>
<FONT color="green">621</FONT>    <a name="line.621"></a>
<FONT color="green">622</FONT>        @Override<a name="line.622"></a>
<FONT color="green">623</FONT>        /**<a name="line.623"></a>
<FONT color="green">624</FONT>         * Connects to CM, sets up container launch context <a name="line.624"></a>
<FONT color="green">625</FONT>         * for shell command and eventually dispatches the container <a name="line.625"></a>
<FONT color="green">626</FONT>         * start request to the CM. <a name="line.626"></a>
<FONT color="green">627</FONT>         */<a name="line.627"></a>
<FONT color="green">628</FONT>        public void run() {<a name="line.628"></a>
<FONT color="green">629</FONT>          // Connect to ContainerManager <a name="line.629"></a>
<FONT color="green">630</FONT>          connectToCM();<a name="line.630"></a>
<FONT color="green">631</FONT>    <a name="line.631"></a>
<FONT color="green">632</FONT>          LOG.info("Setting up container launch container for containerid=" + container.getId());<a name="line.632"></a>
<FONT color="green">633</FONT>          ContainerLaunchContext ctx = Records.newRecord(ContainerLaunchContext.class);<a name="line.633"></a>
<FONT color="green">634</FONT>    <a name="line.634"></a>
<FONT color="green">635</FONT>          ctx.setContainerId(container.getId());<a name="line.635"></a>
<FONT color="green">636</FONT>          ctx.setResource(container.getResource());<a name="line.636"></a>
<FONT color="green">637</FONT>    <a name="line.637"></a>
<FONT color="green">638</FONT>          try {<a name="line.638"></a>
<FONT color="green">639</FONT>            ctx.setUser(UserGroupInformation.getCurrentUser().getShortUserName());<a name="line.639"></a>
<FONT color="green">640</FONT>          } catch (IOException e) {<a name="line.640"></a>
<FONT color="green">641</FONT>            LOG.info("Getting current user info failed when trying to launch the container"<a name="line.641"></a>
<FONT color="green">642</FONT>                + e.getMessage());<a name="line.642"></a>
<FONT color="green">643</FONT>          }<a name="line.643"></a>
<FONT color="green">644</FONT>    <a name="line.644"></a>
<FONT color="green">645</FONT>          // Set the environment <a name="line.645"></a>
<FONT color="green">646</FONT>          ctx.setEnvironment(shellEnv);<a name="line.646"></a>
<FONT color="green">647</FONT>    <a name="line.647"></a>
<FONT color="green">648</FONT>          // Set the local resources <a name="line.648"></a>
<FONT color="green">649</FONT>          Map&lt;String, LocalResource&gt; localResources = new HashMap&lt;String, LocalResource&gt;();<a name="line.649"></a>
<FONT color="green">650</FONT>    <a name="line.650"></a>
<FONT color="green">651</FONT>          // The container for the eventual shell commands needs its own local resources too. <a name="line.651"></a>
<FONT color="green">652</FONT>          // In this scenario, if a shell script is specified, we need to have it copied <a name="line.652"></a>
<FONT color="green">653</FONT>          // and made available to the container. <a name="line.653"></a>
<FONT color="green">654</FONT>          if (!shellScriptPath.isEmpty()) {<a name="line.654"></a>
<FONT color="green">655</FONT>            LocalResource shellRsrc = Records.newRecord(LocalResource.class);<a name="line.655"></a>
<FONT color="green">656</FONT>            shellRsrc.setType(LocalResourceType.FILE);<a name="line.656"></a>
<FONT color="green">657</FONT>            shellRsrc.setVisibility(LocalResourceVisibility.APPLICATION);<a name="line.657"></a>
<FONT color="green">658</FONT>            try {<a name="line.658"></a>
<FONT color="green">659</FONT>              shellRsrc.setResource(ConverterUtils.getYarnUrlFromURI(new URI(shellScriptPath)));<a name="line.659"></a>
<FONT color="green">660</FONT>            } catch (URISyntaxException e) {<a name="line.660"></a>
<FONT color="green">661</FONT>              LOG.error("Error when trying to use shell script path specified in env"<a name="line.661"></a>
<FONT color="green">662</FONT>                  + ", path=" + shellScriptPath);<a name="line.662"></a>
<FONT color="green">663</FONT>              e.printStackTrace();<a name="line.663"></a>
<FONT color="green">664</FONT>    <a name="line.664"></a>
<FONT color="green">665</FONT>              // A failure scenario on bad input such as invalid shell script path <a name="line.665"></a>
<FONT color="green">666</FONT>              // We know we cannot continue launching the container <a name="line.666"></a>
<FONT color="green">667</FONT>              // so we should release it.<a name="line.667"></a>
<FONT color="green">668</FONT>              // TODO<a name="line.668"></a>
<FONT color="green">669</FONT>              numCompletedContainers.incrementAndGet();<a name="line.669"></a>
<FONT color="green">670</FONT>              numFailedContainers.incrementAndGet();<a name="line.670"></a>
<FONT color="green">671</FONT>              return;<a name="line.671"></a>
<FONT color="green">672</FONT>            }<a name="line.672"></a>
<FONT color="green">673</FONT>            shellRsrc.setTimestamp(shellScriptPathTimestamp);<a name="line.673"></a>
<FONT color="green">674</FONT>            shellRsrc.setSize(shellScriptPathLen);<a name="line.674"></a>
<FONT color="green">675</FONT>            localResources.put(ExecShellStringPath, shellRsrc);<a name="line.675"></a>
<FONT color="green">676</FONT>          }<a name="line.676"></a>
<FONT color="green">677</FONT>          ctx.setLocalResources(localResources);<a name="line.677"></a>
<FONT color="green">678</FONT>    <a name="line.678"></a>
<FONT color="green">679</FONT>          // Set the necessary command to execute on the allocated container <a name="line.679"></a>
<FONT color="green">680</FONT>          Vector&lt;CharSequence&gt; vargs = new Vector&lt;CharSequence&gt;(5);<a name="line.680"></a>
<FONT color="green">681</FONT>    <a name="line.681"></a>
<FONT color="green">682</FONT>          // Set executable command <a name="line.682"></a>
<FONT color="green">683</FONT>          vargs.add(shellCommand);<a name="line.683"></a>
<FONT color="green">684</FONT>          // Set shell script path <a name="line.684"></a>
<FONT color="green">685</FONT>          if (!shellScriptPath.isEmpty()) {<a name="line.685"></a>
<FONT color="green">686</FONT>            vargs.add(ExecShellStringPath);<a name="line.686"></a>
<FONT color="green">687</FONT>          }<a name="line.687"></a>
<FONT color="green">688</FONT>    <a name="line.688"></a>
<FONT color="green">689</FONT>          // Set args for the shell command if any<a name="line.689"></a>
<FONT color="green">690</FONT>          vargs.add(shellArgs);<a name="line.690"></a>
<FONT color="green">691</FONT>          // Add log redirect params<a name="line.691"></a>
<FONT color="green">692</FONT>          // TODO<a name="line.692"></a>
<FONT color="green">693</FONT>          // We should redirect the output to hdfs instead of local logs <a name="line.693"></a>
<FONT color="green">694</FONT>          // so as to be able to look at the final output after the containers <a name="line.694"></a>
<FONT color="green">695</FONT>          // have been released. <a name="line.695"></a>
<FONT color="green">696</FONT>          // Could use a path suffixed with /AppId/AppAttempId/ContainerId/std[out|err] <a name="line.696"></a>
<FONT color="green">697</FONT>          vargs.add("1&gt;" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/stdout");<a name="line.697"></a>
<FONT color="green">698</FONT>          vargs.add("2&gt;" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/stderr");<a name="line.698"></a>
<FONT color="green">699</FONT>    <a name="line.699"></a>
<FONT color="green">700</FONT>          // Get final commmand<a name="line.700"></a>
<FONT color="green">701</FONT>          StringBuilder command = new StringBuilder();<a name="line.701"></a>
<FONT color="green">702</FONT>          for (CharSequence str : vargs) {<a name="line.702"></a>
<FONT color="green">703</FONT>            command.append(str).append(" ");<a name="line.703"></a>
<FONT color="green">704</FONT>          }<a name="line.704"></a>
<FONT color="green">705</FONT>    <a name="line.705"></a>
<FONT color="green">706</FONT>          List&lt;String&gt; commands = new ArrayList&lt;String&gt;();<a name="line.706"></a>
<FONT color="green">707</FONT>          commands.add(command.toString());<a name="line.707"></a>
<FONT color="green">708</FONT>          ctx.setCommands(commands);<a name="line.708"></a>
<FONT color="green">709</FONT>    <a name="line.709"></a>
<FONT color="green">710</FONT>          StartContainerRequest startReq = Records.newRecord(StartContainerRequest.class);<a name="line.710"></a>
<FONT color="green">711</FONT>          startReq.setContainerLaunchContext(ctx);<a name="line.711"></a>
<FONT color="green">712</FONT>          try {<a name="line.712"></a>
<FONT color="green">713</FONT>            cm.startContainer(startReq);<a name="line.713"></a>
<FONT color="green">714</FONT>          } catch (YarnRemoteException e) {<a name="line.714"></a>
<FONT color="green">715</FONT>            LOG.info("Start container failed for :"<a name="line.715"></a>
<FONT color="green">716</FONT>                + ", containerId=" + container.getId());<a name="line.716"></a>
<FONT color="green">717</FONT>            e.printStackTrace();<a name="line.717"></a>
<FONT color="green">718</FONT>            // TODO do we need to release this container? <a name="line.718"></a>
<FONT color="green">719</FONT>          }<a name="line.719"></a>
<FONT color="green">720</FONT>    <a name="line.720"></a>
<FONT color="green">721</FONT>          // Get container status?<a name="line.721"></a>
<FONT color="green">722</FONT>          // Left commented out as the shell scripts are short lived <a name="line.722"></a>
<FONT color="green">723</FONT>          // and we are relying on the status for completed containers from RM to detect status<a name="line.723"></a>
<FONT color="green">724</FONT>    <a name="line.724"></a>
<FONT color="green">725</FONT>          //    GetContainerStatusRequest statusReq = Records.newRecord(GetContainerStatusRequest.class);<a name="line.725"></a>
<FONT color="green">726</FONT>          //    statusReq.setContainerId(container.getId());<a name="line.726"></a>
<FONT color="green">727</FONT>          //    GetContainerStatusResponse statusResp;<a name="line.727"></a>
<FONT color="green">728</FONT>          //try {<a name="line.728"></a>
<FONT color="green">729</FONT>          //statusResp = cm.getContainerStatus(statusReq);<a name="line.729"></a>
<FONT color="green">730</FONT>          //    LOG.info("Container Status"<a name="line.730"></a>
<FONT color="green">731</FONT>          //    + ", id=" + container.getId()<a name="line.731"></a>
<FONT color="green">732</FONT>          //    + ", status=" +statusResp.getStatus());<a name="line.732"></a>
<FONT color="green">733</FONT>          //} catch (YarnRemoteException e) {<a name="line.733"></a>
<FONT color="green">734</FONT>          //e.printStackTrace();<a name="line.734"></a>
<FONT color="green">735</FONT>          //}<a name="line.735"></a>
<FONT color="green">736</FONT>        }<a name="line.736"></a>
<FONT color="green">737</FONT>      }<a name="line.737"></a>
<FONT color="green">738</FONT>    <a name="line.738"></a>
<FONT color="green">739</FONT>      /**<a name="line.739"></a>
<FONT color="green">740</FONT>       * Connect to the Resource Manager<a name="line.740"></a>
<FONT color="green">741</FONT>       * @return Handle to communicate with the RM<a name="line.741"></a>
<FONT color="green">742</FONT>       */<a name="line.742"></a>
<FONT color="green">743</FONT>      private AMRMProtocol connectToRM() {<a name="line.743"></a>
<FONT color="green">744</FONT>        YarnConfiguration yarnConf = new YarnConfiguration(conf);<a name="line.744"></a>
<FONT color="green">745</FONT>        InetSocketAddress rmAddress = NetUtils.createSocketAddr(yarnConf.get(<a name="line.745"></a>
<FONT color="green">746</FONT>            YarnConfiguration.RM_SCHEDULER_ADDRESS,<a name="line.746"></a>
<FONT color="green">747</FONT>            YarnConfiguration.DEFAULT_RM_SCHEDULER_ADDRESS));<a name="line.747"></a>
<FONT color="green">748</FONT>        LOG.info("Connecting to ResourceManager at " + rmAddress);<a name="line.748"></a>
<FONT color="green">749</FONT>        return ((AMRMProtocol) rpc.getProxy(AMRMProtocol.class, rmAddress, conf));<a name="line.749"></a>
<FONT color="green">750</FONT>      }<a name="line.750"></a>
<FONT color="green">751</FONT>    <a name="line.751"></a>
<FONT color="green">752</FONT>      /** <a name="line.752"></a>
<FONT color="green">753</FONT>       * Register the Application Master to the Resource Manager<a name="line.753"></a>
<FONT color="green">754</FONT>       * @return the registration response from the RM<a name="line.754"></a>
<FONT color="green">755</FONT>       * @throws YarnRemoteException<a name="line.755"></a>
<FONT color="green">756</FONT>       */<a name="line.756"></a>
<FONT color="green">757</FONT>      private RegisterApplicationMasterResponse registerToRM() throws YarnRemoteException {<a name="line.757"></a>
<FONT color="green">758</FONT>        RegisterApplicationMasterRequest appMasterRequest = Records.newRecord(RegisterApplicationMasterRequest.class);<a name="line.758"></a>
<FONT color="green">759</FONT>    <a name="line.759"></a>
<FONT color="green">760</FONT>        // set the required info into the registration request: <a name="line.760"></a>
<FONT color="green">761</FONT>        // application attempt id, <a name="line.761"></a>
<FONT color="green">762</FONT>        // host on which the app master is running<a name="line.762"></a>
<FONT color="green">763</FONT>        // rpc port on which the app master accepts requests from the client <a name="line.763"></a>
<FONT color="green">764</FONT>        // tracking url for the app master<a name="line.764"></a>
<FONT color="green">765</FONT>        appMasterRequest.setApplicationAttemptId(appAttemptID);<a name="line.765"></a>
<FONT color="green">766</FONT>        appMasterRequest.setHost(appMasterHostname);<a name="line.766"></a>
<FONT color="green">767</FONT>        appMasterRequest.setRpcPort(appMasterRpcPort);<a name="line.767"></a>
<FONT color="green">768</FONT>        appMasterRequest.setTrackingUrl(appMasterTrackingUrl);<a name="line.768"></a>
<FONT color="green">769</FONT>    <a name="line.769"></a>
<FONT color="green">770</FONT>        return resourceManager.registerApplicationMaster(appMasterRequest);<a name="line.770"></a>
<FONT color="green">771</FONT>      }<a name="line.771"></a>
<FONT color="green">772</FONT>    <a name="line.772"></a>
<FONT color="green">773</FONT>      /**<a name="line.773"></a>
<FONT color="green">774</FONT>       * Setup the request that will be sent to the RM for the container ask.<a name="line.774"></a>
<FONT color="green">775</FONT>       * @param numContainers Containers to ask for from RM<a name="line.775"></a>
<FONT color="green">776</FONT>       * @return the setup ResourceRequest to be sent to RM<a name="line.776"></a>
<FONT color="green">777</FONT>       */<a name="line.777"></a>
<FONT color="green">778</FONT>      private ResourceRequest setupContainerAskForRM(int numContainers) {<a name="line.778"></a>
<FONT color="green">779</FONT>        ResourceRequest request = Records.newRecord(ResourceRequest.class);<a name="line.779"></a>
<FONT color="green">780</FONT>    <a name="line.780"></a>
<FONT color="green">781</FONT>        // setup requirements for hosts <a name="line.781"></a>
<FONT color="green">782</FONT>        // whether a particular rack/host is needed <a name="line.782"></a>
<FONT color="green">783</FONT>        // Refer to apis under org.apache.hadoop.net for more <a name="line.783"></a>
<FONT color="green">784</FONT>        // details on how to get figure out rack/host mapping.<a name="line.784"></a>
<FONT color="green">785</FONT>        // using * as any host will do for the distributed shell app<a name="line.785"></a>
<FONT color="green">786</FONT>        request.setHostName("*");<a name="line.786"></a>
<FONT color="green">787</FONT>    <a name="line.787"></a>
<FONT color="green">788</FONT>        // set no. of containers needed<a name="line.788"></a>
<FONT color="green">789</FONT>        request.setNumContainers(numContainers);<a name="line.789"></a>
<FONT color="green">790</FONT>    <a name="line.790"></a>
<FONT color="green">791</FONT>        // set the priority for the request<a name="line.791"></a>
<FONT color="green">792</FONT>        Priority pri = Records.newRecord(Priority.class);<a name="line.792"></a>
<FONT color="green">793</FONT>        // TODO - what is the range for priority? how to decide? <a name="line.793"></a>
<FONT color="green">794</FONT>        pri.setPriority(requestPriority);<a name="line.794"></a>
<FONT color="green">795</FONT>        request.setPriority(pri);<a name="line.795"></a>
<FONT color="green">796</FONT>    <a name="line.796"></a>
<FONT color="green">797</FONT>        // Set up resource type requirements<a name="line.797"></a>
<FONT color="green">798</FONT>        // For now, only memory is supported so we set memory requirements<a name="line.798"></a>
<FONT color="green">799</FONT>        Resource capability = Records.newRecord(Resource.class);<a name="line.799"></a>
<FONT color="green">800</FONT>        capability.setMemory(containerMemory);<a name="line.800"></a>
<FONT color="green">801</FONT>        request.setCapability(capability);<a name="line.801"></a>
<FONT color="green">802</FONT>    <a name="line.802"></a>
<FONT color="green">803</FONT>        return request;<a name="line.803"></a>
<FONT color="green">804</FONT>      }<a name="line.804"></a>
<FONT color="green">805</FONT>    <a name="line.805"></a>
<FONT color="green">806</FONT>      /**<a name="line.806"></a>
<FONT color="green">807</FONT>       * Ask RM to allocate given no. of containers to this Application Master<a name="line.807"></a>
<FONT color="green">808</FONT>       * @param requestedContainers Containers to ask for from RM<a name="line.808"></a>
<FONT color="green">809</FONT>       * @return Response from RM to AM with allocated containers <a name="line.809"></a>
<FONT color="green">810</FONT>       * @throws YarnRemoteException<a name="line.810"></a>
<FONT color="green">811</FONT>       */<a name="line.811"></a>
<FONT color="green">812</FONT>      private AMResponse sendContainerAskToRM(List&lt;ResourceRequest&gt; requestedContainers)<a name="line.812"></a>
<FONT color="green">813</FONT>          throws YarnRemoteException {<a name="line.813"></a>
<FONT color="green">814</FONT>        AllocateRequest req = Records.newRecord(AllocateRequest.class);<a name="line.814"></a>
<FONT color="green">815</FONT>        req.setResponseId(rmRequestID.incrementAndGet());<a name="line.815"></a>
<FONT color="green">816</FONT>        req.setApplicationAttemptId(appAttemptID);<a name="line.816"></a>
<FONT color="green">817</FONT>        req.addAllAsks(requestedContainers);<a name="line.817"></a>
<FONT color="green">818</FONT>        req.addAllReleases(releasedContainers);<a name="line.818"></a>
<FONT color="green">819</FONT>        req.setProgress((float)numCompletedContainers.get()/numTotalContainers);<a name="line.819"></a>
<FONT color="green">820</FONT>    <a name="line.820"></a>
<FONT color="green">821</FONT>        LOG.info("Sending request to RM for containers"<a name="line.821"></a>
<FONT color="green">822</FONT>            + ", requestedSet=" + requestedContainers.size()<a name="line.822"></a>
<FONT color="green">823</FONT>            + ", releasedSet=" + releasedContainers.size()<a name="line.823"></a>
<FONT color="green">824</FONT>            + ", progress=" + req.getProgress());<a name="line.824"></a>
<FONT color="green">825</FONT>    <a name="line.825"></a>
<FONT color="green">826</FONT>        for (ResourceRequest  rsrcReq : requestedContainers) {<a name="line.826"></a>
<FONT color="green">827</FONT>          LOG.info("Requested container ask: " + rsrcReq.toString());<a name="line.827"></a>
<FONT color="green">828</FONT>        }<a name="line.828"></a>
<FONT color="green">829</FONT>        for (ContainerId id : releasedContainers) {<a name="line.829"></a>
<FONT color="green">830</FONT>          LOG.info("Released container, id=" + id.getId());<a name="line.830"></a>
<FONT color="green">831</FONT>        }<a name="line.831"></a>
<FONT color="green">832</FONT>    <a name="line.832"></a>
<FONT color="green">833</FONT>        AllocateResponse resp = resourceManager.allocate(req);<a name="line.833"></a>
<FONT color="green">834</FONT>        return resp.getAMResponse();<a name="line.834"></a>
<FONT color="green">835</FONT>      }<a name="line.835"></a>
<FONT color="green">836</FONT>    }<a name="line.836"></a>




























































</PRE>
</BODY>
</HTML>
