RouteTask.java

/*******************************************************************************
 * Copyright 2012 Urbancode, Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package com.urbancode.terraform.tasks.aws;

import org.apache.log4j.Logger;

import com.amazonaws.services.ec2.AmazonEC2;
import com.urbancode.terraform.tasks.aws.helpers.AWSHelper;
import com.urbancode.x2o.tasks.SubTask;

public class RouteTask extends SubTask {

    //**********************************************************************************************
    // CLASS
    //**********************************************************************************************
    final static private Logger log = Logger.getLogger(RouteTask.class);

    //**********************************************************************************************
    // INSTANCE
    //**********************************************************************************************

    private AmazonEC2 ec2Client;
    private AWSHelper helper;
    private ContextAWS context;

    private String routeTableId;
    private String target;
    private String cidr;
    private String targetName;

    //----------------------------------------------------------------------------------------------
    public RouteTask(ContextAWS context) {
        super(context);
        this.context = context;
        helper = new AWSHelper();
    }

    //----------------------------------------------------------------------------------------------
    public String getTarget() {
        return target;
    }

    //----------------------------------------------------------------------------------------------
    public String getTargetName() {
        return targetName;
    }

    //----------------------------------------------------------------------------------------------
    public String getDest() {
        return cidr;
    }

    //----------------------------------------------------------------------------------------------
    public void setRouteTableId(String id) {
        this.routeTableId = id;
    }

    //----------------------------------------------------------------------------------------------
    public void setTarget(String id) {
        this.target = id;
    }

    //----------------------------------------------------------------------------------------------
    public void setTargetName(String name) {
        this.targetName = name;
    }

    //----------------------------------------------------------------------------------------------
    public void setDest(String cidr) {
        this.cidr = cidr;
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void create() {
        if (ec2Client == null) {
            ec2Client = context.fetchEC2Client();
        }

        log.info("Creating Route");
        try {
            if (target == null || cidr == null || routeTableId == null) {
                log.error("Error: Route is missing data:" +
                          "target: " + target +
                          "dest: " + cidr +
                          "RouteTable-Id: " + routeTableId);
                return;
            }
            helper.createRoute(routeTableId, cidr, target, ec2Client);
            log.info("Route Created.");
        }
        finally {
            ec2Client = null;
        }
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void destroy() {
        if (ec2Client == null) {
            ec2Client = context.fetchEC2Client();
        }

        try {
            helper.deleteRoute(routeTableId, cidr, ec2Client);
        }
        finally {
            ec2Client = null;
        }
    }

}