org.opendaylight.protocol.bgp.state.GlobalUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bgp-openconfig-state Show documentation
BGP Openconfig State
There is a newer version: 0.22.6
/*
 * Copyright (c) 2016 Cisco Systems, Inc. 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.protocol.bgp.state;

import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import javax.annotation.Nonnull;
import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
import org.opendaylight.protocol.bgp.rib.spi.state.BGPRibState;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.State;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.StateBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafisBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.GlobalAfiSafiStateAugmentation;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.GlobalAfiSafiStateAugmentationBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;

public final class GlobalUtil {
    private GlobalUtil() {
        throw new UnsupportedOperationException();
    }

    /**
     * Build Openconfig Global containing RIB group stats from a list of BGP RIB State.
     *
     * @param ribState             containing RIb Operational State
     * @param bgpTableTypeRegistry BGP TableType Registry
     * @return Global containing state
     */
    @Nonnull
    public static Global buildGlobal(final BGPRibState ribState,
            final BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
        return new GlobalBuilder().setState(buildState(ribState))
                .setAfiSafis(new AfiSafisBuilder().setAfiSafi(buildAfisSafis(ribState, bgpTableTypeRegistry)).build())
                .build();
    }

    /**
     * Build per AFI SAFI Openconfig Global State containing RIB group stats from a list of BGP RIB State.
     *
     * @param ribState             containing RIb Operational State
     * @param bgpTableTypeRegistry BGP TableType Registry
     * @return List containing per afi/safi operational state
     */
    public static List buildAfisSafis(final BGPRibState ribState,
            final BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
        return ribState.getPathsCount().keySet().stream()
                .map(tk -> buildAfiSafi(ribState, tk, bgpTableTypeRegistry))
                .collect(Collectors.toList());
    }

    /**
     * Build Openconfig Global State.
     *
     * @param ribState containing RIb Operational State
     * @return Openconfig Global State
     */
    public static org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base
            .State buildState(final BGPRibState ribState) {
        return new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.StateBuilder()
                .setAs(ribState.getAs())
                .setRouterId(ribState.getRouteId())
                .setTotalPaths(ribState.getTotalPathsCount())
                .setTotalPrefixes(ribState.getTotalPrefixesCount())
                .build();
    }

    /**
     * Build Afi Safi containing State.
     *
     * @param ribState             containing RIb Operational State
     * @param tablesKey            table Key
     * @param bgpTableTypeRegistry BGP TableType Registry
     * @return Afi Safi Operational State
     */
    public static AfiSafi buildAfiSafi(final BGPRibState ribState, final TablesKey tablesKey,
            final BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
        final Optional> optAfiSafi = bgpTableTypeRegistry.getAfiSafiType(tablesKey);
        if (!optAfiSafi.isPresent()) {
            return null;
        }
        final State state = new StateBuilder()
                .addAugmentation(GlobalAfiSafiStateAugmentation.class, new GlobalAfiSafiStateAugmentationBuilder()
                        .setTotalPaths(ribState.getPathCount(tablesKey))
                        .setTotalPrefixes(ribState.getPrefixesCount(tablesKey))
                        .build()).build();
        return new AfiSafiBuilder().setAfiSafiName(optAfiSafi.get()).setState(state).build();
    }
}