server.hpp

Go to the documentation of this file.

#ifndef SERVER_HPP
#define SERVER_HPP
 
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
 
#include <cstddef>
#include <functional>
#include <stdexcept>
#include <unordered_map>
 
#define NB_CONNECTION    1000
#define READ_BUFFER_SIZE 4096
 
#include "../message/message.hpp"
 
class Server
{
private:
    int         _socket  = -1;
    int         _max_fd  = -1;
    int         _next_id = 0;
    bool        _running = true;
    std::string _address;
    size_t      _port;
 
    fd_set _active;
    fd_set _readyRead;
 
    std::unordered_map<int, long long> _clients;
    std::unordered_map<long long, int> _clientsToFd;
    std::unordered_map<Message::Type, std::function<void(long long& clientID, const Message& msg)>>
        _tasks;
 
    std::vector<Message>             _msgs;
    std::unordered_map<int, Message> _partialMsgs;
 
    void _acceptNewConnection();
    bool _receiveClientMsg(const int& fd);
 
    void _clearAll();
    void _clearClient(int& fd);
 
public:
    Server();
    ~Server();
    Server(const std::string& address, size_t port);
    void start(const size_t& port = 0);
 
    void defineAction(const Message::Type& messageType,
                      const std::function<void(long long& clientID, const Message& msg)>& action);
 
    void sendTo(const Message& message, long long clientID);
    void sendToArray(const Message& message, std::vector<long long> clientIDs);
    void sendToAll(const Message& message);
 
    void update();
    void stop();
};
#endif