infra/bgp-distribution/src/main/java/io/fd/honeycomb/infra/bgp/BgpNeighboursProvider.java

   1 /*
   2  * Copyright (c) 2017 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.infra.bgp;
  18
  19 import static com.google.common.base.Preconditions.checkNotNull;
  20 import static com.google.common.base.Preconditions.checkState;
  21 import static io.fd.honeycomb.translate.util.JsonUtils.readContainerEntryJson;
  22 import static org.opendaylight.protocol.bgp.openconfig.impl.util.OpenConfigUtil.APPLICATION_PEER_GROUP_NAME;
  23 import static org.opendaylight.yangtools.sal.binding.generator.impl.BindingSchemaContextUtils.findDataNodeContainer;
  24
  25 import com.google.common.base.Optional;
  26 import com.google.inject.Inject;
  27 import io.fd.honeycomb.infra.distro.ProviderTrait;
  28 import java.io.InputStream;
  29 import java.util.Map;
  30 import javax.annotation.Nonnull;
  31 import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodec;
  32 import org.opendaylight.controller.sal.core.api.model.SchemaService;
  33 import org.opendaylight.protocol.bgp.openconfig.spi.BGPOpenConfigMappingService;
  34 import org.opendaylight.protocol.bgp.rib.impl.StrictBGPPeerRegistry;
  35 import org.opendaylight.protocol.bgp.rib.impl.config.AppPeer;
  36 import org.opendaylight.protocol.bgp.rib.impl.config.BgpPeer;
  37 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
  38 import org.opendaylight.protocol.bgp.rib.impl.spi.RIB;
  39 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.BgpNeighborPeerGroupConfig;
  40 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.BgpNeighbors;
  41 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
  42 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
  43 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
  44 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.NetworkInstances;
  45 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstance;
  46 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceKey;
  47 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.Protocols;
  48 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
  49 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
  50 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config2;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Protocol1;
  53 import org.opendaylight.yangtools.yang.binding.DataObject;
  54 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  55 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  56 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  57 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  58 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
  59 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  60 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
  61 import org.slf4j.Logger;
  62 import org.slf4j.LoggerFactory;
  63
  64 final class BgpNeighboursProvider extends ProviderTrait<BgpNeighbors> {
  65     private static final Logger LOG = LoggerFactory.getLogger(BgpNeighboursProvider.class);
  66     private static final String PEERS_CFG = "/bgp-peers.json";
  67     @Inject
  68     private BindingToNormalizedNodeCodec codec;
  69     @Inject
  70     private RIB globalRib;
  71     @Inject
  72     private BGPOpenConfigMappingService mappingService;
  73     @Inject
  74     private SchemaService schemaService;
  75     @Inject
  76     private BGPPeerRegistry peerRegistry;
  77
  78     @Override
  79     protected BgpNeighbors create() {
  80         final BgpNeighbors neighbors = readNeighbours();
  81         for (final Neighbor neighbor : neighbors.getNeighbor()) {
  82             if (isApplicationPeer(neighbor)) {
  83                 LOG.trace("Starting AppPeer for {}", neighbor);
  84                 new AppPeer().start(globalRib, neighbor, mappingService, null);
  85             } else {
  86                 LOG.trace("Starting BgpPeer for {}", neighbor);
  87                 new BgpPeer(null, peerRegistry).start(globalRib, neighbor, mappingService, null);
  88             }
  89         }
  90         LOG.debug("BgpNeighbours initialized: {}", neighbors);
  91         return neighbors;
  92     }
  93
  94     private Neighbors readNeighbours() {
  95         LOG.debug("Reading BGP neighbours from {}", PEERS_CFG);
  96         final InputStream resourceStream = this.getClass().getResourceAsStream(PEERS_CFG);
  97         checkState(resourceStream != null, "Resource %s not found", PEERS_CFG);
  98
  99         final InstanceIdentifier<Bgp> bgpII = InstanceIdentifier.create(NetworkInstances.class)
 100             .child(NetworkInstance.class, new NetworkInstanceKey("dummy-value")).child(Protocols.class)
 101             .child(Protocol.class, new ProtocolKey(BGP.class, "dummy-value")).augmentation(Protocol1.class)
 102             .child(Bgp.class);
 103         final InstanceIdentifier<Neighbors> neighborsII = bgpII.child(Neighbors.class);
 104
 105         final YangInstanceIdentifier neighborsYII = codec.toYangInstanceIdentifier(neighborsII);
 106         final SchemaContext schemaContext = schemaService.getGlobalContext();
 107         final Optional<DataNodeContainer> parentNode = findDataNodeContainer(schemaContext, bgpII);
 108         final ContainerNode parentContainer = readContainerEntryJson(schemaContext, resourceStream,
 109             (SchemaNode) parentNode.get(),
 110             (YangInstanceIdentifier.NodeIdentifier) neighborsYII.getLastPathArgument());
 111         final NormalizedNode<?, ?> neighborsContainer = parentContainer.getValue().iterator().next();
 112
 113         final Map.Entry<InstanceIdentifier<?>, DataObject> entry = codec.fromNormalizedNode(neighborsYII, neighborsContainer);
 114         checkNotNull(entry, "Failed to deserialize neighbours configuration at %s", PEERS_CFG);
 115         return (Neighbors) entry.getValue();
 116     }
 117
 118     private static boolean isApplicationPeer(@Nonnull final Neighbor neighbor) {
 119         return java.util.Optional.of(neighbor.getConfig())
 120             .map(config -> config.getAugmentation(Config2.class))
 121             .map(BgpNeighborPeerGroupConfig::getPeerGroup)
 122             .map(APPLICATION_PEER_GROUP_NAME::equals)
 123             .orElse(false);
 124     }
 125 }