RemoveVirtualNic

RemoveVirtualNic.java

/*
 * ****************************************************************************
 * Copyright VMware, Inc. 2010-2016.  All Rights Reserved.
 * ****************************************************************************
 *
 * This software is made available for use under the terms of the BSD
 * 3-Clause license:
 *
 * Redistribution and use in source and binary forms, with or without 
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright 
 *    notice, this list of conditions and the following disclaimer.
 * 
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in 
 *    the documentation and/or other materials provided with the 
 *    distribution.
 * 
 * 3. Neither the name of the copyright holder nor the names of its 
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 
 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */


package com.vmware.host;

import com.vmware.common.annotations.Action;
import com.vmware.common.annotations.Option;
import com.vmware.common.annotations.Sample;
import com.vmware.connection.ConnectedVimServiceBase;
import com.vmware.vim25.*;

import java.util.List;
import java.util.Map;

/**
 * <pre>
 * RemoveVirtualNic
 *
 * This sample removes a Virtual Nic from a PortGroup on a vSwitch
 *
 * <b>Parameters:</b>
 * url              [required] : url of the web service
 * username         [required] : username for the authentication
 * password         [required] : password for the authentication
 * portgroupname    [required] : Name of port group to remove Virtual Nic from
 * hostname         [required] : Name of host
 * datacentername   [optional] : Name of datacenter
 *
 * <b>Command Line:</b>
 * Remove a VirtualNic from a PortGroup
 * run.bat com.vmware.host.RemoveVirtualNic --url [webserviceurl]
 * --username [username] --password  [password] --datacentername [mydatacenter]
 * --portgroupname [myportgroup] --hostname [hostname]
 *
 * Remove a VirtualNic from a PortGroup without hostname
 * run.bat com.vmware.host.RemoveVirtualNic --url [webserviceurl]
 * --username [username] --password  [password] --datacentername [mydatacenter]
 * --portgroupname [myportgroup]
 *
 * Remove a VirtualNic from a PortGroup without datacentername
 * run.bat com.vmware.host.RemoveVirtualNic --url [webserviceurl]
 * --username [username] --password  [password]
 * --portgroupname [myportgroup] --hostname [name of the host]
 * </pre>
 */
@Sample(
        name = "remove-virtual-nic",
        description = "removes a Virtual Nic from a PortGroup on a vSwitch"
)
public class RemoveVirtualNic extends ConnectedVimServiceBase {
    ManagedObjectReference propCollectorRef = null;
    ManagedObjectReference rootFolder;

    String datacenter;
    String host;
    String portgroupname;

    @Option(name = "datacentername", required = false, description = "Name of datacenter")
    public void setDatacenter(String datacenter) {
        this.datacenter = datacenter;
    }

    @Option(name = "hostname", description = "name of host")
    public void setHost(String host) {
        this.host = host;
    }

    @Option(name = "portgroupname", description = "Name of port group to remove Virtual Nic from")
    public void setPortgroupname(String portgroupname) {
        this.portgroupname = portgroupname;
    }

    void init() {
        propCollectorRef = serviceContent.getPropertyCollector();
        rootFolder = serviceContent.getRootFolder();
    }

    @SuppressWarnings("unchecked")
    void removeVirtualNic() {
        ManagedObjectReference dcmor;
        ManagedObjectReference hostfoldermor;
        ManagedObjectReference hostmor = null;

         try {
            if (((datacenter != null) && (host != null))
                    || ((datacenter != null) && (host == null))) {
                Map<String, ManagedObjectReference> dcResults = getMOREFs.inFolderByType(
                        serviceContent.getRootFolder(), "Datacenter", new RetrieveOptions());
                dcmor = dcResults.get(datacenter);
                if (dcmor == null) {
                    System.out.println("Datacenter not found");
                    return;
                }
                hostfoldermor = (ManagedObjectReference) getMOREFs.entityProps(dcmor,
                        new String[] { "hostFolder" }).get("hostFolder");
                Map<String, ManagedObjectReference> hostResults = getMOREFs.inFolderByType(
                        hostfoldermor, "HostSystem", new RetrieveOptions());
                hostmor = hostResults.get(host);
            } else if ((datacenter == null) && (host != null)) {
                Map<String, ManagedObjectReference> hostResults = getMOREFs.inFolderByType(
                        serviceContent.getRootFolder(), "HostSystem", new RetrieveOptions());
                hostmor = hostResults.get(host);
            }

            if (hostmor != null) {
                HostConfigManager configMgr = (HostConfigManager) getMOREFs.entityProps(hostmor,
                        new String[] { "configManager" }).get("configManager");
                ManagedObjectReference nwSystem = configMgr.getNetworkSystem();
                ArrayOfHostVirtualNic arrayHostVirtualNic = (ArrayOfHostVirtualNic) getMOREFs
                        .entityProps(nwSystem, new String[] { "networkInfo.vnic" }).get(
                                "networkInfo.vnic");
                List<HostVirtualNic> hvncArr = arrayHostVirtualNic.getHostVirtualNic();
                boolean foundOne = false;
                for (HostVirtualNic nic : hvncArr) {
                    String portGroup = nic.getPortgroup();
                    if (portGroup.equals(portgroupname)) {
                        vimPort.removeVirtualNic(nwSystem, nic.getDevice());
                        foundOne = true;
                    }
                }
                if (foundOne) {
                    System.out
                            .println("Successfully removed virtual nic from portgroup : "
                                    + portgroupname);
                } else {
                    System.out.println("No virtual nic found on portgroup : "
                            + portgroupname);
                }
            } else {
                System.out.println("Host not found");
            }
        } catch (HostConfigFaultFaultMsg ex) {
            System.out.println("Failed : Configuration falilures. ");
        } catch (NotFoundFaultMsg ex) {
            System.out.println("Failed : " + ex);
        } catch (RuntimeFaultFaultMsg ex) {
            System.out.println("Failed : " + ex);
        } catch (Exception e) {
            System.out.println("Failed : " + e);
        }

    }

    @Action
    public void run() {
        init();
        removeVirtualNic();
    }

}