VMWareMachineSpecsUtils.java

/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2013, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.redhat.vmtruckloader.vmware;

import org.redhat.vmtruckloader.service.MachineSpecification;

import com.vmware.vim25.Description;
import com.vmware.vim25.VirtualDeviceConfigSpec;
import com.vmware.vim25.VirtualDeviceConfigSpecOperation;
import com.vmware.vim25.VirtualEthernetCard;
import com.vmware.vim25.VirtualEthernetCardNetworkBackingInfo;
import com.vmware.vim25.VirtualMachineConfigSpec;
import com.vmware.vim25.VirtualVmxnet3;
import com.vmware.vim25.mo.ServiceInstance;

/**
 * 
 * @author Romain Pelisse - romain@redhat.com
 *
 */
public class VMWareMachineSpecsUtils {

	private final static long GIGABYTE = 1024L;
	
	private final static String NETWORK_INTERFACE_NAME = "Network Adapter 1";
	private final static String NETWORK_ADDRESS_TYPE = "generated";

	public static VirtualMachineConfigSpec buildVmSpec(ServiceInstance serviceInstance, MachineSpecification spec) {
		VirtualMachineConfigSpec vmSpec = new VirtualMachineConfigSpec();
		vmSpec.setName(spec.getHostname());
		vmSpec.setNumCPUs(spec.getNbCpu());
		vmSpec.setMemoryMB(spec.getVRAM() * GIGABYTE);
		vmSpec.setAnnotation(spec.getRole());

		final String netName = spec.getVLAN();
		VirtualDeviceConfigSpec ethernet = createNicSpec(netName,VirtualDeviceConfigSpecOperation.add);
		
		vmSpec.setDeviceChange(new VirtualDeviceConfigSpec[] { ethernet,null });
		return vmSpec;
	}
	
	private static VirtualDeviceConfigSpec createNicSpec(String netName, VirtualDeviceConfigSpecOperation operation) {
		final String nicName = NETWORK_INTERFACE_NAME;
		VirtualDeviceConfigSpec nicSpec = new VirtualDeviceConfigSpec();
		nicSpec.setOperation(operation);

		VirtualEthernetCard nic = new VirtualVmxnet3();
		VirtualEthernetCardNetworkBackingInfo nicBacking = new VirtualEthernetCardNetworkBackingInfo();
		nicBacking.setDeviceName(netName);

		Description info = new Description();
		info.setLabel(nicName);
		info.setSummary(netName);
		nic.setDeviceInfo(info);

		// type: "generated", "manual", "assigned" by VC
		nic.setAddressType(NETWORK_ADDRESS_TYPE);
		nic.setBacking(nicBacking);
		nic.setKey(0);	

		nicSpec.setDevice(nic);
		return nicSpec;
	}

	
	public  static VirtualMachineConfigSpec buildVmSpecAndSetCpusAndMemory(ServiceInstance serviceInstance, MachineSpecification spec) {
		VirtualMachineConfigSpec vmSpec = new VirtualMachineConfigSpec();
		vmSpec.setName(spec.getVmName());
						
		if ( spec.getNbCpu() != 0 )
			vmSpec.setNumCPUs(spec.getNbCpu());
		if ( spec.getVRAM() != 0 )
			vmSpec.setMemoryMB(Long.valueOf(spec.getVRAM() * 1024));
		return vmSpec;
	}


	public  static VirtualMachineConfigSpec updateVmSpecAndSetCpusAndMemory(ServiceInstance serviceInstance, MachineSpecification spec, VirtualMachineConfigSpec vmSpec) {
		vmSpec.setName(spec.getVmName());
						
		if ( spec.getNbCpu() != 0 )
			vmSpec.setNumCPUs(spec.getNbCpu());
		if ( spec.getVRAM() != 0 )
			vmSpec.setMemoryMB(Long.valueOf(spec.getVRAM() * 1024));
		return vmSpec;
	}
}