/*
 * The MIT License
 *
 *  Copyright (c) 2015, CloudBees, Inc.
 *
 *  Permission is hereby granted, free of charge, to any person obtaining a copy
 *  of this software and associated documentation files (the "Software"), to deal
 *  in the Software without restriction, including without limitation the rights
 *  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *  copies of the Software, and to permit persons to whom the Software is
 *  furnished to do so, subject to the following conditions:
 *
 *  The above copyright notice and this permission notice shall be included in
 *  all copies or substantial portions of the Software.
 *
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 *  THE SOFTWARE.
 *
 */

package com.cloudbees.jenkins.plugins.amazonecs;

import hudson.slaves.AbstractCloudComputer;

/**
 * Amazon EC2 Container Service implementation of {@link hudson.model.Computer}
 *
 * @author <a href="mailto:nicolas.deloof@gmail.com">Nicolas De Loof</a>
 */
public class ECSComputer extends AbstractCloudComputer {
    public ECSComputer(ECSSlave slave) {
        super(slave);
    }
}

@@ -1206,16 +1206,73 @@
 onecs;%0A%0A
+import hudson.model.Executor;%0Aimport hudson.model.Queue;%0A
 import h
@@ -1308,16 +1308,45 @@
 puter;%0A%0A
+import java.io.IOException;%0A%0A
 /**%0A * A
@@ -1599,16 +1599,16 @@
 lave) %7B%0A
-
         
@@ -1626,11 +1626,751 @@
 );%0A    %7D
+%0A%0A    @Override%0A    public void taskCompleted(Executor executor, Queue.Task task, long durationMS) %7B%0A        super.taskCompleted(executor, task, durationMS);%0A        terminate();%0A    %7D%0A%0A    @Override%0A    public void taskCompletedWithProblems(Executor executor, Queue.Task task, long durationMS, Throwable problems) %7B%0A        super.taskCompletedWithProblems(executor, task, durationMS, problems);%0A        terminate();%0A    %7D%0A%0A    /**%0A     * Computer is terminated after build completion so we enforce it will only be used once.%0A     */%0A    private void terminate() %7B%0A        setAcceptingTasks(false);%0A        try %7B%0A            getNode().terminate();%0A        %7D catch (InterruptedException e) %7B%0A        %7D catch (IOException e) %7B%0A        %7D%0A    %7D
 %0A%7D%0A

terminate computer immediately after build completion