transcriber/src/main/java/org/opendaylight/neutron/transcriber/NeutronLoadBalancerPoolInterface.java

   1 /*
   2  * Copyright (c) 2014, 2015 Red Hat, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.neutron.transcriber;
  10
  11 import java.util.ArrayList;
  12 import java.util.HashSet;
  13 import java.util.List;
  14 import java.util.Map.Entry;
  15 import java.util.Set;
  16 import java.util.concurrent.ConcurrentHashMap;
  17 import java.util.concurrent.ConcurrentMap;
  18
  19 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
  20 import org.opendaylight.neutron.spi.INeutronLoadBalancerPoolCRUD;
  21 import org.opendaylight.neutron.spi.NeutronLoadBalancerPool;
  22 import org.opendaylight.neutron.spi.NeutronLoadBalancerPoolMember;
  23 import org.opendaylight.neutron.spi.NeutronLoadBalancer_SessionPersistence;
  24 import org.opendaylight.neutron.spi.Neutron_ID;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev160807.ProtocolBase;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev160807.ProtocolHttp;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev160807.ProtocolHttps;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev160807.ProtocolTcp;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev141002.lbaas.attributes.Pool;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev141002.lbaas.attributes.pool.Pools;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev141002.lbaas.attributes.pool.PoolsBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev141002.pool.attributes.SessionPersistenceBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150325.Neutron;
  35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  36 import org.osgi.framework.BundleContext;
  37 import org.osgi.framework.ServiceRegistration;
  38 import org.slf4j.Logger;
  39 import org.slf4j.LoggerFactory;
  40
  41 import com.google.common.collect.ImmutableBiMap;
  42
  43 public class NeutronLoadBalancerPoolInterface extends AbstractNeutronInterface<Pools, NeutronLoadBalancerPool> implements INeutronLoadBalancerPoolCRUD {
  44     private static final Logger LOGGER = LoggerFactory.getLogger(NeutronLoadBalancerPoolInterface.class);
  45     private ConcurrentMap<String, NeutronLoadBalancerPool> loadBalancerPoolDB = new ConcurrentHashMap<String, NeutronLoadBalancerPool>();
  46
  47     private static final ImmutableBiMap<Class<? extends ProtocolBase>,String> PROTOCOL_MAP
  48             = new ImmutableBiMap.Builder<Class<? extends ProtocolBase>,String>()
  49             .put(ProtocolHttp.class,"HTTP")
  50             .put(ProtocolHttps.class,"HTTPS")
  51             .put(ProtocolTcp.class,"TCP")
  52             .build();
  53
  54     NeutronLoadBalancerPoolInterface(ProviderContext providerContext) {
  55         super(providerContext);
  56     }
  57
  58     @Override
  59     public boolean neutronLoadBalancerPoolExists(String uuid) {
  60         return loadBalancerPoolDB.containsKey(uuid);
  61     }
  62
  63     @Override
  64     public NeutronLoadBalancerPool getNeutronLoadBalancerPool(String uuid) {
  65         if (!neutronLoadBalancerPoolExists(uuid)) {
  66             LOGGER.debug("No LoadBalancerPool has Been Defined");
  67             return null;
  68         }
  69         return loadBalancerPoolDB.get(uuid);
  70     }
  71
  72     @Override
  73     public List<NeutronLoadBalancerPool> getAllNeutronLoadBalancerPools() {
  74         Set<NeutronLoadBalancerPool> allLoadBalancerPools = new HashSet<NeutronLoadBalancerPool>();
  75         for (Entry<String, NeutronLoadBalancerPool> entry : loadBalancerPoolDB.entrySet()) {
  76             NeutronLoadBalancerPool loadBalancerPool = entry.getValue();
  77             allLoadBalancerPools.add(loadBalancerPool);
  78         }
  79         LOGGER.debug("Exiting getLoadBalancerPools, Found {} OpenStackLoadBalancerPool", allLoadBalancerPools.size());
  80         List<NeutronLoadBalancerPool> ans = new ArrayList<NeutronLoadBalancerPool>();
  81         ans.addAll(allLoadBalancerPools);
  82         return ans;
  83     }
  84
  85     @Override
  86     public boolean addNeutronLoadBalancerPool(NeutronLoadBalancerPool input) {
  87         if (neutronLoadBalancerPoolExists(input.getID())) {
  88             return false;
  89         }
  90         loadBalancerPoolDB.putIfAbsent(input.getID(), input);
  91         //TODO: add code to find INeutronLoadBalancerPoolAware services and call newtorkCreated on them
  92         return true;
  93     }
  94
  95     @Override
  96     public boolean removeNeutronLoadBalancerPool(String uuid) {
  97         if (!neutronLoadBalancerPoolExists(uuid)) {
  98             return false;
  99         }
 100         loadBalancerPoolDB.remove(uuid);
 101         //TODO: add code to find INeutronLoadBalancerPoolAware services and call newtorkDeleted on them
 102         return true;
 103     }
 104
 105     @Override
 106     public boolean updateNeutronLoadBalancerPool(String uuid, NeutronLoadBalancerPool delta) {
 107         if (!neutronLoadBalancerPoolExists(uuid)) {
 108             return false;
 109         }
 110         NeutronLoadBalancerPool target = loadBalancerPoolDB.get(uuid);
 111         return overwrite(target, delta);
 112     }
 113
 114     @Override
 115     public boolean neutronLoadBalancerPoolInUse(String loadBalancerPoolUUID) {
 116         return !neutronLoadBalancerPoolExists(loadBalancerPoolUUID);
 117     }
 118
 119     @Override
 120     protected Pools toMd(String uuid) {
 121         PoolsBuilder poolsBuilder = new PoolsBuilder();
 122         poolsBuilder.setUuid(toUuid(uuid));
 123         return poolsBuilder.build();
 124     }
 125
 126     @Override
 127     protected InstanceIdentifier<Pools> createInstanceIdentifier(Pools pools) {
 128         return InstanceIdentifier.create(Neutron.class)
 129                 .child(Pool.class)
 130                 .child(Pools.class, pools.getKey());
 131     }
 132
 133     @Override
 134     protected Pools toMd(NeutronLoadBalancerPool pools) {
 135         PoolsBuilder poolsBuilder = new PoolsBuilder();
 136         poolsBuilder.setAdminStateUp(pools.getLoadBalancerPoolAdminIsStateIsUp());
 137         if (pools.getLoadBalancerPoolDescription() != null) {
 138             poolsBuilder.setDescr(pools.getLoadBalancerPoolDescription());
 139         }
 140         if (pools.getNeutronLoadBalancerPoolHealthMonitorID() != null) {
 141             List<Uuid> listHealthMonitor = new ArrayList<Uuid>();
 142             listHealthMonitor.add(toUuid(pools.getNeutronLoadBalancerPoolHealthMonitorID()));
 143             poolsBuilder.setHealthmonitorIds(listHealthMonitor);
 144         }
 145         if (pools.getLoadBalancerPoolLbAlgorithm() != null) {
 146             poolsBuilder.setLbAlgorithm(pools.getLoadBalancerPoolLbAlgorithm());
 147         }
 148         if (pools.getLoadBalancerPoolListeners() != null) {
 149             List<Uuid> listListener = new ArrayList<Uuid>();
 150             for (Neutron_ID neutron_id : pools.getLoadBalancerPoolListeners()) {
 151                 listListener.add(toUuid(neutron_id.getID()));
 152             }
 153             poolsBuilder.setListeners(listListener);
 154         }
 155         if (pools.getLoadBalancerPoolMembers() != null) {
 156             List<Uuid> listMember = new ArrayList<Uuid>();
 157             for (NeutronLoadBalancerPoolMember laodBalancerPoolMember : pools.getLoadBalancerPoolMembers()) {
 158                 listMember.add(toUuid(laodBalancerPoolMember.getID()));
 159
 160             }
 161             poolsBuilder.setMembers(listMember);
 162         }
 163         if (pools.getLoadBalancerPoolName() != null) {
 164             poolsBuilder.setName(pools.getLoadBalancerPoolName());
 165         }
 166         if (pools.getLoadBalancerPoolProtocol() != null) {
 167             ImmutableBiMap<String, Class<? extends ProtocolBase>> mapper =
 168                 PROTOCOL_MAP.inverse();
 169             poolsBuilder.setProtocol((Class<? extends ProtocolBase>) mapper.get(pools.getLoadBalancerPoolProtocol()));
 170         }
 171         if (pools.getLoadBalancerPoolSessionPersistence() != null) {
 172             NeutronLoadBalancer_SessionPersistence sessionPersistence = pools.getLoadBalancerPoolSessionPersistence();
 173             SessionPersistenceBuilder sessionPersistenceBuilder = new SessionPersistenceBuilder();
 174             sessionPersistenceBuilder.setCookieName(sessionPersistence.getCookieName());
 175             sessionPersistenceBuilder.setType(sessionPersistence.getType());
 176             poolsBuilder.setSessionPersistence(sessionPersistenceBuilder.build());
 177         }
 178         if (pools.getLoadBalancerPoolTenantID() != null) {
 179             poolsBuilder.setTenantId(toUuid(pools.getLoadBalancerPoolTenantID()));
 180         }
 181         if (pools.getID() != null) {
 182             poolsBuilder.setUuid(toUuid(pools.getID()));
 183         } else {
 184             LOGGER.warn("Attempting to write neutron load balancer pool without UUID");
 185         }
 186         return poolsBuilder.build();
 187     }
 188
 189     public static void registerNewInterface(BundleContext context,
 190                                             ProviderContext providerContext,
 191                                             List<ServiceRegistration<?>> registrations) {
 192         NeutronLoadBalancerPoolInterface neutronLoadBalancerPoolInterface = new NeutronLoadBalancerPoolInterface(providerContext);
 193         ServiceRegistration<INeutronLoadBalancerPoolCRUD> neutronLoadBalancerPoolInterfaceRegistration = context.registerService(INeutronLoadBalancerPoolCRUD.class, neutronLoadBalancerPoolInterface, null);
 194         if(neutronLoadBalancerPoolInterfaceRegistration != null) {
 195             registrations.add(neutronLoadBalancerPoolInterfaceRegistration);
 196         }
 197     }
 198 }