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