Vijava (更改虚拟机配置信息（内存、CPU和添加新磁盘）终极版本)

package com.vmware.server;

import java.util.List;
import java.util.Map;
import org.apache.log4j.FileAppender;
import org.apache.log4j.Logger;
import org.apache.log4j.SimpleLayout;
import com.vmware.pojo.VirtualMachineConfig;
import com.vmware.util.DataBase;
import com.vmware.util.Session;
import com.vmware.util.StringRandom;
import com.vmware.util.VmwareUtil;
import com.vmware.vim25.VirtualDevice;
import com.vmware.vim25.VirtualDeviceConfigSpec;
import com.vmware.vim25.VirtualDeviceConfigSpecFileOperation;
import com.vmware.vim25.VirtualDeviceConfigSpecOperation;
import com.vmware.vim25.VirtualDisk;
import com.vmware.vim25.VirtualDiskFlatVer2BackingInfo;
import com.vmware.vim25.VirtualMachineConfigSpec;
import com.vmware.vim25.VirtualMachinePowerState;
import com.vmware.vim25.mo.Folder;
import com.vmware.vim25.mo.InventoryNavigator;
import com.vmware.vim25.mo.ServiceInstance;
import com.vmware.vim25.mo.Task;
import com.vmware.vim25.mo.VirtualMachine;

/**
 * VirtualMachineConfigServer:虚拟机配置更改服务
 **/

public class VirtualMachineConfigServer {
	public static Logger logger = Logger
			.getLogger(VirtualMachineConfigServer.class);

	//
	public void vmConfig(List<VirtualMachineConfig> list) {
		// 自定义工具栏
		VmwareUtil util = new VmwareUtil();
		// 日志处理
		SimpleLayout layout = new SimpleLayout();
		FileAppender appender = null;
		try {
			appender = new FileAppender(layout, "log.txt", false);
		} catch (Exception e) {
		}
		logger.addAppender(appender);
		// 逻辑处理
		try {
			Map map = util.getURL();
			String ip = String.valueOf(map.get("ip"));
			String username = String.valueOf(map.get("username"));
			String password = String.valueOf(map.get("password"));

			if (list.size() > 0 && list != null) {
				for (VirtualMachineConfig config : list) {
					ServiceInstance instance = Session.getInstance(ip,
							username, password);
					if (instance != null) {
						logger.debug("vmware 连接成功");
					} else {
						logger.debug("vmware 连接失败,请检查vmware连接相关配置信息");
					}
					Folder rootFolder = instance.getRootFolder();
					InventoryNavigator inventoryNavigator = new InventoryNavigator(
							rootFolder);
					try {
						com.vmware.vim25.mo.VirtualMachine VM = (com.vmware.vim25.mo.VirtualMachine) inventoryNavigator
								.searchManagedEntity("VirtualMachine",
										config.getXnjName());
						if (VM != null) {
							logger.debug("虚拟机 查询成功");
						} else {
							logger.debug("虚拟机 查询失败，请仔细检查配置模板是否存在");
						}
						if(!VM.getRuntime().getPowerState().name().equalsIgnoreCase(VirtualMachinePowerState.poweredOff.name())){
							Task tasks = VM.powerOffVM_Task();
							tasks.waitForTask();							
						}
						// 修改虚拟机相关配置属性
						VirtualMachineConfigSpec configSpec = new VirtualMachineConfigSpec();
						// 设置CPU核数
						configSpec
								.setNumCPUs(Integer.parseInt(config.getCpu()));
						// 设置内存大小
						configSpec.setMemoryMB((long) Integer.parseInt(config
								.getRam()));
						  int diskSize = Integer.parseInt(config.getRl())*1024*1024;
				            // mode: persistent|independent_persistent,independent_nonpersistent
				            String diskMode = "persistent";				         
				            VirtualDeviceConfigSpec vdiskSpec = createDiskSpec(config.getDsName(),diskSize, diskMode,VM);
				            VirtualDeviceConfigSpec[] vdiskSpecArray = {vdiskSpec};
				            configSpec.setDeviceChange(vdiskSpecArray);

					
						Task task = VM.reconfigVM_Task(configSpec);
						String result = task.waitForTask();
						if (result.equals(Task.SUCCESS)) {
							logger.debug("模板生成虚拟机成功");
							logger.debug("模板生成虚拟机成功");
							System.out.println("flwId:"+config.getFlwId());
							String sql = "update tb_flw_content set `status`='3' where `flw_id`='"
									+ config.getFlwId() + "'";
							DataBase.update(sql);
							String deleteAll = "update tb_flw_content set `status`='6' where `flw_id`='"
									+ config.getFlwId() + "'";
							DataBase.update(deleteAll);
						} else {
							logger.debug("模板生成虚拟机失败，请查询Vcenter 上相关日志信息");
						}
						instance.getServerConnection().logout();

					} catch (Exception e) {
						logger.debug("template 查询失败，请仔细检查配置模板是否存在");
						instance.getServerConnection().logout();
						break;
					}
				}

			}
		} catch (Exception e) {
		}
	}

	public static VirtualDeviceConfigSpec createDiskSpec(String dsName, long diskSizeKB, String diskMode, VirtualMachine vm)
			 {
		VirtualDeviceConfigSpec diskSpec = new VirtualDeviceConfigSpec();
		diskSpec.setOperation(VirtualDeviceConfigSpecOperation.add);
		diskSpec.setFileOperation(VirtualDeviceConfigSpecFileOperation.create);
		VirtualDisk vd = new VirtualDisk();
		vd.setCapacityInKB(diskSizeKB);
		diskSpec.setDevice(vd);
		vd.setKey(getkey(vm));
		vd.setUnitNumber(getUnitNumber(vm));
		vd.setControllerKey(getcontrollerkey(vm));
		VirtualDiskFlatVer2BackingInfo diskfileBacking = new VirtualDiskFlatVer2BackingInfo();
		String diskName=StringRandom.getStringRandom(12);
		String fileName = "["+ dsName +"] "+ vm.getName() + "/" + diskName + ".vmdk";
		diskfileBacking.setFileName(fileName);
		diskfileBacking.setDiskMode(diskMode);
		diskfileBacking.setThinProvisioned(true);
		vd.setBacking(diskfileBacking);
		return diskSpec;
	}

	// 获取虚拟机磁盘管理的controlerkey
	public static int getcontrollerkey(VirtualMachine vm) {
		int controllerkey = 0;
		if (vm != null) {
			VirtualDevice[] devices = vm.getConfig().getHardware().getDevice();
			if (devices != null && devices.length > 0) {
				for (int i = 0; i < devices.length; i++) {
					VirtualDevice device = devices[i];
					if (device instanceof VirtualDisk) {
						controllerkey = device.getControllerKey();
					}
				}
			}
		}
		return controllerkey;
	}

	// 获取虚拟机已生成key
	public static int getkey(VirtualMachine vm) {
		int key = 0;
		if (vm != null) {
			VirtualDevice[] devices = vm.getConfig().getHardware().getDevice();
			if (devices != null && devices.length > 0) {
				for (int i = 0; i < devices.length; i++) {
					VirtualDevice device = devices[i];
					if (device instanceof VirtualDisk) {
						key = device.getKey();
					}
				}
			}
		}
		key = key + 1;
		return key;
	}

	// 获取虚拟机已生成UnitNumber
	public static int getUnitNumber(VirtualMachine vm) {
		int UnitNumber = 0;
		if (vm != null) {
			VirtualDevice[] devices = vm.getConfig().getHardware().getDevice();
			if (devices != null && devices.length > 0) {
				for (int i = 0; i < devices.length; i++) {
					VirtualDevice device = devices[i];
					if (device instanceof VirtualDisk) {
						UnitNumber = device.getUnitNumber();
					}
				}
			}
		}
		UnitNumber = UnitNumber + 1;
		return UnitNumber;
	}

}