bgp/openconfig-state/src/main/java/org/opendaylight/protocol/bgp/state/NeighborUtil.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8
   9 package org.opendaylight.protocol.bgp.state;
  10
  11 import com.google.common.primitives.UnsignedInteger;
  12 import java.math.BigDecimal;
  13 import java.util.ArrayList;
  14 import java.util.HashSet;
  15 import java.util.List;
  16 import java.util.Map;
  17 import java.util.Objects;
  18 import java.util.Optional;
  19 import java.util.Set;
  20 import java.util.function.Function;
  21 import java.util.stream.Collectors;
  22 import org.eclipse.jdt.annotation.NonNull;
  23 import org.eclipse.jdt.annotation.Nullable;
  24 import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
  25 import org.opendaylight.protocol.bgp.rib.spi.state.BGPAfiSafiState;
  26 import org.opendaylight.protocol.bgp.rib.spi.state.BGPErrorHandlingState;
  27 import org.opendaylight.protocol.bgp.rib.spi.state.BGPGracelfulRestartState;
  28 import org.opendaylight.protocol.bgp.rib.spi.state.BGPLlGracelfulRestartState;
  29 import org.opendaylight.protocol.bgp.rib.spi.state.BGPPeerMessagesState;
  30 import org.opendaylight.protocol.bgp.rib.spi.state.BGPPeerState;
  31 import org.opendaylight.protocol.bgp.rib.spi.state.BGPSessionState;
  32 import org.opendaylight.protocol.bgp.rib.spi.state.BGPTimersState;
  33 import org.opendaylight.protocol.bgp.rib.spi.state.BGPTransportState;
  34 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
  35 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
  36 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiKey;
  37 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.operational.rev151009.BgpNeighborState.SessionState;
  38 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.operational.rev151009.bgp.neighbor.prefix.counters_state.PrefixesBuilder;
  39 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.graceful.restart.GracefulRestart;
  40 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.graceful.restart.GracefulRestartBuilder;
  41 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafis;
  42 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafisBuilder;
  43 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.ErrorHandling;
  44 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.ErrorHandlingBuilder;
  45 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.State;
  46 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.StateBuilder;
  47 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.Timers;
  48 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TimersBuilder;
  49 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.Transport;
  50 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TransportBuilder;
  51 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
  52 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborBuilder;
  53 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
  54 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.NeighborsBuilder;
  55 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.ADDPATHS;
  56 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.ASN32;
  57 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
  58 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.BgpCapability;
  59 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.GRACEFULRESTART;
  60 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.MPBGP;
  61 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.ROUTEREFRESH;
  62 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  63 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressNoZone;
  64 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
  65 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Timeticks;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.BgpNeighborStateAugmentation;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.BgpNeighborStateAugmentationBuilder;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborAfiSafiGracefulRestartStateAugmentation;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborAfiSafiGracefulRestartStateAugmentationBuilder;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborAfiSafiStateAugmentationBuilder;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborErrorHandlingStateAugmentation;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborErrorHandlingStateAugmentationBuilder;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborGracefulRestartStateAugmentation;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborGracefulRestartStateAugmentationBuilder;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborStateAugmentation;
  76 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborStateAugmentationBuilder;
  77 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborTimersStateAugmentationBuilder;
  78 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NeighborTransportStateAugmentationBuilder;
  79 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.MessagesBuilder;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.messages.Received;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.messages.ReceivedBuilder;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.messages.Sent;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.messages.SentBuilder;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
  85 import org.opendaylight.yangtools.yang.common.Uint16;
  86 import org.opendaylight.yangtools.yang.common.Uint32;
  87 import org.opendaylight.yangtools.yang.common.Uint64;
  88
  89 /**
  90  * Util for create OpenConfig Neighbor with corresponding openConfig state.
  91  */
  92 public final class NeighborUtil {
  93     private static final long TIMETICK_ROLLOVER_VALUE = UnsignedInteger.MAX_VALUE.longValue() + 1;
  94
  95     private NeighborUtil() {
  96         // Hidden on purpose
  97     }
  98
  99     /**
 100      * Build a Openconfig Neighbors container with all Neighbors Stats from a list of
 101      * BGPPeerGroupState.
 102      *
 103      * @param peerStats            List of BGPPeerState containing Neighbor state counters
 104      * @param bgpTableTypeRegistry BGP TableType Registry
 105      * @return Openconfig Neighbors Stats
 106      */
 107     public static @Nullable Neighbors buildNeighbors(final @NonNull List<BGPPeerState> peerStats,
 108             final @NonNull BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
 109         if (peerStats.isEmpty()) {
 110             return null;
 111         }
 112         return new NeighborsBuilder().setNeighbor(peerStats.stream()
 113                 .filter(Objects::nonNull)
 114                 .map(neighbor -> buildNeighbor(neighbor, bgpTableTypeRegistry))
 115                 .collect(Collectors.toUnmodifiableMap(Neighbor::key, Function.identity()))).build();
 116     }
 117
 118     /**
 119      * Build a list of neighbors containing Operational State from a list of BGPPeerState.
 120      *
 121      * @param neighbor containing Neighbor state counters
 122      * @return neighbor containing Neighbor State
 123      */
 124     public static @NonNull Neighbor buildNeighbor(final @NonNull BGPPeerState neighbor,
 125             final @NonNull BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
 126         return new NeighborBuilder()
 127                 .setNeighborAddress(convertIpAddress(neighbor.getNeighborAddress()))
 128                 .setState(buildNeighborState(neighbor.getBGPSessionState(), neighbor.getBGPPeerMessagesState()))
 129                 .setTimers(buildTimer(neighbor.getBGPTimersState()))
 130                 .setTransport(buildTransport(neighbor.getBGPTransportState()))
 131                 .setErrorHandling(buildErrorHandling(neighbor.getBGPErrorHandlingState()))
 132                 .setGracefulRestart(buildGracefulRestart(neighbor.getBGPGracelfulRestart()))
 133                 .setAfiSafis(buildAfisSafis(neighbor, bgpTableTypeRegistry))
 134                 .build();
 135     }
 136
 137     private static IpAddress convertIpAddress(final IpAddressNoZone addr) {
 138         if (addr == null) {
 139             return null;
 140         }
 141         final Ipv4AddressNoZone ipv4 = addr.getIpv4AddressNoZone();
 142         if (ipv4 != null) {
 143             return new IpAddress(ipv4);
 144         }
 145         return new IpAddress(addr.getIpv6AddressNoZone());
 146     }
 147
 148     /**
 149      * Builds Neighbor State from BGPPeerState counters.
 150      *
 151      * @param sessionState         BGPPeerState containing Operational state counters
 152      * @param bgpPeerMessagesState message state
 153      * @return Neighbor State
 154      */
 155     public static @Nullable State buildNeighborState(final @Nullable BGPSessionState sessionState,
 156             final BGPPeerMessagesState bgpPeerMessagesState) {
 157         if (sessionState == null && bgpPeerMessagesState == null) {
 158             return null;
 159         }
 160         final StateBuilder builder = new StateBuilder();
 161         if (sessionState != null) {
 162             builder.addAugmentation(buildCapabilityState(sessionState));
 163         }
 164         if (bgpPeerMessagesState != null) {
 165             builder.addAugmentation(buildMessageState(bgpPeerMessagesState));
 166         }
 167         return builder.build();
 168     }
 169
 170     /**
 171      * Builds Neighbor State from BGPPeerState counters.
 172      *
 173      * @param neighbor BGPPeerState containing Operational state counters
 174      * @return Timer State
 175      */
 176     public static @Nullable Timers buildTimer(final @Nullable BGPTimersState neighbor) {
 177         if (neighbor == null) {
 178             return null;
 179         }
 180         // convert neighbor uptime which is in milliseconds to time-ticks which is
 181         // hundredth of a second, and handle roll-over scenario
 182         final long uptimeTicks = neighbor.getUpTime() / 10 % TIMETICK_ROLLOVER_VALUE;
 183
 184         return new TimersBuilder()
 185                 .setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group
 186                     .timers.StateBuilder()
 187                         .addAugmentation(new NeighborTimersStateAugmentationBuilder()
 188                             .setNegotiatedHoldTime(BigDecimal.valueOf(neighbor.getNegotiatedHoldTime()))
 189                             .setUptime(new Timeticks(Uint32.valueOf(uptimeTicks))).build())
 190                         .build())
 191                 .build();
 192     }
 193
 194     /**
 195      * Builds Transport State from BGPTransportState counters.
 196      *
 197      * @param neighbor BGPPeerState containing Operational state counters
 198      * @return Transport State
 199      */
 200     public static @Nullable Transport buildTransport(final @Nullable BGPTransportState neighbor) {
 201         if (neighbor == null) {
 202             return null;
 203         }
 204
 205         return new TransportBuilder().setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009
 206                 .bgp.neighbor.group.transport.StateBuilder()
 207                     .addAugmentation(new NeighborTransportStateAugmentationBuilder()
 208                         .setLocalPort(neighbor.getLocalPort())
 209                         .setRemoteAddress(convertIpAddress(neighbor.getRemoteAddress()))
 210                         .setRemotePort(neighbor.getRemotePort())
 211                         .build())
 212                     .build())
 213                 .build();
 214     }
 215
 216     /**
 217      * Builds Error Handling State from BGPPeerState counters.
 218      *
 219      * @param errorHandlingState BGPErrorHandlingState containing ErrorHandlingState Operational state counters
 220      * @return Error Handling State
 221      */
 222     public static ErrorHandling buildErrorHandling(final @Nullable BGPErrorHandlingState errorHandlingState) {
 223         if (errorHandlingState == null) {
 224             return null;
 225         }
 226         return new ErrorHandlingBuilder()
 227                 .setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group
 228                     .error.handling.StateBuilder()
 229                         .addAugmentation(buildErrorHandlingState(errorHandlingState.getErroneousUpdateReceivedCount()))
 230                         .build())
 231                 .build();
 232     }
 233
 234     /**
 235      * Builds Graceful Restart containing Graceful Restart State from BGPGracelfulRestartState counters.
 236      *
 237      * @param neighbor BGPPeerState containing Operational state counters
 238      * @return Graceful Restart
 239      */
 240     public static @NonNull GracefulRestart buildGracefulRestart(final @NonNull BGPGracelfulRestartState neighbor) {
 241         final NeighborGracefulRestartStateAugmentation gracefulRestartState =
 242                 new NeighborGracefulRestartStateAugmentationBuilder()
 243                         .setLocalRestarting(neighbor.isLocalRestarting())
 244                         .setPeerRestartTime(Uint16.valueOf(neighbor.getPeerRestartTime()))
 245                         .setMode(neighbor.getMode())
 246                         .setPeerRestarting(neighbor.isPeerRestarting()).build();
 247
 248         return new GracefulRestartBuilder().setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp
 249                 .rev151009.bgp.graceful.restart.graceful.restart.StateBuilder()
 250                 .addAugmentation(gracefulRestartState).build()).build();
 251     }
 252
 253     /**
 254      * Builds  Neighbor Afi Safi containing AfiSafi State.
 255      *
 256      * @param neighbor BGPPeerState containing Operational state counters
 257      * @return Afi Safis
 258      */
 259     public static AfiSafis buildAfisSafis(final @NonNull BGPPeerState neighbor,
 260             final @NonNull BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
 261         return new AfiSafisBuilder().setAfiSafi(buildAfisSafisState(neighbor.getBGPAfiSafiState(),
 262                 bgpTableTypeRegistry)).build();
 263     }
 264
 265     /**
 266      * Builds  Neighbor State containing Capabilities State, session State.
 267      *
 268      * @return Neighbor State
 269      */
 270     public static NeighborStateAugmentation buildCapabilityState(final @NonNull BGPSessionState neighbor) {
 271
 272         final List<Class<? extends BgpCapability>> supportedCapabilities = buildSupportedCapabilities(neighbor);
 273         SessionState sessionState = null;
 274         switch (neighbor.getSessionState()) {
 275             case IDLE:
 276                 sessionState = SessionState.IDLE;
 277                 break;
 278             case UP:
 279                 sessionState = SessionState.ESTABLISHED;
 280                 break;
 281             case OPEN_CONFIRM:
 282                 sessionState = SessionState.OPENCONFIRM;
 283                 break;
 284             default:
 285         }
 286         return new NeighborStateAugmentationBuilder().setSupportedCapabilities(supportedCapabilities)
 287                 .setSessionState(sessionState).build();
 288     }
 289
 290     /**
 291      * Builds Bgp Neighbor State containing Message State.
 292      *
 293      * @return BgpNeighborState containing Message State
 294      */
 295     public static @NonNull BgpNeighborStateAugmentation buildMessageState(
 296             final @NonNull BGPPeerMessagesState neighbor) {
 297         return new BgpNeighborStateAugmentationBuilder()
 298                 .setMessages(new MessagesBuilder()
 299                         .setReceived(buildMessagesReceived(neighbor))
 300                         .setSent(buildMessagesSent(neighbor)).build()).build();
 301     }
 302
 303     private static Received buildMessagesReceived(final @NonNull BGPPeerMessagesState neighbor) {
 304         return new ReceivedBuilder()
 305                 .setUPDATE(Uint64.valueOf(neighbor.getUpdateMessagesReceivedCount()))
 306                 .setNOTIFICATION(Uint64.valueOf(neighbor.getNotificationMessagesReceivedCount()))
 307                 .build();
 308     }
 309
 310     private static Sent buildMessagesSent(final @NonNull BGPPeerMessagesState neighbor) {
 311         return new SentBuilder()
 312                 .setUPDATE(Uint64.valueOf(neighbor.getUpdateMessagesSentCount()))
 313                 .setNOTIFICATION(Uint64.valueOf(neighbor.getNotificationMessagesSentCount()))
 314                 .build();
 315     }
 316
 317     /**
 318      * Builds Neighbor Error Handling State.
 319      *
 320      * @param erroneousUpdateCount erroneous Update Count
 321      * @return Error Handling State
 322      */
 323     public static @NonNull NeighborErrorHandlingStateAugmentation buildErrorHandlingState(
 324             final long erroneousUpdateCount) {
 325         return new NeighborErrorHandlingStateAugmentationBuilder()
 326                 .setErroneousUpdateMessages(saturatedUint32(erroneousUpdateCount)).build();
 327     }
 328
 329     /**
 330      * Build List of afi safi containing State per Afi Safi.
 331      *
 332      * @return AfiSafi List
 333      */
 334     public static @NonNull Map<AfiSafiKey, AfiSafi> buildAfisSafisState(final @NonNull BGPAfiSafiState neighbor,
 335             final @NonNull BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
 336         final Set<TablesKey> afiSafiJoin = new HashSet<>(neighbor.getAfiSafisAdvertized());
 337         afiSafiJoin.addAll(neighbor.getAfiSafisReceived());
 338         return afiSafiJoin.stream().map(tableKey -> buildAfiSafi(neighbor, tableKey, bgpTableTypeRegistry))
 339                 .filter(Objects::nonNull)
 340                 .collect(Collectors.toUnmodifiableMap(AfiSafi::key, Function.identity()));
 341     }
 342
 343     private static AfiSafi buildAfiSafi(final @NonNull BGPAfiSafiState neighbor,
 344             final @NonNull TablesKey tablesKey, final @NonNull BGPTableTypeRegistryConsumer bgpTableTypeRegistry) {
 345         final Optional<Class<? extends AfiSafiType>> afiSafi = bgpTableTypeRegistry.getAfiSafiType(tablesKey);
 346         return afiSafi.map(aClass -> new AfiSafiBuilder().setAfiSafiName(aClass)
 347                 .setState(buildAfiSafiState(neighbor, tablesKey, neighbor.isAfiSafiSupported(tablesKey)))
 348                 .setGracefulRestart(buildAfiSafiGracefulRestartState(neighbor, tablesKey)).build()).orElse(null);
 349
 350     }
 351
 352     private static org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi
 353             .safi.list.afi.safi.State buildAfiSafiState(final @NonNull BGPAfiSafiState neighbor,
 354             final @NonNull TablesKey tablesKey, final boolean afiSafiSupported) {
 355         final NeighborAfiSafiStateAugmentationBuilder builder = new NeighborAfiSafiStateAugmentationBuilder();
 356         builder.setActive(afiSafiSupported);
 357         if (afiSafiSupported) {
 358             builder.setPrefixes(new PrefixesBuilder()
 359                     .setInstalled(saturatedUint32(neighbor.getPrefixesInstalledCount(tablesKey)))
 360                     .setReceived(saturatedUint32(neighbor.getPrefixesReceivedCount(tablesKey)))
 361                     .setSent(saturatedUint32(neighbor.getPrefixesSentCount(tablesKey))).build());
 362         }
 363         return new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi
 364                 .safi.list.afi.safi.StateBuilder().addAugmentation(builder.build()).build();
 365     }
 366
 367     // FIXME: remove this with YANGTOOLS-5.0.7+
 368     private static Uint32 saturatedUint32(final long value) {
 369         return value < 4294967295L ? Uint32.valueOf(value) : Uint32.MAX_VALUE;
 370     }
 371
 372     private static org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi
 373             .safi.list.afi.safi.GracefulRestart buildAfiSafiGracefulRestartState(
 374             final @NonNull BGPLlGracelfulRestartState neighbor, final @NonNull TablesKey tablesKey) {
 375         final NeighborAfiSafiGracefulRestartStateAugmentation builder =
 376                 new NeighborAfiSafiGracefulRestartStateAugmentationBuilder()
 377                         .setAdvertised(neighbor.isGracefulRestartAdvertized(tablesKey))
 378                         .setReceived(neighbor.isGracefulRestartReceived(tablesKey))
 379                         .setLlAdvertised(neighbor.isLlGracefulRestartAdvertised(tablesKey))
 380                         .setLlReceived(neighbor.isLlGracefulRestartReceived(tablesKey))
 381                         .setLlStaleTimer(Uint32.valueOf(neighbor.getLlGracefulRestartTimer(tablesKey))).build();
 382         return new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi
 383                 .safi.list.afi.safi.GracefulRestartBuilder().setState(new org.opendaylight.yang.gen.v1.http.openconfig
 384                 .net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.graceful.restart.StateBuilder()
 385                 .addAugmentation(builder).build()).build();
 386     }
 387
 388     /**
 389      * Builds List of BgpCapability supported capabilities.
 390      *
 391      * @return List containing supported capabilities
 392      */
 393     public static @NonNull List<Class<? extends BgpCapability>> buildSupportedCapabilities(
 394             final @NonNull BGPSessionState neighbor) {
 395         final List<Class<? extends BgpCapability>> supportedCapabilities = new ArrayList<>();
 396         if (neighbor.isAddPathCapabilitySupported()) {
 397             supportedCapabilities.add(ADDPATHS.class);
 398         }
 399         if (neighbor.isAsn32CapabilitySupported()) {
 400             supportedCapabilities.add(ASN32.class);
 401         }
 402         if (neighbor.isGracefulRestartCapabilitySupported()) {
 403             supportedCapabilities.add(GRACEFULRESTART.class);
 404         }
 405         if (neighbor.isMultiProtocolCapabilitySupported()) {
 406             supportedCapabilities.add(MPBGP.class);
 407         }
 408         if (neighbor.isRouterRefreshCapabilitySupported()) {
 409             supportedCapabilities.add(ROUTEREFRESH.class);
 410         }
 411         return supportedCapabilities;
 412     }
 413 }