package hudson.plugins.ec2.one; import hudson.Util; import hudson.plugins.ec2.EC2Cloud; import hudson.plugins.ec2.InstanceState; import hudson.slaves.SlaveComputer; import java.io.IOException; import java.util.Collections; import org.kohsuke.stapler.HttpRedirect; import org.kohsuke.stapler.HttpResponse; import com.amazonaws.AmazonClientException; import com.amazonaws.services.ec2.AmazonEC2; import com.amazonaws.services.ec2.model.DescribeInstancesRequest; import com.amazonaws.services.ec2.model.GetConsoleOutputRequest; import com.amazonaws.services.ec2.model.Instance; /** * @author Kohsuke Kawaguchi */ public class OneEC2Computer extends SlaveComputer { /** * Cached description of this EC2 instance. Lazily fetched. */ private volatile Instance ec2InstanceDescription; public OneEC2Computer(final OneEC2Slave slave) { super(slave); } @Override public OneEC2Slave getNode() { return (OneEC2Slave) super.getNode(); } public String getInstanceId() { return getName(); } /** * Gets the EC2 console output. */ public String getConsoleOutput() throws AmazonClientException { AmazonEC2 ec2 = OneEC2Cloud.get().connect(); GetConsoleOutputRequest request = new GetConsoleOutputRequest(getInstanceId()); return ec2.getConsoleOutput(request).getOutput(); } /** * Obtains the instance state description in EC2. *

* This method returns a cached state, so it's not suitable to check * {@link Instance#getState()} and {@link Instance#getStateCode()} from the * returned instance (but all the other fields are valid as it won't * change.) The cache can be flushed using * {@link #updateInstanceDescription()} */ public Instance describeInstance() throws AmazonClientException { if (ec2InstanceDescription == null) { ec2InstanceDescription = _describeInstance(); } return ec2InstanceDescription; } /** * This will flush any cached description held by * {@link #describeInstance()}. */ public Instance updateInstanceDescription() throws AmazonClientException { return ec2InstanceDescription = _describeInstance(); } /** * Gets the current state of the instance. *

* Unlike {@link #describeInstance()}, this method always return the current * status by calling EC2. */ public InstanceState getState() throws AmazonClientException { ec2InstanceDescription = _describeInstance(); return InstanceState.find(ec2InstanceDescription.getState() .getName()); } /** * Number of milli-secs since the instance was started. */ public long getUptime() throws AmazonClientException { return System.currentTimeMillis() - describeInstance().getLaunchTime().getTime(); } /** * Returns uptime in the human readable form. */ public String getUptimeString() throws AmazonClientException { return Util.getTimeSpanString(getUptime()); } private Instance _describeInstance() throws AmazonClientException { DescribeInstancesRequest request = new DescribeInstancesRequest(); request.setInstanceIds(Collections . singletonList(getNode().getInstanceId())); return EC2Cloud.get().connect().describeInstances(request) .getReservations().get(0).getInstances().get(0); } /** * When the slave is deleted, terminate the instance. */ @Override public HttpResponse doDoDelete() throws IOException { checkPermission(DELETE); if (getNode() != null) { getNode().terminate(); } return new HttpRedirect(".."); } /** * What username to use to run root-like commands */ public String getRemoteAdmin() { return getNode().getRemoteAdmin(); } public int getSshPort() { return getNode().getSshPort(); } public String getRootCommandPrefix() { return getNode().getRootCommandPrefix(); } }