/***************************************************************************
* Copyright (c) 2012-2014 VMware, Inc. All Rights Reserved.
* 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.vmware.aurora.composition;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import javax.xml.bind.JAXBException;
import org.apache.log4j.Logger;
import com.vmware.aurora.util.AuAssert;
import com.vmware.aurora.vc.VcCluster;
import com.vmware.aurora.vc.VcNetwork;
import com.vmware.aurora.vc.VcVirtualMachine;
import com.vmware.aurora.vc.VmConfigUtil;
import com.vmware.vim.binding.impl.vim.vm.ConfigSpecImpl;
import com.vmware.vim.binding.impl.vim.vm.device.VirtualDeviceSpecImpl;
import com.vmware.vim.binding.vim.vm.ConfigSpec;
import com.vmware.vim.binding.vim.vm.device.VirtualDevice;
import com.vmware.vim.binding.vim.vm.device.VirtualDeviceSpec;
import com.vmware.vim.binding.vim.vm.device.VirtualEthernetCard;
/**
* Utility Class for the NetworkSchema
*
* @author sridharr
*
*/
public class NetworkSchemaUtil {
private static final Logger logger = Logger
.getLogger(NetworkSchemaUtil.class);
public static NetworkSchema getSchema(String xmlSchema) throws JAXBException {
return SchemaUtil.getSchema(xmlSchema, NetworkSchema.class);
}
public static NetworkSchema getSchema(File file) throws JAXBException {
return SchemaUtil.getSchema(file, NetworkSchema.class);
}
public static void setNetworkSchema(ConfigSpecImpl spec, VcCluster cluster,
NetworkSchema networkSchema, VcVirtualMachine vcVm) throws Exception {
List<VirtualDeviceSpec> changes = new ArrayList<VirtualDeviceSpec>();
for (NetworkSchema.Network network : networkSchema.networks) {
VcNetwork vN = cluster.getNetwork(network.vcNetwork);
if (vN == null) {
logger.error("Network " + network.vcNetwork + " is not defined on cluster " + cluster.getName());
throw new Exception("Network " + network.vcNetwork + " is not defined on cluster " + cluster.getName());
}
if (network.nicLabel != null
&& vcVm.getDeviceByLabel(network.nicLabel) != null) {
// Edit existing networks
changes.add(vcVm.reconfigNetworkSpec(network.nicLabel, vN));
} else {
// Add new networks
VirtualDeviceSpec deviceSpec =
VmConfigUtil.createNetworkDevice(
VmConfigUtil.EthernetControllerType.VMXNET3,
network.nicLabel, vN);
changes.add(deviceSpec);
}
}
spec.setDeviceChange(changes.toArray(new VirtualDeviceSpec[changes.size()]));
}
public static void copyMacAddresses(ConfigSpec configSpec,
VcVirtualMachine parentVm, VcVirtualMachine childVm,
NetworkSchema networkSchema) {
List<VirtualDeviceSpec> changes = new ArrayList<VirtualDeviceSpec>();
for (NetworkSchema.Network network : networkSchema.networks) {
VirtualDevice nic = null;
String macAddr = null;
if (network.nicLabel != null) {
nic = parentVm.getDeviceByLabel(network.nicLabel);
if (nic == null) {
continue;
}
macAddr = ((VirtualEthernetCard) nic).getMacAddress();
if (macAddr == null) {
continue;
}
logger.info("get parent vm's mac address " + macAddr);
// edit mac address
nic = childVm.getDeviceByLabel(network.nicLabel);
if (nic == null) {
logger.info("child vm does not have nic " + network.nicLabel);
continue;
}
// set mac address
((VirtualEthernetCard) nic).setAddressType("Manual");
((VirtualEthernetCard) nic).setMacAddress(macAddr);
VirtualDeviceSpec deviceSpec = new VirtualDeviceSpecImpl();
deviceSpec.setOperation(VirtualDeviceSpec.Operation.edit);
deviceSpec.setDevice(nic);
changes.add(deviceSpec);
}
}
configSpec.setDeviceChange(changes.toArray(new VirtualDeviceSpec[changes
.size()]));
}
}