natservice/impl/src/main/java/org/opendaylight/netvirt/natservice/ha/NatSwitchCacheImpl.java

   1 /*
   2  * Copyright (c) 2018 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 package org.opendaylight.netvirt.natservice.ha;
   9
  10 import java.math.BigInteger;
  11 import java.util.ArrayList;
  12 import java.util.HashSet;
  13 import java.util.List;
  14 import java.util.Map;
  15 import java.util.Set;
  16 import java.util.concurrent.ConcurrentHashMap;
  17 import java.util.concurrent.ConcurrentMap;
  18
  19 import javax.inject.Inject;
  20 import javax.inject.Singleton;
  21
  22 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  23 import org.opendaylight.netvirt.natservice.api.NatSwitchCache;
  24 import org.opendaylight.netvirt.natservice.api.NatSwitchCacheListener;
  25 import org.opendaylight.netvirt.natservice.api.SwitchInfo;
  26 import org.opendaylight.netvirt.natservice.internal.NatUtil;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29
  30 @Singleton
  31 public class NatSwitchCacheImpl implements NatSwitchCache {
  32
  33     private static final Logger LOG = LoggerFactory.getLogger(NatSwitchCacheImpl.class);
  34     ConcurrentMap<BigInteger,SwitchInfo> switchMap = new ConcurrentHashMap<>();
  35
  36     private final  List<NatSwitchCacheListener> centralizedSwitchCacheListenerList =
  37             new ArrayList<NatSwitchCacheListener>();
  38     private final DataBroker dataBroker;
  39
  40     @Inject
  41     public NatSwitchCacheImpl(final DataBroker dataBroker) {
  42         this.dataBroker = dataBroker;
  43     }
  44
  45     @Override
  46     public void addSwitch(BigInteger dpnId) {
  47         LOG.info("addSwitch: Retrieving the provider config for {}", dpnId);
  48         Map<String, String> providerMappingsMap = NatUtil.getOpenvswitchOtherConfigMap(dpnId, dataBroker);
  49         SwitchInfo switchInfo = new SwitchInfo();
  50         switchInfo.setDpnId(dpnId);
  51         switchInfo.setProviderNets(providerMappingsMap.keySet());
  52         switchMap.put(dpnId, switchInfo);
  53         for (NatSwitchCacheListener centralizedSwitchCacheListener : centralizedSwitchCacheListenerList) {
  54             centralizedSwitchCacheListener.switchAddedToCache(switchInfo);
  55         }
  56     }
  57
  58     @Override
  59     public void removeSwitch(BigInteger dpnId) {
  60         LOG.info("removeSwitch: Removing {} dpnId to switchWeightsMap", dpnId);
  61         SwitchInfo switchInfo = switchMap.get(dpnId);
  62         for (NatSwitchCacheListener centralizedSwitchCacheListener : centralizedSwitchCacheListenerList) {
  63             centralizedSwitchCacheListener.switchRemovedFromCache(switchInfo);
  64         }
  65     }
  66
  67     @Override
  68     public boolean isSwitchConnectedToExternal(BigInteger dpnId, String providerNet) {
  69         SwitchInfo switchInfo = switchMap.get(dpnId);
  70         if (switchInfo != null) {
  71             return switchInfo.getProviderNets().contains(providerNet);
  72         }
  73         return false;
  74     }
  75
  76     @Override
  77     public Set<BigInteger> getSwitchesConnectedToExternal(String providerNet) {
  78         Set<BigInteger> switches = new HashSet<>();
  79         for (Map.Entry<BigInteger,SwitchInfo> switchesEntrySet : switchMap.entrySet()) {
  80             Set<String> providerNetSet = switchesEntrySet.getValue().getProviderNets();
  81             if (providerNetSet != null && providerNetSet.contains(providerNet)) {
  82                 switches.add(switchesEntrySet.getKey());
  83             }
  84         }
  85         return switches;
  86     }
  87
  88     public void register(NatSwitchCacheListener centralizedSwitchCacheListener) {
  89         if (centralizedSwitchCacheListener != null) {
  90             centralizedSwitchCacheListenerList.add(centralizedSwitchCacheListener);
  91         }
  92     }
  93
  94     public void deregister(NatSwitchCacheListener centralizedSwitchCacheListener) {
  95         if (centralizedSwitchCacheListener != null) {
  96             centralizedSwitchCacheListenerList.remove(centralizedSwitchCacheListener);
  97         }
  98     }
  99
 100     @Override
 101     public Map<BigInteger,SwitchInfo> getSwitches() {
 102         return switchMap;
 103     }
 104 }