VcHost.java

/***************************************************************************
 * Copyright (c) 2012-2015 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.vc;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import com.vmware.aurora.util.AuAssert;
import com.vmware.vim.binding.vim.ClusterComputeResource;
import com.vmware.vim.binding.vim.HostSystem;
import com.vmware.vim.binding.vim.HostSystem.ConnectionState;
import com.vmware.vim.binding.vim.host.RuntimeInfo;
import com.vmware.vim.binding.vmodl.ManagedObject;
import com.vmware.vim.binding.vmodl.ManagedObjectReference;

public interface VcHost extends VcObject {

   abstract String getName();

   /**
    * @return the cluster of this host
    * @throws Exception
    */
   abstract VcCluster getCluster();

   /**
    * @return the VMs on this host
    * @throws Exception
    */
   abstract List<VcVirtualMachine> getVMs() throws Exception;

   /**
    * @return a list of networks on this host
    */
   abstract List<VcNetwork> getNetworks();

   /**
    * Return a list of all datastores (shared by all hosts and not).
    * @return all datastores
    * @throws Exception
    */
   abstract List<VcDatastore> getDatastores();

   abstract boolean isConnected();
   abstract boolean isInMaintenanceMode();
   abstract boolean isUnavailbleForManagement();
   abstract String  getVersion();
   abstract long getCpuHz();
}

@SuppressWarnings("serial")
class VcHostImpl extends VcObjectImpl implements VcHost {
   private String name;

   // either a folder or datacenter
   private ManagedObjectReference parent;
   private ManagedObjectReference[] network;
   private ManagedObjectReference[] datastore;
   private RuntimeInfo runtime;
   private String version;
   private long cpuHz;

   protected VcHostImpl(HostSystem host) throws Exception {
      super(host);
      update(host);
   }

   @Override
   protected void update(ManagedObject mo) throws Exception {
      AuAssert.check(this.moRef.equals(mo._getRef()));
      HostSystem host = (HostSystem)mo;
      name = host.getName();
      parent = checkReady(host.getParent());
      network = checkReady(host.getNetwork());
      datastore = checkReady(host.getDatastore());
      runtime = checkReady(host.getRuntime());
      if (host.getConfig() != null && host.getConfig().getProduct() != null) {
         version = host.getConfig().getProduct().getVersion();
      }
      cpuHz = host.getHardware().getCpuInfo().getHz();
   }

   /* (non-Javadoc)
    * @see com.vmware.aurora.vc.VcCluster#getName()
    */
   @Override
   public String getName() {
      return MoUtil.fromURLString(name);
   }

   @Override
   public String toString() {
      return String.format("HOST[%s]()",
            name);
   }

   @Override
   public int hashCode() {
      final int prime = 31;
      int result = super.hashCode();
      result = prime * result + ((name == null) ? 0 : name.hashCode());
      result = prime * result + ((version == null) ? 0 : version.hashCode());
      return result;
   }

   @Override
   public boolean equals(Object obj) {
      if (this == obj)
         return true;
      if (!super.equals(obj))
         return false;
      if (getClass() != obj.getClass())
         return false;
      VcHostImpl other = (VcHostImpl) obj;
      if (name == null) {
         if (other.name != null)
            return false;
      } else if (!name.equals(other.name))
         return false;
      if (version == null) {
         if (other.version != null)
            return false;
      } else if (!version.equals(other.version))
         return false;
      return true;
   }

   @Override
   public List<VcNetwork> getNetworks() {
      List<VcNetwork> results = new ArrayList<VcNetwork>();
      for (ManagedObjectReference ref : network) {
         VcNetwork net = VcCache.get(ref);
         if (!net.isUplink()) {
            results.add(net);
         }
      }
      return results;
   }



   @Override
   public List<VcDatastore> getDatastores() {
      return VcCache.<VcDatastore>getPartialList(Arrays.asList(datastore), getMoRef());
   }




   @Override
   public VcCluster getCluster() {
      return VcCache.get(
            MoUtil.getAncestorMoRef(parent, ClusterComputeResource.class));
   }



   @Override
   public List<VcVirtualMachine> getVMs() throws Exception {
      return null;
   }

   @Override
   public boolean isConnected() {
      return ConnectionState.connected.equals(runtime.getConnectionState());
   }

   @Override
   public boolean isInMaintenanceMode() {
      return runtime.isInMaintenanceMode();
   }

   @Override
   public boolean isUnavailbleForManagement() {
      return (runtime.isInMaintenanceMode() || !isConnected());
   }

   @Override
   public String getVersion() {
      return version;
   }

   @Override
   public long getCpuHz() {

      return cpuHz;
   }
}