TCPServerBase.h

#ifndef _ots_TCPServerBase_h_
#define _ots_TCPServerBase_h_

#include <future>
#include <map>
#include <string>
#include <vector>
#include "otsdaq/NetworkUtilities/TCPSocket.h"

namespace ots
{
class TCPServerBase : public TCPSocket
{
  public:
    TCPServerBase(int serverPort, unsigned int maxNumberOfClients);
    virtual ~TCPServerBase(void);

    void startAccept(void);
    void broadcastPacket(const char* message, std::size_t length);
    void broadcastPacket(const std::string& message);
    void broadcast(const char* message, std::size_t length);
    void broadcast(const std::string& message);
    void broadcast(const std::vector<char>& message);
    void broadcast(const std::vector<uint16_t>& message);

  protected:
    virtual void acceptConnections() = 0;

    void closeClientSocket(int socket);
    template<class T>
    T* acceptClient(bool blocking = true)
    {
        int socketId = accept(blocking);
        fConnectedClients.emplace(socketId, new T(socketId));
        return dynamic_cast<T*>(fConnectedClients[socketId]);
    }

    std::promise<bool>        fAcceptPromise;
    std::map<int, TCPSocket*> fConnectedClients;
    const int                 E_SHUTDOWN = 0;

  private:
    void closeClientSockets(void);
    int  accept            (bool blocking = true);
    void shutdownAccept    (void);

    const int         fMaxConnectionBacklog = 5;
    //unsigned int      fMaxNumberOfClients;
    std::atomic_bool  fAccept;
    std::future<bool> fAcceptFuture;
};
}

#endif