natservice/impl/src/main/java/org/opendaylight/netvirt/natservice/internal/SnatServiceManagerImpl.java

   1 /*
   2  * Copyright (c) 2017 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.internal;
   9
  10 import java.math.BigInteger;
  11 import java.util.List;
  12 import java.util.concurrent.CopyOnWriteArrayList;
  13 import javax.inject.Inject;
  14 import javax.inject.Singleton;
  15 import org.opendaylight.genius.infra.Datastore;
  16 import org.opendaylight.genius.infra.TypedReadWriteTransaction;
  17 import org.opendaylight.netvirt.natservice.api.SnatServiceListener;
  18 import org.opendaylight.netvirt.natservice.api.SnatServiceManager;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.ext.routers.Routers;
  20 import org.slf4j.Logger;
  21 import org.slf4j.LoggerFactory;
  22
  23 @Singleton
  24 public class SnatServiceManagerImpl implements SnatServiceManager {
  25
  26     private static final Logger LOG = LoggerFactory.getLogger(SnatServiceManagerImpl.class);
  27
  28     private final List<SnatServiceListener> snatServiceListeners = new CopyOnWriteArrayList<>();
  29
  30     @Inject
  31     public SnatServiceManagerImpl(final SnatServiceImplFactory factory) {
  32         AbstractSnatService flatVlaSnatServiceImpl = factory.createFlatVlanSnatServiceImpl();
  33         if (flatVlaSnatServiceImpl != null) {
  34             addNatServiceListener(flatVlaSnatServiceImpl);
  35         }
  36         AbstractSnatService vxlanGreSnatServiceImpl = factory.createVxlanGreSnatServiceImpl();
  37         if (vxlanGreSnatServiceImpl != null) {
  38             addNatServiceListener(vxlanGreSnatServiceImpl);
  39         }
  40     }
  41
  42     @Override
  43     public void addNatServiceListener(SnatServiceListener natServiceListner) {
  44         snatServiceListeners.add(natServiceListner);
  45     }
  46
  47     @Override
  48     public void removeNatServiceListener(SnatServiceListener natServiceListner) {
  49         snatServiceListeners.remove(natServiceListner);
  50     }
  51
  52     @Override
  53     public void notify(TypedReadWriteTransaction<Datastore.Configuration> confTx,
  54         Routers router, BigInteger primarySwitchId, BigInteger dpnId, Action action) {
  55         for (SnatServiceListener snatServiceListener : snatServiceListeners) {
  56             boolean result = false;
  57             switch (action) {
  58                 case SNAT_ALL_SWITCH_ENBL:
  59                     result = snatServiceListener.addSnatAllSwitch(confTx, router, primarySwitchId);
  60                     break;
  61
  62                 case SNAT_ALL_SWITCH_DISBL:
  63                     result = snatServiceListener.removeSnatAllSwitch(confTx, router, primarySwitchId);
  64                     break;
  65
  66                 case SNAT_ROUTER_ENBL:
  67                     result = snatServiceListener.addSnat(confTx, router, primarySwitchId, dpnId);
  68                     break;
  69
  70                 case SNAT_ROUTER_DISBL:
  71                     result = snatServiceListener.removeSnat(confTx, router, primarySwitchId, dpnId);
  72                     break;
  73
  74                 default:
  75                     break;
  76             }
  77
  78             if (result) {
  79                 LOG.debug("notify : Nat action {} invoking listener {} succeeded", action,
  80                     snatServiceListener.getClass().getName());
  81             } else {
  82                 LOG.warn("notify : Nat action {} invoking listener {} failed",
  83                         action, snatServiceListener.getClass().getName());
  84             }
  85         }
  86     }
  87
  88 }