peer_manager.h

Go to the documentation of this file.

/*
 * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
 */

#ifndef SRC_BGP_ROUTING_INSTANCE_PEER_MANAGER_H_
#define SRC_BGP_ROUTING_INSTANCE_PEER_MANAGER_H_

#include <boost/asio/ip/tcp.hpp>

#include <map>
#include <string>
#include <vector>

#include "bgp/bgp_peer_key.h"
#include "bgp/ipeer.h"
#include "io/tcp_session.h"

class BgpPeer;
class BgpServer;
class RoutingInstance;
class BgpNeighborResp;
class BgpSandeshContext;

class PeerManager {
public:
    typedef std::multimap<BgpPeerKey, BgpPeer *> BgpPeerKeyMap;
    typedef std::map<std::string, BgpPeer *> BgpPeerNameMap;

    explicit PeerManager(RoutingInstance *instance) : instance_(instance) { }
    virtual ~PeerManager() { }

    virtual BgpPeer *PeerFind(std::string address) const;
    virtual BgpPeer *PeerLookup(std::string name) const;
    virtual BgpPeer *PeerLookup(TcpSession::Endpoint remote_endpoint) const;
    virtual BgpPeer *PeerLocate(BgpServer *server,
                                const BgpNeighborConfig *config);
    void PeerResurrect(std::string name);
    BgpPeer *TriggerPeerDeletion(const BgpNeighborConfig *config);
    virtual void DestroyIPeer(IPeer *ipeer);
    void ClearAllPeers();
    void ClearAllInternalPeers();

    const BgpPeer *NextPeer(const BgpPeerKey &key) const;
    size_t GetNeighborCount(std::string up_or_down);

    size_t size() { return peers_by_key_.size(); }
    const std::string &name() const;
    const RoutingInstance *instance() const { return instance_; }
    RoutingInstance *instance() { return instance_; }
    BgpServer *server() const;

    const BgpPeerKeyMap &peer_map() const { return peers_by_key_; }
    BgpPeerKeyMap *peer_map_mutable() { return &peers_by_key_; }

private:
    friend class PeerManagerTest;
    friend class BgpServerTest;

    void InsertPeerByKey(BgpPeerKey key, BgpPeer *peer);
    void RemovePeerByKey(BgpPeerKey key, BgpPeer *peer);
    void InsertPeerByName(const std::string name, BgpPeer *peer);
    void RemovePeerByName(const std::string name, BgpPeer *peer);

    BgpPeerKeyMap peers_by_key_;
    BgpPeerNameMap peers_by_name_;
    RoutingInstance *instance_;
};

#endif  // SRC_BGP_ROUTING_INSTANCE_PEER_MANAGER_H_