vmware-base/src/main/java/com/cloud/hypervisor/vmware/mo/VirtualMachineDiskInfoBuilder.java

// 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;
    }
}