content: // 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.network.element;
  
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  import java.util.Set;
  
  import javax.inject.Inject;
  import javax.naming.ConfigurationException;
  
  import org.springframework.stereotype.Component;
  
  import org.apache.cloudstack.framework.config.dao.ConfigurationDao;
  
  import com.cloud.agent.api.to.LoadBalancerTO;
  import com.cloud.configuration.Config;
  import com.cloud.deploy.DeployDestination;
  import com.cloud.exception.ConcurrentOperationException;
  import com.cloud.exception.InsufficientCapacityException;
  import com.cloud.exception.ResourceUnavailableException;
  import com.cloud.network.Network;
  import com.cloud.network.Network.Capability;
  import com.cloud.network.Network.Provider;
  import com.cloud.network.Network.Service;
  import com.cloud.network.NetworkModel;
  import com.cloud.network.Networks.TrafficType;
  import com.cloud.network.PhysicalNetworkServiceProvider;
  import com.cloud.network.PublicIpAddress;
  import com.cloud.network.dao.NetworkDao;
  import com.cloud.network.lb.ElasticLoadBalancerManager;
  import com.cloud.network.lb.LoadBalancingRule;
  import com.cloud.network.rules.LoadBalancerContainer;
  import com.cloud.offering.NetworkOffering;
  import com.cloud.offerings.dao.NetworkOfferingDao;
  import com.cloud.utils.component.AdapterBase;
  import com.cloud.vm.NicProfile;
  import com.cloud.vm.ReservationContext;
  import com.cloud.vm.VirtualMachineProfile;
  
  @Component
  public class ElasticLoadBalancerElement extends AdapterBase implements LoadBalancingServiceProvider, IpDeployer {
      private static final Map<Service, Map<Capability, String>> capabilities = setCapabilities();
      @Inject
      NetworkModel _networkManager;
      @Inject
      ElasticLoadBalancerManager _lbMgr;
      @Inject
      ConfigurationDao _configDao;
      @Inject
      NetworkOfferingDao _networkOfferingDao;
      @Inject
      NetworkDao _networksDao;
  
      boolean _enabled;
      TrafficType _frontEndTrafficType = TrafficType.Guest;
  
      private boolean canHandle(Network network, List<LoadBalancingRule> rules) {
          if (network.getGuestType() != Network.GuestType.Shared || network.getTrafficType() != TrafficType.Guest) {
              logger.debug("Not handling network {} with type {} and traffic type {}", network, network.getGuestType(), network.getTrafficType());
              return false;
          }
  
          Map<Capability, String> lbCaps = this.getCapabilities().get(Service.Lb);
          if (!lbCaps.isEmpty()) {
              String schemeCaps = lbCaps.get(Capability.LbSchemes);
              if (schemeCaps != null) {
                  for (LoadBalancingRule rule : rules) {
                      if (!schemeCaps.contains(rule.getScheme().toString())) {
                          logger.debug("Scheme " + rules.get(0).getScheme() + " is not supported by the provider " + this.getName());
                          return false;
                      }
                  }
              }
          }
  
          return true;
      }
  
      @Override
      public Provider getProvider() {
          return Provider.ElasticLoadBalancerVm;
      }
  
      @Override
      public Map<Service, Map<Capability, String>> getCapabilities() {
          return capabilities;
      }
  
      private static Map<Service, Map<Capability, String>> setCapabilities() {
          Map<Service, Map<Capability, String>> capabilities = new HashMap<Service, Map<Capability, String>>();
  
          Map<Capability, String> lbCapabilities = new HashMap<Capability, String>();
          lbCapabilities.put(Capability.SupportedLBAlgorithms, "roundrobin,leastconn,source");
          lbCapabilities.put(Capability.SupportedLBIsolation, "shared");
          lbCapabilities.put(Capability.SupportedProtocols, "tcp, udp");
          lbCapabilities.put(Capability.LbSchemes, LoadBalancerContainer.Scheme.Public.toString());
  
          capabilities.put(Service.Lb, lbCapabilities);
          return capabilities;
      }
  
      @Override
      public boolean implement(Network network, NetworkOffering offering, DeployDestination dest, ReservationContext context) throws ConcurrentOperationException,
          ResourceUnavailableException, InsufficientCapacityException {
  
          return true;
      }
  
      @Override
      public boolean prepare(Network network, NicProfile nic, VirtualMachineProfile vm, DeployDestination dest, ReservationContext context)
          throws ConcurrentOperationException, ResourceUnavailableException, InsufficientCapacityException {
  
          return true;
      }
  
      @Override
      public boolean release(Network network, NicProfile nic, VirtualMachineProfile vm, ReservationContext context) throws ConcurrentOperationException,
          ResourceUnavailableException {
  
          return true;
      }
  
      @Override
      public boolean shutdown(Network network, ReservationContext context, boolean cleanup) throws ConcurrentOperationException, ResourceUnavailableException {
          // TODO kill all loadbalancer vms by calling the ElasticLoadBalancerManager
          return false;
      }
  
      @Override
      public boolean destroy(Network network, ReservationContext context) throws ConcurrentOperationException, ResourceUnavailableException {
          // TODO  kill all loadbalancer vms by calling the ElasticLoadBalancerManager
          return false;
      }
  
      @Override
      public boolean validateLBRule(Network network, LoadBalancingRule rule) {
          return true;
      }
  
      @Override
      public boolean applyLBRules(Network network, List<LoadBalancingRule> rules) throws ResourceUnavailableException {
          if (!canHandle(network, rules)) {
              return false;
          }
  
          return _lbMgr.applyLoadBalancerRules(network, rules);
      }
  
      @Override
      public boolean configure(String name, Map<String, Object> params) throws ConfigurationException {
  
          super.configure(name, params);
          String enabled = _configDao.getValue(Config.ElasticLoadBalancerEnabled.key());
          _enabled = (enabled == null) ? false : Boolean.parseBoolean(enabled);
          if (_enabled) {
              String traffType = _configDao.getValue(Config.ElasticLoadBalancerNetwork.key());
              if ("guest".equalsIgnoreCase(traffType)) {
                  _frontEndTrafficType = TrafficType.Guest;
              } else if ("public".equalsIgnoreCase(traffType)) {
                  _frontEndTrafficType = TrafficType.Public;
              } else
                  throw new ConfigurationException("Traffic type for front end of load balancer has to be guest or public; found : " + traffType);
          }
          return true;
      }
  
      @Override
      public boolean isReady(PhysicalNetworkServiceProvider provider) {
          // TODO Auto-generated method stub
          return true;
      }
  
      @Override
      public boolean shutdownProviderInstances(PhysicalNetworkServiceProvider provider, ReservationContext context) throws ConcurrentOperationException,
          ResourceUnavailableException {
          // TODO Auto-generated method stub
          return true;
      }
  
      @Override
      public boolean canEnableIndividualServices() {
          return false;
      }
  
      @Override
      public boolean verifyServicesCombination(Set<Service> services) {
          return true;
      }
  
      @Override
      public boolean applyIps(Network network, List<? extends PublicIpAddress> ipAddress, Set<Service> services) throws ResourceUnavailableException {
          // TODO Auto-generated method stub
          return false;
      }
  
      @Override
      public IpDeployer getIpDeployer(Network network) {
          return this;
      }
  
      @Override
      public List<LoadBalancerTO> updateHealthChecks(Network network, List<LoadBalancingRule> lbrules) {
          return null;
      }
  
      @Override
      public boolean handlesOnlyRulesInTransitionState() {
          return true;
      }
  
      @Override
      public void expungeLbVmRefs(List<Long> vmIds, Long batchSize) {
          _lbMgr.expungeLbVmRefs(vmIds, batchSize);
      }
  }
  