-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Generate network map from BGP local rib
Signed-off-by: jingxuanzhang <jingxuan.n.zhang@gmail.com>
- Loading branch information
Showing
30 changed files
with
626 additions
and
154 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
151 changes: 151 additions & 0 deletions
151
...-maps/impl/src/main/java/org/opendaylight/alto/basic/impl/AltoAutoMapsBgpIpv4Updater.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,151 @@ | ||
| /* | ||
| * Copyright © 2015 Yale University and others. All rights reserved. | ||
| * | ||
| * This program and the accompanying materials are made available under the | ||
| * terms of the Eclipse Public License v1.0 which accompanies this distribution, | ||
| * and is available at http://www.eclipse.org/legal/epl-v10.html | ||
| */ | ||
| package org.opendaylight.alto.basic.impl; | ||
|
|
||
| import com.google.common.base.Optional; | ||
| import org.opendaylight.alto.basic.manual.maps.ManualMapsUtils; | ||
| import org.opendaylight.controller.md.sal.binding.api.DataBroker; | ||
| import org.opendaylight.controller.md.sal.binding.api.ReadTransaction; | ||
| import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction; | ||
| import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.auto.maps.rev150105.config.context.ResourceNetworkMap; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.auto.maps.rev150105.config.context.resource.network.map.algorithm.BgpSimpleAsCluster; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.EndpointAddressType; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.endpoint.address.group.EndpointAddressGroup; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.endpoint.address.group.EndpointAddressGroupBuilder; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.network.map.Map; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.network.map.MapBuilder; | ||
| import org.opendaylight.yang.gen.v1.urn.alto.types.rev150921.PidName; | ||
| import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber; | ||
| import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.ipv4.routes.ipv4.routes.Ipv4Route; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.as.path.Segments; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.BgpRib; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.RibId; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.Rib; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.RibKey; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.rib.LocRib; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.Tables; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.TablesKey; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily; | ||
| import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily; | ||
| import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; | ||
| import org.slf4j.Logger; | ||
| import org.slf4j.LoggerFactory; | ||
|
|
||
| import java.util.Arrays; | ||
| import java.util.LinkedHashMap; | ||
| import java.util.LinkedList; | ||
| import java.util.List; | ||
| import java.util.concurrent.ExecutionException; | ||
|
|
||
| public class AltoAutoMapsBgpIpv4Updater implements AutoCloseable { | ||
|
|
||
| private static final Logger LOG = LoggerFactory.getLogger(AltoAutoMapsBgpIpv4Updater.class); | ||
|
|
||
| private final DataBroker dataBroker; | ||
| // private ListenerRegistration<?> registration; | ||
| private String contextId; | ||
| private ResourceNetworkMap networkMapConfig; | ||
|
|
||
| public AltoAutoMapsBgpIpv4Updater(String contextId, ResourceNetworkMap networkMapConfig, final DataBroker dataBroker) { | ||
| this.dataBroker = dataBroker; | ||
| this.contextId = contextId; | ||
| this.networkMapConfig = networkMapConfig; | ||
| registerBGPListener(); | ||
| } | ||
|
|
||
| private void registerBGPListener() { | ||
| final ReadWriteTransaction wrx = dataBroker.newReadWriteTransaction(); | ||
| try { | ||
| List<Map> networkMap = computeNetworkMapByBgpIpv4(networkMapConfig); | ||
| LOG.info("Putting auto generated network-map to manual map config..."); | ||
| ManualMapsUtils.createResourceNetworkMap(contextId, networkMapConfig.getResourceId().getValue(), | ||
| networkMap, wrx); | ||
| } catch (InterruptedException e) { | ||
| e.printStackTrace(); | ||
| } catch (ExecutionException e) { | ||
| e.printStackTrace(); | ||
| } | ||
| wrx.submit(); | ||
| } | ||
|
|
||
| private List<Map> computeNetworkMapByBgpIpv4(ResourceNetworkMap networkMapConfig) throws InterruptedException, ExecutionException { | ||
| final ReadTransaction rx = dataBroker.newReadOnlyTransaction(); | ||
| List<Map> networkMap = new LinkedList<>(); | ||
| if (networkMapConfig.getAlgorithm() instanceof BgpSimpleAsCluster) { | ||
| BgpSimpleAsCluster algorithm = (BgpSimpleAsCluster) networkMapConfig.getAlgorithm(); | ||
| String ribId = algorithm.getBgpSimpleAsParams().getBgpRib(); | ||
| InstanceIdentifier<Tables> ribIID = InstanceIdentifier.builder(BgpRib.class) | ||
| .child(Rib.class, new RibKey(new RibId(ribId))) | ||
| .child(LocRib.class) | ||
| .child(Tables.class, new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class)) | ||
| .build(); | ||
| try { | ||
| Optional<Tables> optional = rx.read(LogicalDatastoreType.OPERATIONAL, ribIID).get(); | ||
| if (optional.isPresent()) { | ||
| Tables table = optional.get(); | ||
| if (table.getRoutes() instanceof Ipv4RoutesCase) { | ||
| Ipv4RoutesCase routesCase = (Ipv4RoutesCase) table.getRoutes(); | ||
| java.util.Map<String, List<IpPrefix>> pids = new LinkedHashMap<>(); | ||
| for (Ipv4Route route : routesCase.getIpv4Routes().getIpv4Route()) { | ||
| List<Segments> segments = route.getAttributes().getAsPath().getSegments(); | ||
| String pidName = "PID0"; | ||
| if (segments != null && !segments.isEmpty()) { | ||
| List<AsNumber> asSequence = segments.get(segments.size() - 1).getAsSequence(); | ||
| if (asSequence != null && !asSequence.isEmpty()) { | ||
| pidName = "PID" + asSequence.get(asSequence.size() - 1).getValue().toString(); | ||
| } else { | ||
| List<AsNumber> asSet = segments.get(segments.size() - 1).getAsSet(); | ||
| if (asSet != null && !asSet.isEmpty()) { | ||
| pidName = "PID" + String.join("-", | ||
| (String[]) Arrays.stream((AsNumber[]) asSet.toArray()) | ||
| .map(s -> s.getValue().toString()) | ||
| .toArray()); | ||
| } | ||
| } | ||
| } | ||
| if (!pids.containsKey(pidName)) { | ||
| pids.put(pidName, new LinkedList<>()); | ||
| } | ||
| pids.get(pidName).add(new IpPrefix(route.getPrefix())); | ||
| } | ||
| for (java.util.Map.Entry<String, List<IpPrefix>> entry : pids.entrySet()) { | ||
| String pidName = entry.getKey(); | ||
| List<IpPrefix> prefixList = entry.getValue(); | ||
| networkMap.add(new MapBuilder() | ||
| .setPid(new PidName(pidName)) | ||
| .setEndpointAddressGroup(Arrays.asList(new EndpointAddressGroup[]{ | ||
| new EndpointAddressGroupBuilder() | ||
| .setAddressType(new EndpointAddressType(EndpointAddressType.Enumeration.Ipv4)) | ||
| .setEndpointPrefix(prefixList) | ||
| .build() | ||
| })) | ||
| .build()); | ||
| } | ||
| } else { | ||
| throw new ExecutionException("Unsupported route type", null); | ||
| } | ||
| } else { | ||
| throw new InterruptedException("BGP Local RIB not found"); | ||
| } | ||
| } catch (InterruptedException | ExecutionException e) { | ||
| throw e; | ||
| } | ||
| } else { | ||
| throw new InterruptedException("Unsupported algorithm for bgp topology"); | ||
| } | ||
| return networkMap; | ||
| } | ||
|
|
||
| @Override | ||
| public void close() throws Exception { | ||
| LOG.info("BGP IPv4 updater closed"); | ||
| } | ||
| } |
Oops, something went wrong.