org.opendaylight.protocol.bgp.rib.impl.BGPPeerTrackerImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bgp-rib-impl Show documentation

BGP RIB implementation

The newest version!

/*
 * Copyright (c) 2018 AT&T Intellectual Property. 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.rib.impl;

import com.google.common.collect.ImmutableList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.checkerframework.checker.lock.qual.GuardedBy;
import org.opendaylight.protocol.bgp.rib.spi.BGPPeerTracker;
import org.opendaylight.protocol.bgp.rib.spi.Peer;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerRole;
import org.opendaylight.yangtools.concepts.AbstractRegistration;
import org.opendaylight.yangtools.concepts.Registration;

public final class BGPPeerTrackerImpl implements BGPPeerTracker {
    @GuardedBy("this")
    private final Map peers = new HashMap<>();
    private ImmutableList peersList;
    private ImmutableList peersFilteredList;

    @Override
    public synchronized Registration registerPeer(final Peer peer) {
        this.peers.put(peer.getPeerId(), peer);
        this.peersList = ImmutableList.copyOf(this.peers.values());
        this.peersFilteredList = this.peers.values().stream()
                .filter(p1 -> p1.getRole() != PeerRole.Internal)
                .collect(ImmutableList.toImmutableList());
        return new AbstractRegistration() {
            @Override
            protected void removeRegistration() {
                synchronized (BGPPeerTrackerImpl.this) {
                    BGPPeerTrackerImpl.this.peers.remove(peer.getPeerId());
                }
            }
        };
    }

    @Override
    public synchronized Peer getPeer(final PeerId peerId) {
        return this.peers.get(peerId);
    }

    @Override
    public synchronized List getPeers() {
        return this.peersList;
    }

    @Override
    public synchronized List getNonInternalPeers() {
        return this.peersFilteredList;
    }
}