bgp/topology-provider/src/main/java/org/opendaylight/bgpcep/bgp/topology/provider/config/AbstractBgpTopologyProvider.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 package org.opendaylight.bgpcep.bgp.topology.provider.config;
   9
  10 import java.util.HashMap;
  11 import java.util.Map;
  12 import org.opendaylight.bgpcep.bgp.topology.provider.AbstractTopologyBuilder;
  13 import org.opendaylight.bgpcep.bgp.topology.provider.spi.BgpTopologyDeployer;
  14 import org.opendaylight.bgpcep.bgp.topology.provider.spi.BgpTopologyProvider;
  15 import org.opendaylight.bgpcep.bgp.topology.provider.spi.TopologyReferenceSingletonService;
  16 import org.opendaylight.mdsal.binding.api.DataBroker;
  17 import org.opendaylight.protocol.bgp.rib.DefaultRibReference;
  18 import org.opendaylight.protocol.bgp.rib.RibReference;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.BgpRib;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.bgp.rib.Rib;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.bgp.rib.RibKey;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.topology.config.rev180329.Topology1;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.topology.types.rev160524.TopologyTypes1;
  24 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  25 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  26 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.TopologyTypes;
  27 import org.opendaylight.yangtools.concepts.AbstractRegistration;
  28 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  29 import org.slf4j.Logger;
  30 import org.slf4j.LoggerFactory;
  31
  32 abstract class AbstractBgpTopologyProvider implements BgpTopologyProvider, AutoCloseable {
  33
  34     private static final Logger LOG = LoggerFactory.getLogger(AbstractBgpTopologyProvider.class);
  35
  36     private final Map<TopologyId, TopologyReferenceSingletonService> topologyBuilders = new HashMap<>();
  37     private final AbstractRegistration registration;
  38     private final DataBroker dataBroker;
  39     private final BgpTopologyDeployer deployer;
  40
  41     AbstractBgpTopologyProvider(final BgpTopologyDeployer deployer) {
  42         this.deployer = deployer;
  43         this.dataBroker = deployer.getDataBroker();
  44         this.registration = deployer.registerTopologyProvider(this);
  45     }
  46
  47     @Override
  48     public final void onTopologyBuilderCreated(final Topology topology) {
  49         LOG.debug("Creating topology builder instance {}", topology);
  50         final TopologyReferenceSingletonService currentInstance = this.topologyBuilders.get(topology.getTopologyId());
  51         if (currentInstance == null || !currentInstance.getConfiguration().equals(topology)) {
  52             final TopologyReferenceSingletonService topologyBuilder = createInstance(topology);
  53             this.topologyBuilders.put(topology.getTopologyId(), topologyBuilder);
  54             LOG.debug("Topology builder instance created {}", topologyBuilder);
  55         }
  56     }
  57
  58     @Override
  59     public final void onTopologyBuilderRemoved(final Topology topology) {
  60         LOG.debug("Removing topology builder instance {}", topology);
  61         final TopologyReferenceSingletonService topologyBuilder =
  62                 this.topologyBuilders.remove(topology.getTopologyId());
  63         if (topologyBuilder != null) {
  64             topologyBuilder.close();
  65             LOG.debug("Topology builder instance removed {}", topologyBuilder);
  66         }
  67     }
  68
  69     @Override
  70     public void close() {
  71         this.registration.close();
  72     }
  73
  74     private TopologyReferenceSingletonService createInstance(final Topology topology) {
  75         final RibReference ribReference = new DefaultRibReference(InstanceIdentifier.create(BgpRib.class)
  76                 .child(Rib.class, new RibKey(topology.augmentation(Topology1.class).getRibId())));
  77         final AbstractTopologyBuilder<?> topologyBuilder = createTopologyBuilder(this.dataBroker,
  78                 ribReference, topology.getTopologyId());
  79         return new TopologyReferenceSingletonServiceImpl(topologyBuilder, this.deployer, topology);
  80     }
  81
  82     abstract AbstractTopologyBuilder<?> createTopologyBuilder(DataBroker dataProvider, RibReference locRibReference,
  83             TopologyId topologyId);
  84
  85     static final TopologyTypes1 getTopologyAug(final Topology topology) {
  86         final TopologyTypes topologyTypes = topology.getTopologyTypes();
  87         if (topologyTypes == null) {
  88             return null;
  89         }
  90
  91         return topologyTypes.augmentation(TopologyTypes1.class);
  92     }
  93 }