bgp/openconfig-impl/src/main/java/org/opendaylight/protocol/bgp/openconfig/impl/moduleconfig/PathSelectionModeFunction.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, 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.protocol.bgp.openconfig.impl.moduleconfig;
  10
  11 import com.google.common.base.Function;
  12 import com.google.common.base.Optional;
  13 import com.google.common.base.Predicate;
  14 import com.google.common.collect.FluentIterable;
  15 import com.google.common.collect.ImmutableList;
  16 import com.google.common.collect.ImmutableMap;
  17 import com.google.common.collect.ImmutableMap.Builder;
  18 import com.google.common.collect.Iterables;
  19 import java.util.ArrayList;
  20 import java.util.Collections;
  21 import java.util.HashMap;
  22 import java.util.List;
  23 import java.util.Map;
  24 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  25 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
  26 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  27 import org.opendaylight.protocol.bgp.openconfig.impl.util.OpenConfigUtil;
  28 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
  29 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.use.multiple.paths.use.multiple.paths.Config;
  30 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.use.multiple.paths.use.multiple.paths.Ibgp;
  31 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.AfiSafi1;
  32 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.path.selection.mode.rev160301.AdvertiseAllPaths;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.path.selection.mode.rev160301.AdvertiseNPaths;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.path.selection.mode.rev160301.PathSelectionModeFactory;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev160330.BgpPathSelectionMode;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev160330.BgpPsmImpl;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.ModuleType;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.ServiceRef;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.modules.Module;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.modules.ModuleKey;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.services.Service;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.services.ServiceKey;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.services.service.Instance;
  45 import org.opendaylight.yangtools.yang.binding.ChildOf;
  46
  47 final class PathSelectionModeFunction {
  48
  49     private static final Map<String, Class<? extends ModuleType>> PATH_SELECTION_MODULE_TYPES;
  50
  51     static {
  52         final Builder<String, Class<? extends ModuleType>> builder = ImmutableMap.builder();
  53         builder.put(AdvertiseNPaths.QNAME.getLocalName(), AdvertiseNPaths.class);
  54         builder.put(AdvertiseAllPaths.QNAME.getLocalName(), AdvertiseAllPaths.class);
  55         PATH_SELECTION_MODULE_TYPES = builder.build();
  56     }
  57
  58     private PathSelectionModeFunction() {
  59         throw new UnsupportedOperationException();
  60     }
  61
  62     public static <T extends ServiceRef & ChildOf<Module>> List<T> getPathSelectionMode(final ReadOnlyTransaction rTx, final BGPConfigModuleProvider
  63             configModuleWriter, final Function<String, T> function, final List<AfiSafi> afiSafis) {
  64         final ImmutableList<AfiSafi> afiSafisMultipath = FluentIterable.from(afiSafis).filter(new Predicate<AfiSafi>() {
  65             @Override
  66             public boolean apply(final AfiSafi afisafi) {
  67                 final AfiSafi1 afiSafi1 = afisafi.getAugmentation(AfiSafi1.class);
  68                 if (afiSafi1 != null) {
  69                     final org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.use.multiple.paths.UseMultiplePaths useMultiplePaths = afiSafi1.getUseMultiplePaths();
  70                     if (useMultiplePaths != null) {
  71                         final Config useMultiplePathsConfig = useMultiplePaths.getConfig();
  72                         if (useMultiplePathsConfig != null && useMultiplePathsConfig.isEnabled() != null) {
  73                             return useMultiplePathsConfig.isEnabled();
  74                         }
  75                     }
  76                 }
  77                 return false;
  78             }}).toList();
  79         try {
  80             final Optional<Service> maybeService = configModuleWriter.readConfigService(new ServiceKey(BgpPathSelectionMode.class), rTx);
  81             if (maybeService.isPresent()) {
  82                 final Service service = maybeService.get();
  83                 final List<Module> modules = new ArrayList<>();
  84                 final Map<String, String> moduleNameToService = new HashMap<>();
  85                 for (final Instance instance : service.getInstance()) {
  86                     final String moduleName = OpenConfigUtil.getModuleName(instance.getProvider());
  87                     final ModuleKey moduleKey = new ModuleKey(moduleName, BgpPsmImpl.class);
  88                     final Optional<Module> moduleConfig = configModuleWriter.readModuleConfiguration(moduleKey, rTx);
  89                     if (moduleConfig.isPresent()) {
  90                         modules.add(moduleConfig.get());
  91                     }
  92                     moduleNameToService.put(moduleName, instance.getName());
  93                 }
  94
  95                 return TableTypesFunction.toServices(function, afiSafisMultipath, afiSafiToModuleName(afiSafisMultipath, modules, configModuleWriter, rTx), moduleNameToService);
  96             }
  97             return Collections.emptyList();
  98         } catch (final ReadFailedException e) {
  99             throw new IllegalStateException(OpenConfigUtil.FAILED_TO_READ_SERVICE, e);
 100         }
 101     }
 102
 103     private static Class<? extends ModuleType> getModuleTypeClass(final String moduleType) {
 104         return PATH_SELECTION_MODULE_TYPES.get(moduleType);
 105     }
 106
 107     private static Map<Class<? extends AfiSafiType>, String> afiSafiToModuleName(final List<AfiSafi> afiSafis, final List<Module> modules, final BGPConfigModuleProvider configModuleWriter, final ReadTransaction rTx) throws ReadFailedException {
 108         final Map<Class<? extends AfiSafiType>, String> afiSafiToModuleName = new HashMap<>(afiSafis.size());
 109         for (final Module module : modules) {
 110             final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev160330.modules.module.configuration.BgpPsmImpl config =
 111                     ((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev160330.modules.module.configuration.BgpPsmImpl) module.getConfiguration());
 112
 113             final Optional<AfiSafi> tryFind = Iterables.tryFind(afiSafis, new Predicate<AfiSafi>() {
 114                 @Override
 115                 public boolean apply(final AfiSafi input) {
 116                     final Optional<org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType> bgpTableType = OpenConfigUtil.toBgpTableType(input.getAfiSafiName());
 117                     if (bgpTableType.isPresent() && AddPathFunction.tableTypeExists(configModuleWriter, rTx,
 118                             (String) config.getPathAddressFamily().getName(), bgpTableType.get())) {
 119                         final Ibgp ibgp = input.getAugmentation(AfiSafi1.class).getUseMultiplePaths().getIbgp();
 120                         return pathSelectionModeExists(configModuleWriter, rTx, (String) config.getPathSelectionMode().getName(), ibgp);
 121                     }
 122                     return false;
 123                 }
 124             });
 125             if (tryFind.isPresent()) {
 126                 afiSafiToModuleName.put(tryFind.get().getAfiSafiName(), module.getName());
 127             }
 128         }
 129         return afiSafiToModuleName;
 130     }
 131
 132     public static boolean pathSelectionModeExists(final BGPConfigModuleProvider configModuleWriter, final ReadTransaction rTx,
 133             final String instanceName, final Ibgp ibgp) {
 134
 135         try {
 136             final Optional<Service> maybeService = configModuleWriter.readConfigService(new ServiceKey(PathSelectionModeFactory.class), rTx);
 137             if (maybeService.isPresent()) {
 138                 for (final Instance instance : maybeService.get().getInstance()) {
 139                     final String provider = instance.getProvider();
 140                     final String moduleName = OpenConfigUtil.getModuleName(provider);
 141                     final String moduleType = OpenConfigUtil.getModuleType(provider);
 142                     if (moduleName.equals(instanceName)) {
 143                         final ModuleKey moduleKey = new ModuleKey(moduleName, getModuleTypeClass(moduleType));
 144                         final Optional<Module> moduleConfig = configModuleWriter.readModuleConfiguration(moduleKey, rTx);
 145                         if (moduleConfig.isPresent()) {
 146                             if (ibgp != null && ibgp.getConfig() != null && ibgp.getConfig().getMaximumPaths() != null) {
 147                                 final long maxPaths = ibgp.getConfig().getMaximumPaths();
 148                                 if (moduleConfig.get().getType().equals(AdvertiseNPaths.class)) {
 149                                     final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.path.selection.mode.rev160301.modules.module.configuration.AdvertiseNPaths config =
 150                                             (org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.path.selection.mode.rev160301.modules.module.configuration.AdvertiseNPaths)moduleConfig.get().getConfiguration();
 151                                     if (maxPaths == config.getNBestPaths().longValue()) {
 152                                         return true;
 153                                     }
 154                                 }
 155                             } else {
 156                                 if (moduleConfig.get().getType().equals(AdvertiseAllPaths.class)) {
 157                                     return true;
 158                                 }
 159                             }
 160                         }
 161                     }
 162                 }
 163             }
 164             return false;
 165         } catch (final ReadFailedException e) {
 166             throw new IllegalStateException(OpenConfigUtil.FAILED_TO_READ_SERVICE, e);
 167         }
 168     }
 169
 170 }