nexthop_server.h

Go to the documentation of this file.

/*
 * NexthopDBServer class: represents the master server interface that
 *                        has a Nexthop DB store and a list of clients.
 */
#ifndef _AGENT_NHS_NEXTHOP_SERVER_H_
#define _AGENT_NHS_NEXTHOP_SERVER_H_

#include <boost/array.hpp>
#include <boost/asio.hpp>
#include <boost/asio/signal_set.hpp>
#include <boost/bind.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/scoped_ptr.hpp>
#include <boost/shared_ptr.hpp>
#include <cstdio>
#include <iostream>
#include "io/usock_server.h"
#include "nexthop_client.h"
#include "nexthop_entry.h"
#include <tbb/mutex.h>

/*
 * The main server object for serving nexthops to connected clients.
 */
class NexthopDBServer
{
  public:

    typedef std::map <uint64_t, NexthopDBClient::ClientPtr> ClientDB;
    typedef std::map <uint64_t, NexthopDBClient::ClientPtr>::iterator
      ClientIterator;
    typedef std::map <std::string, NexthopDBEntry::NexthopPtr> NexthopDB;
    typedef std::map <std::string, NexthopDBEntry::NexthopPtr>::iterator
        NexthopIterator;

    NexthopDBServer(boost::asio::io_context &io, const std::string &path);

    NexthopDBEntry::NexthopPtr FindOrCreateNexthop(const std::string &str);
    void FindAndRemoveNexthop(const std::string &str);
    void Run();
    void EventHandler(UnixDomainSocketServer *, UnixDomainSocketSession *,
                      UnixDomainSocketServer::Event);
    void RemoveClient(uint64_t);

  private:
    void AddClient(NexthopDBClient::ClientPtr cl);
    void TriggerClients();
    void AddNexthop(NexthopDBEntry::NexthopPtr nh);
    void RemoveNexthop(NexthopDBEntry::NexthopPtr nh);

    boost::asio::io_context &io_service_;
    std::string endpoint_path_;
    boost::scoped_ptr<UnixDomainSocketServer> io_server_;
    NexthopDB nexthop_table_;
    ClientDB client_table_;
    tbb::mutex mutex_;
};

#endif