NetworkRefTask.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.vmware;

import java.rmi.RemoteException;
import java.util.List;

import org.apache.log4j.Logger;

import com.urbancode.terraform.tasks.vmware.util.SwitchResult;
import com.urbancode.terraform.tasks.vmware.util.VirtualHost;
import com.urbancode.x2o.tasks.SubTask;
import com.vmware.vim25.VirtualEthernetCard;
import com.vmware.vim25.mo.VirtualMachine;

public class NetworkRefTask extends SubTask {

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

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

    private int nicIndex;
    private String networkName;
    private String nicType = "E1000";
    private VirtualMachine vm = null;
    private VirtualHost host = null;
    private SwitchResult vSwitch = null;

    //----------------------------------------------------------------------------------------------
    public NetworkRefTask() {
        super();
    }

    //----------------------------------------------------------------------------------------------
    public int getNicIndex() {
        return this.nicIndex;
    }

    //----------------------------------------------------------------------------------------------
    public String getNetworkName() {
        return this.networkName;
    }

    //----------------------------------------------------------------------------------------------
    public String getNicType() {
        return nicType;
    }

    //----------------------------------------------------------------------------------------------
    public SwitchResult fetchSwitch() {
        return this.vSwitch;
    }

    //----------------------------------------------------------------------------------------------
    public void setVirtualMachine(VirtualMachine vm) {
        this.vm = vm;
    }

    //----------------------------------------------------------------------------------------------
    public void setNetworkName(String networkName) {
        this.networkName = networkName;
    }

    //----------------------------------------------------------------------------------------------
    public void setNicIndex(int nicIndex) {
        this.nicIndex = nicIndex;
    }

    //----------------------------------------------------------------------------------------------
    public void setNicType(String nicType) {
        this.nicType = nicType;
    }

    //----------------------------------------------------------------------------------------------
    public void setVirtualSwitch(SwitchResult vSwitch) {
        this.vSwitch = vSwitch;
    }

    //----------------------------------------------------------------------------------------------
    public void setVirtualHost(VirtualHost host) {
        this.host = host;
    }

    //----------------------------------------------------------------------------------------------
    /**
     * Attaches a NIC to a VM.
     * @see VirtualHost
     * @throws RemoteException
     * @throws InterruptedException
     */
    public void attachNic()
    throws RemoteException, InterruptedException {
        try {
            List<VirtualEthernetCard> nics = host.getAllNics(vm);
            if (nicIndex < nics.size()) {
                VirtualEthernetCard nic = host.getNic(vm, nicIndex);
                host.attachNic(vm, nic, vSwitch.getNetwork());
            }
        }
        catch (IndexOutOfBoundsException e) {
            log.warn("Tried to attach NIC before card was created. NIC not attached.");
        }
        catch (IllegalArgumentException e) {
            log.warn("Tried to attach NIC before card was created. NIC not attached.");
        }

    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void create() {
        try {
            attachNic();
        }
        catch (RemoteException e) {
            log.warn("RemoteException while attaching NIC", e);
        }
        catch (InterruptedException e) {
            log.warn("InterruptedException while attaching NIC", e);
        }
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void destroy() {
        //this gets destroyed when the clone is destroyed
    }

}