VirtualizationHostComponent.java

/*
 * RHQ Management Platform
 * Copyright (C) 2005-2008 Red Hat, Inc.
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
package org.rhq.plugins.virt;

import java.util.Set;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.libvirt.LibvirtException;

import org.rhq.core.domain.configuration.Configuration;
import org.rhq.core.domain.configuration.PropertySimple;
import org.rhq.core.domain.measurement.AvailabilityType;
import org.rhq.core.domain.measurement.MeasurementDataNumeric;
import org.rhq.core.domain.measurement.MeasurementDataTrait;
import org.rhq.core.domain.measurement.MeasurementReport;
import org.rhq.core.domain.measurement.MeasurementScheduleRequest;
import org.rhq.core.pluginapi.configuration.ConfigurationFacet;
import org.rhq.core.pluginapi.configuration.ConfigurationUpdateReport;
import org.rhq.core.pluginapi.inventory.CreateChildResourceFacet;
import org.rhq.core.pluginapi.inventory.CreateResourceReport;
import org.rhq.core.pluginapi.inventory.InvalidPluginConfigurationException;
import org.rhq.core.pluginapi.inventory.ResourceComponent;
import org.rhq.core.pluginapi.inventory.ResourceContext;
import org.rhq.core.pluginapi.measurement.MeasurementFacet;
import org.rhq.plugins.virt.LibVirtConnection.HVInfo;

/**
 * Component for managing both virtual hosts and guests though some features are guest only and only
 * defined in the guest metadata.
 *
 * @author Greg Hinkle
 */
public class VirtualizationHostComponent implements ResourceComponent<ResourceComponent<?>>, MeasurementFacet, ConfigurationFacet,
    CreateChildResourceFacet {

    private Log log = LogFactory.getLog(VirtualizationDomainComponent.class);
    private String uri = "";
    private LibVirtConnection virt;
    private long cpuNanosLast;
    private long cpuCheckedLast;

    public void start(ResourceContext resourceContext) throws InvalidPluginConfigurationException, Exception {
        uri = resourceContext.getPluginConfiguration().getSimpleValue("connectionURI", "");
        virt = new LibVirtConnection(uri);
        if (!virt.isConnected()) {
            virt = null;
        }
    }

    public void stop() {
        try {
            virt.close();
        } catch (Exception e) {
            log.error("Exception Stopping Libvirt", e);
        }
    }

    public AvailabilityType getAvailability() {
        /*try {
            System.out.println("In get Availability " + virt.getConnectionURI());
        } catch (Exception e) {
            System.out.println("WOIERUOIWUROIEWUROWUROIEUROIEUOIR");
        }*/
        if (virt.isConnected()) {
            try {
                virt.getHVInfo();
                return AvailabilityType.UP;
            } catch (Exception e) {
                return AvailabilityType.DOWN;
            }
        } else {
            return AvailabilityType.DOWN;
        }
    }

    public void getValues(MeasurementReport report, Set<MeasurementScheduleRequest> metrics) throws Exception {
        if (virt.isConnected()) {
            HVInfo hi = virt.getHVInfo();
            for (MeasurementScheduleRequest request : metrics) {
                if (request.getName().equals("cpus")) {
                    report.addData(new MeasurementDataTrait(request, "" + hi.nodeInfo.cpus));
                } else if (request.getName().equals("memory")) {
                    report.addData(new MeasurementDataTrait(request, "" + hi.nodeInfo.memory));
                } else if (request.getName().equals("memoryUsage")) {
                    report.addData(new MeasurementDataNumeric(request, virt.getMemoryPercentage()));
                } else if (request.getName().equals("cpuUsage")) {
                    long checked = System.nanoTime();
                    long cpuNanos = virt.getCPUTime();

                    if (cpuCheckedLast != 0) {
                        long duration = checked - cpuCheckedLast;

                        long diff = cpuNanos - cpuNanosLast;

                        double percentage = ((double) diff) / ((double) duration);
                        report.addData(new MeasurementDataNumeric(request, percentage));
                    }
                    cpuCheckedLast = checked;
                    cpuNanosLast = cpuNanos;
                } else if (request.getName().equals("cpuTime")) {
                    report.addData(new MeasurementDataNumeric(request, (double) virt.getCPUTime()));
                }
            }
        }
    }

    public Configuration loadResourceConfiguration() throws LibvirtException {
        Configuration config = new Configuration();
        if (virt.isConnected()) {
            HVInfo info = virt.getHVInfo();
            config.put(new PropertySimple("hypervisorType", info.hvType));
            config.put(new PropertySimple("hostName", info.hostname));
            config.put(new PropertySimple("libvirtVersion", getLibvirtVersion(info.libvirtVersion)));
        }
        return config;
    }

    public String getLibvirtVersion(long version) {
        long major, minor, micro, remainder;
        major = version / 1000000;
        remainder = version % 1000000;
        minor = remainder / 1000;
        micro = remainder % 1000;
        return major + "." + minor + "." + micro;

    }

    public void updateResourceConfiguration(ConfigurationUpdateReport report) {
        // This configuration can not be updated.
    }

    // TODO 
    public CreateResourceReport createResource(CreateResourceReport report) {
        /*try {
            System.out.println(virt.getConnectionURI());
        } catch (Exception e) {
            e.printStackTrace();
        }*/
        /*String xml = DomainConfigurationEditor.getXml(report.getResourceConfiguration());

        log.info("Defining new domain");
        log.debug("New virtualization domain xml:\n" + xml);

        try {
            this.virt.defineDomain(xml);
            report.setStatus(CreateResourceStatus.SUCCESS);
        } catch (LibvirtException e) {
            log.error("Exception creating the domain", e);
            report.setStatus(CreateResourceStatus.FAILURE);
        }
        */
        return report;
    }

    public LibVirtConnection getConnection() throws LibvirtException {
        return virt;
    }

}