com.netflix.simianarmy.client.vsphere.VSphereServiceConnection.java Source code

Introduction

Here is the source code for com.netflix.simianarmy.client.vsphere.VSphereServiceConnection.java

Source

/*
 *  Copyright 2012 Immobilien Scout GmbH
 *
 * 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.netflix.simianarmy.client.vsphere;

import java.net.MalformedURLException;
import java.net.URL;
import java.rmi.RemoteException;
import java.util.Arrays;

import com.amazonaws.AmazonServiceException;
import com.netflix.simianarmy.MonkeyConfiguration;
import com.vmware.vim25.InvalidProperty;
import com.vmware.vim25.RuntimeFault;
import com.vmware.vim25.mo.InventoryNavigator;
import com.vmware.vim25.mo.ManagedEntity;
import com.vmware.vim25.mo.ServiceInstance;
import com.vmware.vim25.mo.VirtualMachine;

/**
 * Wraps the connection to VSphere and handles the raw service calls.
 *
 * The following properties can be overridden in the client.properties
 * simianarmy.client.vsphere.url                                = https://YOUR_VSPHERE_SERVER/sdk
 * simianarmy.client.vsphere.username                           = YOUR_SERVICE_ACCOUNT_USERNAME
 * simianarmy.client.vsphere.password                           = YOUR_SERVICE_ACCOUNT_PASSWORD
 *
 * @author ingmar.krusch@immobilienscout24.de
 */
public class VSphereServiceConnection {
    /** The type of managedEntity we operate on are virtual machines. */
    public static final String VIRTUAL_MACHINE_TYPE_NAME = "VirtualMachine";

    /** The username that is used to connect to VSpehere Center. */
    private String username = null;

    /** The password that is used to connect to VSpehere Center. */
    private String password = null;

    /** The url that is used to connect to VSpehere Center. */
    private String url = null;

    /** The ServiceInstance that is used to issue multiple requests to VSpehere Center. */
    private ServiceInstance service = null;

    /**
     * Constructor.
     */
    public VSphereServiceConnection(MonkeyConfiguration config) {
        this.url = config.getStr("simianarmy.client.vsphere.url");
        this.username = config.getStr("simianarmy.client.vsphere.username");
        this.password = config.getStr("simianarmy.client.vsphere.password");
    }

    /** disconnect from the service if not already disconnected. */
    public void disconnect() {
        if (service != null) {
            service.getServerConnection().logout();
            service = null;
        }
    }

    /** connect to the service if not already connected. */
    public void connect() throws AmazonServiceException {
        try {
            if (service == null) {
                service = new ServiceInstance(new URL(url), username, password, true);
            }
        } catch (RemoteException e) {
            throw new AmazonServiceException("cannot connect to VSphere", e);
        } catch (MalformedURLException e) {
            throw new AmazonServiceException("cannot connect to VSphere", e);
        }
    }

    /**
     * Gets the named VirtualMachine.
     */
    public VirtualMachine getVirtualMachineById(String instanceId) throws RemoteException {
        InventoryNavigator inventoryNavigator = getInventoryNavigator();
        VirtualMachine virtualMachine = (VirtualMachine) inventoryNavigator
                .searchManagedEntity(VIRTUAL_MACHINE_TYPE_NAME, instanceId);

        return virtualMachine;
    }

    /**
     * Return all VirtualMachines from VSpehere Center.
     *
     * @throws AmazonServiceException
     *             If there is any communication error or if no VirtualMachine's are found. */
    public VirtualMachine[] describeVirtualMachines() throws AmazonServiceException {
        ManagedEntity[] mes = null;

        try {
            mes = getInventoryNavigator().searchManagedEntities(VIRTUAL_MACHINE_TYPE_NAME);
        } catch (InvalidProperty e) {
            throw new AmazonServiceException("cannot query VSphere", e);
        } catch (RuntimeFault e) {
            throw new AmazonServiceException("cannot query VSphere", e);
        } catch (RemoteException e) {
            throw new AmazonServiceException("cannot query VSphere", e);
        }

        if (mes == null || mes.length == 0) {
            throw new AmazonServiceException(
                    "vsphere returned zero entities of type \"" + VIRTUAL_MACHINE_TYPE_NAME + "\"");
        } else {
            return Arrays.copyOf(mes, mes.length, VirtualMachine[].class);
        }
    }

    protected InventoryNavigator getInventoryNavigator() {
        return new InventoryNavigator(service.getRootFolder());
    }

    public String getUsername() {
        return username;
    }

    public String getPassword() {
        return password;
    }

    public String getUrl() {
        return url;
    }
}