infra/minimal-distribution/src/main/java/io/fd/honeycomb/infra/distro/bgp/BGPPeerRegistryProvider.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.distro.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.bgp.neighbors.Neighbor;
  41 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
  42 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
  43 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.NetworkInstances;
  44 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstance;
  45 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceKey;
  46 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.Protocols;
  47 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
  48 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
  49 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config2;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Protocol1;
  52 import org.opendaylight.yangtools.yang.binding.DataObject;
  53 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  54 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  55 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  56 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  57 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
  58 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  59 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
  60
  61 final class BGPPeerRegistryProvider extends ProviderTrait<BGPPeerRegistry> {
  62     private static final String PEERS_CFG = "/bgp-peers.json";
  63     @Inject
  64     private BindingToNormalizedNodeCodec codec;
  65     @Inject
  66     private RIB globalRib;
  67     @Inject
  68     private BGPOpenConfigMappingService mappingService;
  69     @Inject
  70     private SchemaService schemaService;
  71
  72     @Override
  73     protected BGPPeerRegistry create() {
  74         final BGPPeerRegistry peerRegistry = StrictBGPPeerRegistry.instance();
  75         final Neighbors neighbors = readNeighbours();
  76         for (final Neighbor neighbor : neighbors.getNeighbor()) {
  77             if (isApplicationPeer(neighbor)) {
  78                 new AppPeer().start(globalRib, neighbor, mappingService, null);
  79             } else {
  80                 new BgpPeer(null, peerRegistry).start(globalRib, neighbor, mappingService, null);
  81             }
  82         }
  83         return peerRegistry;
  84     }
  85
  86     private Neighbors readNeighbours() {
  87         final InputStream resourceStream = this.getClass().getResourceAsStream(PEERS_CFG);
  88         checkState(resourceStream != null, "Resource %s not found", PEERS_CFG);
  89
  90         final InstanceIdentifier<Bgp> bgpII = InstanceIdentifier.create(NetworkInstances.class)
  91             .child(NetworkInstance.class, new NetworkInstanceKey("dummy-value")).child(Protocols.class)
  92             .child(Protocol.class, new ProtocolKey(BGP.class, "dummy-value")).augmentation(Protocol1.class)
  93             .child(Bgp.class);
  94         final InstanceIdentifier<Neighbors> neighborsII = bgpII.child(Neighbors.class);
  95
  96         final YangInstanceIdentifier neighborsYII = codec.toYangInstanceIdentifier(neighborsII);
  97         final SchemaContext schemaContext = schemaService.getGlobalContext();
  98         final Optional<DataNodeContainer> parentNode = findDataNodeContainer(schemaContext, bgpII);
  99         final ContainerNode parentContainer = readContainerEntryJson(schemaContext, resourceStream,
 100             (SchemaNode) parentNode.get(),
 101             (YangInstanceIdentifier.NodeIdentifier) neighborsYII.getLastPathArgument());
 102         final NormalizedNode<?, ?> neighborsContainer = parentContainer.getValue().iterator().next();
 103
 104         final Map.Entry<InstanceIdentifier<?>, DataObject> entry = codec.fromNormalizedNode(neighborsYII, neighborsContainer);
 105         checkNotNull(entry, "Failed to deserialize neighbours configuration at %s", PEERS_CFG);
 106         return (Neighbors) entry.getValue();
 107     }
 108
 109     private static boolean isApplicationPeer(@Nonnull final Neighbor neighbor) {
 110         return java.util.Optional.of(neighbor.getConfig())
 111             .map(config -> config.getAugmentation(Config2.class))
 112             .map(BgpNeighborPeerGroupConfig::getPeerGroup)
 113             .map(APPLICATION_PEER_GROUP_NAME::equals)
 114             .orElse(false);
 115     }
 116 }