blob: ee98e3af9f0ce5e6d0fe177c0beaf5ec33165e3a [file] [log] [blame]
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you 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.cloud.hypervisor.vmware.mo;
import org.apache.cloudstack.utils.volume.VirtualMachineDiskInfo;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class VirtualMachineDiskInfoBuilder {
Map<String, List<String>> disks;
public VirtualMachineDiskInfoBuilder() {
disks = new HashMap<String, List<String>>();
}
public void addDisk(String diskDeviceBusName, String diskBackingFilePath) {
List<String> chain = getDiskChainContainer(diskDeviceBusName);
chain.add(diskBackingFilePath);
}
public int getDiskCount() {
return disks.keySet().size();
}
public List<VirtualMachineDiskInfo> getAllDiskInfo() {
List<VirtualMachineDiskInfo> infoList = new ArrayList<VirtualMachineDiskInfo>();
for (Map.Entry<String, List<String>> entry : disks.entrySet()) {
VirtualMachineDiskInfo diskInfo = new VirtualMachineDiskInfo();
diskInfo.setDiskDeviceBusName(entry.getKey());
diskInfo.setDiskChain(entry.getValue().toArray(new String[1]));
infoList.add(diskInfo);
}
return infoList;
}
public VirtualMachineDiskInfo getDiskInfoByDeviceBusName(String diskDeviceBusName) {
List<String> chain = disks.get(diskDeviceBusName);
if (chain != null && chain.size() > 0) {
VirtualMachineDiskInfo diskInfo = new VirtualMachineDiskInfo();
diskInfo.setDiskDeviceBusName(diskDeviceBusName);
diskInfo.setDiskChain(chain.toArray(new String[1]));
return diskInfo;
}
return null;
}
public VirtualMachineDiskInfo getDiskInfoByBackingFileBaseName(String diskBackingFileBaseName, String dataStoreName) {
for (Map.Entry<String, List<String>> entry : disks.entrySet()) {
if (chainContains(entry.getValue(), diskBackingFileBaseName, dataStoreName)) {
VirtualMachineDiskInfo diskInfo = new VirtualMachineDiskInfo();
diskInfo.setDiskDeviceBusName(entry.getKey());
diskInfo.setDiskChain(entry.getValue().toArray(new String[1]));
return diskInfo;
}
}
return null;
}
private List<String> getDiskChainContainer(String diskDeviceBusName) {
assert (diskDeviceBusName != null);
List<String> chain = disks.get(diskDeviceBusName);
if (chain == null) {
chain = new ArrayList<String>();
disks.put(diskDeviceBusName, chain);
}
return chain;
}
private static boolean chainContains(List<String> chain, String diskBackingFileBaseName, String dataStoreName) {
for (String backing : chain) {
DatastoreFile file = new DatastoreFile(backing);
// Ensure matching disk exists in the right datastore
if (file.getFileBaseName().equals(diskBackingFileBaseName) && file.getDatastoreName().equals(dataStoreName))
return true;
}
return false;
}
}