2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.protocol.bgp.openconfig.impl.moduleconfig;
11 import com.google.common.base.Function;
12 import com.google.common.base.Optional;
13 import com.google.common.base.Preconditions;
14 import com.google.common.base.Predicate;
15 import com.google.common.collect.FluentIterable;
16 import com.google.common.collect.ImmutableList;
17 import com.google.common.util.concurrent.AsyncFunction;
18 import com.google.common.util.concurrent.Futures;
19 import com.google.common.util.concurrent.ListenableFuture;
20 import java.util.ArrayList;
21 import java.util.HashMap;
22 import java.util.List;
24 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
25 import org.opendaylight.protocol.bgp.openconfig.impl.util.OpenConfigUtil;
26 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev150515.bgp.common.afi.safi.list.AfiSafi;
27 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev150515.AfiSafiType;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev130409.BgpTableType;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev130409.BgpTableTypeImpl;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.ServiceRef;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.modules.Module;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.modules.ModuleKey;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.services.Service;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.services.ServiceKey;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.rev130405.services.service.Instance;
36 import org.opendaylight.yangtools.yang.binding.ChildOf;
38 final class TableTypesFunction<T extends ServiceRef & ChildOf<Module>> implements AsyncFunction<List<AfiSafi>, List<T>> {
40 private final ReadTransaction rTx;
41 private final BGPConfigModuleProvider configModuleWriter;
42 private final Function<String, T> function;
44 public TableTypesFunction(final ReadTransaction rTx, final BGPConfigModuleProvider configModuleWriter, final Function<String, T> function) {
45 this.rTx = Preconditions.checkNotNull(rTx);
46 this.configModuleWriter = Preconditions.checkNotNull(configModuleWriter);
47 this.function = Preconditions.checkNotNull(function);
51 public ListenableFuture<List<T>> apply(final List<AfiSafi> afiSafis) {
52 final ListenableFuture<Optional<Service>> readFuture = configModuleWriter.readConfigService(new ServiceKey(BgpTableType.class), rTx);
53 return Futures.transform(readFuture, new AsyncFunction<Optional<Service>, List<T>>() {
56 public ListenableFuture<List<T>> apply(final Optional<Service> maybeService) {
57 if (maybeService.isPresent()) {
58 final Service service = maybeService.get();
59 final List<ListenableFuture<Optional<Module>>> modulesFuture = new ArrayList<>();
60 final Map<String, String> moduleNameToService = new HashMap<>();
61 for (final Instance instance : service.getInstance()) {
62 final String moduleName = OpenConfigUtil.getModuleName(instance.getProvider());
63 modulesFuture.add(configModuleWriter.readModuleConfiguration(new ModuleKey(moduleName, BgpTableTypeImpl.class), rTx));
64 moduleNameToService.put(moduleName, instance.getName());
66 return Futures.transform(Futures.successfulAsList(modulesFuture), new ModulesToLocalTablesFunction(afiSafis, moduleNameToService));
68 return Futures.immediateFailedFuture(new IllegalStateException("No BgpTableType service present in configuration."));
74 private final class ModulesToLocalTablesFunction implements Function<List<Optional<Module>>, List<T>> {
76 private final List<AfiSafi> afiSafis;
77 private final Map<String, String> moduleNameToService;
79 public ModulesToLocalTablesFunction(final List<AfiSafi> afiSafis, final Map<String, String> moduleNameToService) {
80 this.afiSafis = afiSafis;
81 this.moduleNameToService = moduleNameToService;
85 public List<T> apply(final List<Optional<Module>> maybeModules) {
86 final ImmutableList<Module> modules = FluentIterable.from(maybeModules)
87 .filter(new Predicate<Optional<Module>>() {
89 public boolean apply(final Optional<Module> input) {
90 return input.isPresent();
92 }).transform(new Function<Optional<Module>, Module>() {
94 public Module apply(final Optional<Module> input) {
99 return toTableTypes(afiSafiToModuleName(modules));
102 private Map<Class<? extends AfiSafiType>, String> afiSafiToModuleName(final List<Module> modules) {
103 final Map<Class<? extends AfiSafiType>, String> afiSafiToModuleName = new HashMap<>(afiSafis.size());
104 for (final Module module : modules) {
105 final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev130409.modules.module.configuration.BgpTableTypeImpl config =
106 ((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.bgp.rib.impl.rev130409.modules.module.configuration.BgpTableTypeImpl) module.getConfiguration());
107 final Optional<AfiSafi> afiSafi = OpenConfigUtil.toAfiSafi(new org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl(config.getAfi(), config.getSafi()));
108 if (afiSafi.isPresent()) {
109 afiSafiToModuleName.put(afiSafi.get().getAfiSafiName(), module.getName());
112 return afiSafiToModuleName;
115 private List<T> toTableTypes(final Map<Class<? extends AfiSafiType>, String> afiSafiToModuleName) {
116 final List<T> tableTypes = new ArrayList<>(afiSafis.size());
117 for (final AfiSafi afiSafi : afiSafis) {
118 final String moduleName = afiSafiToModuleName.get(afiSafi.getAfiSafiName());
119 if (moduleName != null) {
120 tableTypes.add(function.apply(moduleNameToService.get(moduleName)));