2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.protocol.bgp.state;
11 import java.util.List;
12 import java.util.Optional;
13 import java.util.stream.Collectors;
14 import javax.annotation.Nonnull;
15 import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
16 import org.opendaylight.protocol.bgp.rib.spi.state.BGPRibState;
17 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
18 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
19 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.State;
20 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.StateBuilder;
21 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafisBuilder;
22 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
23 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
24 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.GlobalAfiSafiStateAugmentation;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.GlobalAfiSafiStateAugmentationBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
29 public final class GlobalUtil {
30 private GlobalUtil() {
31 throw new UnsupportedOperationException();
35 * Build Openconfig Global containing RIB group stats from a list of BGP RIB State.
37 * @param ribState containing RIb Operational State
38 * @param bgpTableTypeRegistry BGP TableType Registry
39 * @return Global containing state
42 public static Global buildGlobal(final BGPRibState ribState,
43 final BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
44 return new GlobalBuilder().setState(buildState(ribState))
45 .setAfiSafis(new AfiSafisBuilder().setAfiSafi(buildAfisSafis(ribState, bgpTableTypeRegistry)).build())
50 * Build per AFI SAFI Openconfig Global State containing RIB group stats from a list of BGP RIB State.
52 * @param ribState containing RIb Operational State
53 * @param bgpTableTypeRegistry BGP TableType Registry
54 * @return List containing per afi/safi operational state
56 public static List<AfiSafi> buildAfisSafis(final BGPRibState ribState,
57 final BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
58 return ribState.getPathsCount().keySet().stream()
59 .map(tk -> buildAfiSafi(ribState, tk, bgpTableTypeRegistry))
60 .collect(Collectors.toList());
64 * Build Openconfig Global State.
66 * @param ribState containing RIb Operational State
67 * @return Openconfig Global State
69 public static org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base
70 .State buildState(final BGPRibState ribState) {
71 return new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.StateBuilder()
72 .setAs(ribState.getAs())
73 .setRouterId(ribState.getRouteId())
74 .setTotalPaths(ribState.getTotalPathsCount())
75 .setTotalPrefixes(ribState.getTotalPrefixesCount())
80 * Build Afi Safi containing State.
82 * @param ribState containing RIb Operational State
83 * @param tablesKey table Key
84 * @param bgpTableTypeRegistry BGP TableType Registry
85 * @return Afi Safi Operational State
87 public static AfiSafi buildAfiSafi(final BGPRibState ribState, final TablesKey tablesKey,
88 final BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
89 final Optional<Class<? extends AfiSafiType>> optAfiSafi = bgpTableTypeRegistry.getAfiSafiType(tablesKey);
90 if (!optAfiSafi.isPresent()) {
93 final State state = new StateBuilder()
94 .addAugmentation(GlobalAfiSafiStateAugmentation.class, new GlobalAfiSafiStateAugmentationBuilder()
95 .setTotalPaths(ribState.getPathCount(tablesKey))
96 .setTotalPrefixes(ribState.getPrefixesCount(tablesKey))
98 return new AfiSafiBuilder().setAfiSafiName(optAfiSafi.get()).setState(state).build();